Fix: Update image generation and clean config files

agent.py

@@ -68,13 +68,31 @@ def ad_copy_agent(product, description, audience, tone):
     user_input = f"Product: {product}\nDescription: {description}\nAudience: {audience}\nTone: {tone}"
     output = call_groq(system_prompt, user_input)
     
-    # Split the output into ad copy and image prompt
+    # Split out the image prompt section
     parts = output.split('Image Prompt:')
-    ad_copy = parts[0].replace('Ad Copy:', '').strip()
+    ad_section = parts[0].strip()
     image_prompt = parts[1].strip() if len(parts) > 1 else ''
     
-    return ad_copy, image_prompt
+    # Extract clean ad copies
+    ad_copies = []
+    current_ad = []
+    for line in ad_section.split('\n'):
+        line = line.strip()
+        if line.startswith('Ad Copy'):
+            if current_ad:
+                ad_copies.append(' '.join(current_ad).strip())
+                current_ad = []
+        else:
+            current_ad.append(line)
+    if current_ad:
+        ad_copies.append(' '.join(current_ad).strip())
+
+    # Join ad copies as one block or return as list
+    ad_copy_text = '\n'.join(ad_copies)
     
+    # ✅ Return ad copy block + separate image prompt
+    return ad_copy_text, image_prompt
+
 def sentiment_agent(social_data):
     system_prompt = load_prompt("sentiment_prompt.txt")
     return call_groq(system_prompt, social_data)

config/system_prompts/ad_copy_prompt.txt

@@ -10,10 +10,10 @@ Rules for Ad Copy:
 - If details are missing, infer creatively.
 
 Rules for Image Prompt:
-- Write one vivid, concise description of the image that should accompany the ads.
-- Focus on mood, setting, colors, and key visual elements.
-- Make sure it matches the product, audience, and tone.
-- Keep it under 100 words.
+- Write **one** clear, concise image description that visually represents the **main idea or mood** across the three ads.
+- Focus on mood, setting, colors, and one central visual concept.
+- Do NOT generate multiple image prompts.
+- Keep it under 50 words.
 
 Output format:
 Ad Copy 1:
@@ -26,4 +26,4 @@ Ad Copy 3:
 [ad copy text]
 
 Image Prompt:
-[visual description]
+[visual description, one prompt only]

config/system_prompts/image_prompt_style.txt

@@ -0,0 +1,15 @@
+You are a senior marketing art director specializing in digital ads. Your job is to craft vivid, concise, and on-brand image prompts based on the ad copy and product details provided.
+
+Rules:
+- Focus on one clear, central visual idea that supports the ads.
+- Include mood, setting, colors, and key objects or people.
+- Avoid abstract or generic descriptions — be specific and marketing-driven.
+- Keep it under 50 words.
+- Make sure the style fits the product’s audience and the tone of the ads.
+- Do NOT generate multiple prompts — just one unified description.
+
+Example:
+“A joyful young woman jogging through a sunny park in stylish sneakers, vibrant colors, dynamic motion, uplifting mood.”
+
+Purpose:
+The prompt you generate will be sent to an AI image generator (like OpenAI’s DALL·E) to create marketing images that match the campaign.