Fix: Update image generation and clean config files

agent.py

@@ -9,7 +9,7 @@ if not GROQ_API_KEY:
     raise ValueError("GROQ_API_KEY is missing! Set it in Hugging Face Secrets.")
 
 
-OPENAI_API_URL = "https://api.openai.com/v1/images/generations"
+OPENAI_API_URL = "https://api.openai.com/v1/chat/completions"
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")  # Set this in Hugging Face Secrets
 
 if not OPENAI_API_KEY:
@@ -46,16 +46,17 @@ def call_groq(system_prompt, user_input):
 
     return data['choices'][0]['message']['content']
 
-def generate_ad_image(prompt, n_images=1, size="1024x1024", model="gpt-image-1"):
+def generate_ad_image(prompt, model="gpt-image-1"):
     headers = {
         "Authorization": f"Bearer {OPENAI_API_KEY}",
         "Content-Type": "application/json"
     }
     payload = {
-        "model": model,        # Add model explicitly (dall-e-3 recommended)
-        "prompt": prompt,
-        "n": n_images,
-        "size": size
+        "model": model,
+        "messages": [
+            {"role": "system", "content": "You are an expert image generator. Provide only a valid image URL, no text."},
+            {"role": "user", "content": prompt}
+        ]
     }
 
     response = requests.post(OPENAI_API_URL, json=payload, headers=headers)
@@ -63,11 +64,17 @@ def generate_ad_image(prompt, n_images=1, size="1024x1024", model="gpt-image-1")
         raise Exception(f"OpenAI API error: {response.status_code} - {response.text}")
 
     data = response.json()
-    if 'data' not in data or not data['data']:
-        raise Exception("OpenAI API returned no image data.")
+    if 'choices' not in data or not data['choices']:
+        raise Exception("OpenAI API returned no choices.")
+
+    # Extract image URL(s) from message content
+    image_urls = []
+    for choice in data['choices']:
+        content = choice['message'].get('content', '')
+        # Simple heuristic to extract URLs
+        urls = [word for word in content.split() if word.startswith('http')]
+        image_urls.extend(urls)
 
-    # Extract the URLs cleanly
-    image_urls = [item.get('url') for item in data['data'] if 'url' in item]
     if not image_urls:
         raise Exception("No valid image URLs found in OpenAI API response.")