idea to image

dalle3/main.py

@@ -7,7 +7,7 @@
 from selenium.webdriver.support import expected_conditions as EC
 from undetected_chromedriver import Chrome, ChromeOptions
 
-
+# Main class
 class Dalle:
     """
     A class used to interact with the DALL-E 3 Unofficial API

example.py

@@ -4,7 +4,7 @@
 from dalle3 import Dalle
 
 # Define cookie using env or empty string
-cookie = os.getenv("BING_COOKIE") or "1Dv1qGH6TcGzEQUJqU6AfxnMfxPR4V9yCtKk8ox2kpl-_YwmKzM9CnzPmlge158c0DTSDAEqrDV5itj3Ll0GyuaRrWmXf0Gk_2FIZuL2NvCLUrCF3VALngS9qw9vESZ9EY9vqVVVvjGoIJpxf9HEz6OV3fih4dw_Abc7GrfRC7p8iRDiCmotsgy9lPKf6sSPg9yy2lMTSU9-MNmkga0CACSZsV4La9NU5V5nNowUAMhU"
+cookie = os.getenv("BING_COOKIE") or ""
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)

idea_to_image.py

@@ -54,8 +54,12 @@ def main():
     llm = OpenAIChat(args.openai_api_key)
     dalle = Dalle(args.cookie)
 
+    # Refine the prompt using the llm
+    image_to_generate = llm_prompt(args.image_to_generate)
+    refined_prompt = llm(image_to_generate)
+
     # Open the website with your query
-    dalle.open_website(llm_prompt(args.image_to_generate))
+    dalle.open_website(refined_prompt)
 
     # Get the image URLs
     urls = dalle.get_urls()

tests/dalle.py

@@ -1,9 +1,9 @@
 import pytest
 import os
 from unittest.mock import patch, MagicMock
-from dalle3.main import Dalle
+from dalle3 import Dalle
 
-test_cookie = os.getenv("BING_COOKIE") or "1Dv1qGH6TcGzEQUJqU6AfxnMfxPR4V9yCtKk8ox2kpl-_YwmKzM9CnzPmlge158c0DTSDAEqrDV5itj3Ll0GyuaRrWmXf0Gk_2FIZuL2NvCLUrCF3VALngS9qw9vESZ9EY9vqVVVvjGoIJpxf9HEz6OV3fih4dw_Abc7GrfRC7p8iRDiCmotsgy9lPKf6sSPg9yy2lMTSU9-MNmkga0CACSZsV4La9NU5V5nNowUAMhU"
+test_cookie = os.getenv("BING_COOKIE") or ""
 
 @pytest.fixture
 def dalle():