Fix: Update image generation and clean config files

agent.py

@@ -46,7 +46,9 @@ def call_groq(system_prompt, user_input):
 
     return data['choices'][0]['message']['content']
 
-def generate_ad_image(prompt, model="gpt-image-1"):
+import time
+
+def generate_ad_image(prompt, model="gpt-image-1", retries=3, backoff=2):
     headers = {
         "Authorization": f"Bearer {OPENAI_API_KEY}",
         "Content-Type": "application/json"
@@ -59,26 +61,33 @@ def generate_ad_image(prompt, model="gpt-image-1"):
         ]
     }
 
-    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 '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)
-
-    if not image_urls:
-        raise Exception("No valid image URLs found in OpenAI API response.")
-
-    return image_urls
+    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.")
 
 def ad_copy_agent(product, description, audience, tone):
     system_prompt = load_prompt("ad_copy_prompt.txt")