Fix: Update image generation and clean config files

agent.py

@@ -6,6 +6,14 @@ GROQ_API_KEY = os.getenv("GROQ_API_KEY")  # Secure key from Hugging Face Secrets
 
 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_KEY = os.getenv("OPENAI_API_KEY")  # Set this in Hugging Face Secrets
+
+if not OPENAI_API_KEY:
+    raise ValueError("OPENAI_API_KEY is missing! Set it in Hugging Face Secrets.")
+
     
 # Helper function to load system prompt from txt files
 def load_prompt(filename):
@@ -37,11 +45,36 @@ def call_groq(system_prompt, user_input):
 
     return data['choices'][0]['message']['content']
 
+def generate_ad_image(prompt, n_images=1, size="512x512"):
+    headers = {
+        "Authorization": f"Bearer {OPENAI_API_KEY}",
+        "Content-Type": "application/json"
+    }
+    payload = {
+        "prompt": prompt,
+        "n": n_images,
+        "size": size
+    }
+    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']]
+
 def ad_copy_agent(product, description, audience, tone):
     system_prompt = load_prompt("ad_copy_prompt.txt")
     user_input = f"Product: {product}\nDescription: {description}\nAudience: {audience}\nTone: {tone}"
-    return call_groq(system_prompt, user_input)
-
+    output = call_groq(system_prompt, user_input)
+    
+    # Split the output into ad copy and image prompt
+    parts = output.split('Image Prompt:')
+    ad_copy = parts[0].replace('Ad Copy:', '').strip()
+    image_prompt = parts[1].strip() if len(parts) > 1 else ''
+    
+    return ad_copy, image_prompt
+    
 def sentiment_agent(social_data):
     system_prompt = load_prompt("sentiment_prompt.txt")
     return call_groq(system_prompt, social_data)

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from agent import ad_copy_agent, sentiment_agent, timing_agent
+from agent import ad_copy_agent, generate_ad_image, sentiment_agent, timing_agent
 
 # Gradio UI definition
 with gr.Blocks() as demo:
@@ -11,8 +11,15 @@ with gr.Blocks() as demo:
         audience = gr.Textbox(label="Audience")
         tone = gr.Textbox(label="Tone")
         ad_output = gr.Textbox(label="Generated Ads")
-        ad_button = gr.Button("Generate Ads")
-        ad_button.click(ad_copy_agent, [prod, desc, audience, tone], ad_output)
+        image_output = gr.Image(label="Generated Ad Image")
+        ad_button = gr.Button("Generate Ads + Image")
+
+        def generate_ad_and_image(product, description, audience, tone):
+            ad_copy, image_prompt = ad_copy_agent(product, description, audience, tone)
+            image_urls = generate_ad_image(image_prompt)
+            return ad_copy, image_urls[0]  # Return first image URL
+
+        ad_button.click(generate_ad_and_image, [prod, desc, audience, tone], [ad_output, image_output])
 
     with gr.Tab("Sentiment Analyzer"):
         social_input = gr.Textbox(label="Social Media Mentions / Input")

config/system_prompts/ad_copy_prompt.txt

@@ -1,6 +1,6 @@
-You are a senior marketing copywriter specializing in digital ads. Your job is to generate highly persuasive, creative, and audience-tailored ad copy.
+You are a senior marketing copywriter specializing in digital ads. Your job is to generate highly persuasive, creative, and audience-tailored ad copy **and** a vivid image description prompt for visual generation.
 
-Rules:
+Rules for Ad Copy:
 - Write 3 distinct ad variations.
 - Focus on **benefits**, not just features.
 - Use emotionally engaging language.
@@ -8,3 +8,22 @@ Rules:
 - Adapt tone (friendly, professional, funny, bold).
 - Keep ads under 280 characters.
 - 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.
+
+Output format:
+Ad Copy 1:
+[ad copy text]
+
+Ad Copy 2:
+[ad copy text]
+
+Ad Copy 3:
+[ad copy text]
+
+Image Prompt:
+[visual description]

config/system_prompts/sentiment_prompt.txt

@@ -4,4 +4,4 @@ You are a social media analyst. Analyze user input or a batch of social media me
 2. A brief summary of why (1-2 sentences).
 3. Suggested marketing action (e.g., amplify, address complaints, run promo).
 
-Be concise but actionable.
+Be concise but actionable.

config/system_prompts/timing_prompt.txt

@@ -4,4 +4,4 @@ Rules:
 - Base suggestions on typical platform engagement patterns.
 - Mention time zones.
 - Briefly explain why these times work.
-- Suggest one experimental or underused time.
+- Suggest one experimental or underused time.

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+groq
+requests