Fix: Update image generation and clean config files

agent.py

@@ -48,46 +48,33 @@ def call_groq(system_prompt, user_input):
 
 import time
 
-def generate_ad_image(prompt, model="gpt-image-1", retries=3, backoff=2):
+def generate_ad_image(prompt, n_images=1, size="1024x1024", model="dall-e-3"):
     headers = {
         "Authorization": f"Bearer {OPENAI_API_KEY}",
         "Content-Type": "application/json"
     }
     payload = {
         "model": model,
-        "messages": [
-            {"role": "system", "content": "You are an expert image generator. Provide only a valid image URL, no text."},
-            {"role": "user", "content": prompt}
-        ]
+        "prompt": prompt,
+        "n": n_images,
+        "size": size
     }
 
-    for attempt in range(retries):
-        response = requests.post(OPENAI_API_URL, json=payload, headers=headers)
-        if response.status_code == 200:
-            data = response.json()
-            if 'choices' not in data or not data['choices']:
-                raise Exception(f"{model} returned no choices.")
-
-            image_urls = []
-            for choice in data['choices']:
-                content = choice['message'].get('content', '')
-                urls = [word for word in content.split() if word.startswith('http')]
-                image_urls.extend(urls)
-
-            if not image_urls:
-                raise Exception(f"No valid image URLs found in {model} response.")
-
-            return image_urls
-
-        elif response.status_code >= 500:
-            # server error → retry after backoff
-            print(f"Attempt {attempt + 1} failed with 500 error. Retrying in {backoff} seconds...")
-            time.sleep(backoff)
-        else:
-            # client error → don’t retry
-            raise Exception(f"{model} API error: {response.status_code} - {response.text}")
-
-    raise Exception(f"{model} failed after {retries} retries due to server errors.")
+    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()
+    print("OpenAI API raw response:", data)  # 👈 ADDED for debugging
+
+    if 'data' not in data or not data['data']:
+        raise Exception("OpenAI API returned no image data. Check prompt, model, or account limits.")
+
+    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. Possible safety filter block or model issue.")
+
+    return image_urls
 
 def ad_copy_agent(product, description, audience, tone):
     system_prompt = load_prompt("ad_copy_prompt.txt")