Fix: Update image generation and clean config files

agent.py

@@ -1,6 +1,7 @@
 import os
 import requests
 
+
 GROQ_API_URL = "https://api.groq.com/openai/v1/chat/completions"
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")  # Secure key from Hugging Face Secrets
 
@@ -45,23 +46,33 @@ def call_groq(system_prompt, user_input):
 
     return data['choices'][0]['message']['content']
 
-def generate_ad_image(prompt, n_images=1, size="1024x1024"):
+def generate_ad_image(prompt, n_images=1, size="1024x1024", 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
+        "size": size,
+        "response_format": "url"  # Ensures we get back URLs
     }
+
     response = requests.post(OPENAI_API_URL, json=payload, headers=headers)
     if response.status_code != 200:
         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.")
-    return [item['url'] for item in data['data']]
+
+    # 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.")
+
+    return image_urls
 
 def ad_copy_agent(product, description, audience, tone):
     system_prompt = load_prompt("ad_copy_prompt.txt")