Update app.py

app.py

@@ -1,321 +1,73 @@
 import gradio as gr
-import os
-from PIL import Image, ImageDraw, ImageFont, ImageStat
-import io
-import json
+from PIL import Image, ImageDraw, ImageFont
 import random
 
-# Simple function to create a working demo image
 def create_image(prompt, style):
-    """Create a simple colored image that actually works"""
+    """Create a demo image"""
+    colors = {"Professional": "#3B82F6", "Creative": "#8B5CF6", "Minimalist": "#6B7280", "Corporate": "#1E40AF", "Modern": "#059669"}
+    color = colors.get(style, "#3B82F6")
+    rgb = tuple(int(color[i:i+2], 16) for i in (1, 4, 7))
     
-    colors = {
-        "Professional": "#3B82F6",
-        "Creative": "#8B5CF6", 
-        "Minimalist": "#6B7280",
-        "Corporate": "#1E40AF",
-        "Modern": "#059669"
-    }
-    
-    color_hex = colors.get(style, "#3B82F6")
-    color_rgb = tuple(int(color_hex[i:i+2], 16) for i in (1, 4, 7))
-    
-    # Create image
-    img = Image.new('RGB', (512, 512), color_rgb)
+    img = Image.new("RGB", (512, 512), rgb)
     draw = ImageDraw.Draw(img)
     font = ImageFont.load_default()
-
-    # Use textlength for better positioning
-    title = "Marketing Image"
-    title_width = draw.textlength(title, font=font)
-    draw.text(((512 - title_width) // 2, 200), title, fill="white", font=font)
-
-    style_text = f"{style} Style"
-    style_width = draw.textlength(style_text, font=font)
-    draw.text(((512 - style_width) // 2, 230), style_text, fill="white", font=font)
-
-    prompt_short = prompt[:40] + "..." if len(prompt) > 40 else prompt
-    prompt_width = draw.textlength(prompt_short, font=font)
-    draw.text(((512 - prompt_width) // 2, 260), prompt_short, fill=(200, 200, 200), font=font)
+    
+    draw.text((50, 200), "Marketing Image", fill="white", font=font)
+    draw.text((50, 230), f"{style} Style", fill="white", font=font)
+    draw.text((50, 260), prompt[:35] + "..." if len(prompt) > 35 else prompt, fill=(200, 200, 200), font=font)
     
     return img
 
-def generate_variants(prompt, style):
-    """Generate 3 variants of the same prompt"""
-    variants = []
-    for i in range(3):
-        # Create slight variations by adding random elements
-        variant_img = create_image(prompt, style)
-        variants.append(variant_img)
-    return variants
-
 def generate_and_review(prompt, style):
-    """Main function that generates image and provides review"""
-    
+    """Generate image and review"""
     if not prompt.strip():
-        return None, "⚠️ **Please enter a description for your marketing image.**"
-
-    # Generate image
-    image = create_image(prompt, style)
+        return None, "Please enter a prompt"
     
-    # Create review with Markdown formatting
+    image = create_image(prompt, style)
     word_count = len(prompt.split())
+    quality = "Excellent" if word_count > 15 else "Good" if word_count > 8 else "Basic"
     
-    if word_count > 15:
-        quality = "Excellent"
-        quality_emoji = "🌟"
-    elif word_count > 8:
-        quality = "Good"
-        quality_emoji = "✅"
-    else:
-        quality = "Basic"
-        quality_emoji = "⚠️"
-
-    review_md = f"""### 🤖 Agent Workflow Results
-
-**Agent1 (Generator):** Created *{style.lower()}* style marketing image  
-**Agent2 (Reviewer):** Analyzed image content and quality
-
----
-
-### 📊 Analysis Summary
+    review = f"""**Agent Review Complete**
 
-**{quality_emoji} Prompt Quality:** {quality} ({word_count} words)  
-**🎨 Style Applied:** {style}  
-**📈 Marketing Suitability:** Good for *{style.lower()}* marketing campaigns  
-**🎯 Brand Alignment:** Professional marketing ready
+Agent1: Generated {style.lower()} marketing image
+Agent2: Quality analysis complete
 
----
+Prompt Quality: {quality} ({word_count} words)
+Style: {style}
+Status: Ready for use
 
-### 💡 Recommendations
-
-- Image successfully generated in **{style.lower()}** style
-- Consider adding more specific visual descriptors for enhanced results
-- Ready for marketing use across digital platforms
-
-**✅ Status:** Workflow Complete | **🚀 Ready to Deploy**
-"""
+Recommendation: {f"Great detail level!" if word_count > 15 else "Consider adding more descriptive details"}"""
     
-    return image, review_md
+    return image, review
 
-def detect_brand_tone(uploaded_img):
-    """Analyze uploaded image for brand tone"""
-    if uploaded_img is None:
-        return ""
+# Create simple interface
+with gr.Blocks(title="AI Marketing Generator") as demo:
+    gr.Markdown("# 🎨 AI Marketing Image Generator")
+    gr.Markdown("Agent1 creates → Agent2 reviews → Professional results")
     
-    try:
-        stat = ImageStat.Stat(uploaded_img.convert("RGB"))
-        avg_brightness = sum(stat.mean) / 3
+    with gr.Row():
+        with gr.Column():
+            prompt = gr.Textbox(
+                label="Marketing Image Description",
+                placeholder="A professional team meeting in a modern office",
+                lines=3
+            )
+            style = gr.Dropdown(
+                choices=["Professional", "Creative", "Minimalist", "Corporate", "Modern"],
+                value="Professional",
+                label="Brand Style"
+            )
+            generate_btn = gr.Button("Generate Image", variant="primary")
         
-        if avg_brightness > 180:
-            tone = "Creative"
-            tone_emoji = "🎨"
-            description = "Bright, vibrant imagery suggests creative, innovative branding"
-        elif avg_brightness > 100:
-            tone = "Professional"
-            tone_emoji = "💼"
-            description = "Balanced lighting indicates professional, corporate tone"
-        else:
-            tone = "Corporate"
-            tone_emoji = "🏢"
-            description = "Darker tones suggest serious, authoritative branding"
-        
-        return f"""### 🎯 Brand Tone Analysis
-
-**{tone_emoji} Detected Tone:** {tone}  
-**📊 Average Brightness:** {int(avg_brightness)}/255  
-**🧠 Analysis:** {description}
-
-**💡 Suggestion:** Use "{tone}" style for consistent brand messaging in your next generation!
-"""
-    except Exception as e:
-        return f"⚠️ **Error analyzing image:** {str(e)}"
-
-def image_to_bytes(img):
-    """Convert PIL image to bytes for download"""
-    if img is None:
-        return None
-    buf = io.BytesIO()
-    img.save(buf, format='PNG')
-    buf.seek(0)
-    return buf
-
-# Create improved Blocks interface
-with gr.Blocks(
-    title="Marketing Image Generator - AI Agent System",
-    theme=gr.themes.Soft(primary_hue="blue", secondary_hue="purple")
-) as demo:
-    
-    # Header with better styling
-    gr.Markdown("""
-    # 🎨 AI Marketing Image Generator
-    ## Agent1 generates → Agent2 reviews → ✅ Professional results
-    *Multi-agent system for creating professional marketing visuals*
-    """)
-    
-    with gr.Tab("🚀 Single Image Generation"):
-        with gr.Row():
-            # Input column with better spacing
-            with gr.Column(scale=1):
-                gr.Markdown("### 📝 Describe Your Marketing Visual")
-                
-                prompt = gr.Textbox(
-                    label="Marketing Image Description", 
-                    placeholder="e.g. A professional team meeting in a modern office with natural lighting",
-                    lines=3,
-                    max_lines=5,
-                    elem_classes=["prompt-input"]
-                )
-                
-                style = gr.Dropdown(
-                    choices=["Professional", "Creative", "Minimalist", "Corporate", "Modern"],
-                    value="Professional",
-                    label="Brand Style"
-                )
-                
-                generate_btn = gr.Button(
-                    "🚀 Generate Marketing Image", 
-                    variant="primary", 
-                    size="lg"
-                )
-            
-            # Output column with side-by-side layout
-            with gr.Column(scale=2):
-                with gr.Row():
-                    image_output = gr.Image(
-                        label="Generated Marketing Image", 
-                        type="pil", 
-                        show_label=True,
-                        height=400,
-                        show_download_button=True
-                    )
-                    
-                    review_output = gr.Markdown(
-                        label="AI Agent Review",
-                        value="*Generate an image to see detailed AI analysis*",
-                        elem_classes=["review-panel"]
-                    )
-        
-        # Examples with better formatting
-        gr.Markdown("### 💡 Example Prompts")
-        gr.Examples(
-            examples=[
-                ["Professional team collaboration with laptops in bright office", "Professional"],
-                ["Clean minimalist product showcase on white background", "Minimalist"],
-                ["Friendly customer service representative with headset", "Corporate"],
-                ["Modern workspace with plants and natural lighting", "Modern"],
-                ["Creative brainstorming session with colorful sticky notes", "Creative"]
-            ],
-            inputs=[prompt, style],
-            label="Click any example to try it out!"
-        )
-    
-    with gr.Tab("🎭 Multi-Variant Generator"):
-        gr.Markdown("### Generate 3 variants and pick the best one")
-        
-        with gr.Row():
-            with gr.Column(scale=1):
-                multi_prompt = gr.Textbox(
-                    label="Marketing Image Description",
-                    placeholder="Describe your marketing visual...",
-                    lines=2,
-                    max_lines=4
-                )
-                multi_style = gr.Dropdown(
-                    choices=["Professional", "Creative", "Minimalist", "Corporate", "Modern"],
-                    value="Professional",
-                    label="Brand Style"
-                )
-                variants_btn = gr.Button("Generate 3 Variants", variant="primary")
-            
-            with gr.Column(scale=2):
-                selected_image = gr.Image(label="✅ Selected Image", type="pil")
-                download_file = gr.File(label="📥 Download Selected Image")
-        
-        # Variant display
-        with gr.Row():
-            variant1 = gr.Image(label="Variant 1", interactive=True, height=200)
-            variant2 = gr.Image(label="Variant 2", interactive=True, height=200)
-            variant3 = gr.Image(label="Variant 3", interactive=True, height=200)
-        
-        variant_review = gr.Markdown()
-        
-        # Handle variant generation and selection
-        def handle_variant_selection(img, prompt):
-            if img is None:
-                return None, None, ""
-            
-            download_bytes = image_to_bytes(img)
-            word_count = len(prompt.split())
-            quality = "Excellent" if word_count > 15 else "Good" if word_count > 8 else "Basic"
-            
-            review = f"""### ✅ Variant Selected
-            
-**Image Quality:** {quality}  
-**Prompt Analysis:** {word_count} descriptive words  
-**Status:** Ready for download and use  
-
-💡 **Tip:** This variant is optimized for professional marketing use.
-"""
-            return img, download_bytes, review
-        
-        variants_btn.click(
-            fn=lambda p, s: generate_variants(p, s),
-            inputs=[multi_prompt, multi_style],
-            outputs=[variant1, variant2, variant3]
-        )
-        
-        # Connect variant selection
-        variant1.select(
-            fn=lambda img: handle_variant_selection(img, multi_prompt.value if 'multi_prompt' in locals() else ""),
-            inputs=[variant1],
-            outputs=[selected_image, download_file, variant_review]
-        )
-        variant2.select(
-            fn=lambda img: handle_variant_selection(img, multi_prompt.value if 'multi_prompt' in locals() else ""),
-            inputs=[variant2],
-            outputs=[selected_image, download_file, variant_review]
-        )
-        variant3.select(
-            fn=lambda img: handle_variant_selection(img, multi_prompt.value if 'multi_prompt' in locals() else ""),
-            inputs=[variant3],
-            outputs=[selected_image, download_file, variant_review]
-        )
+        with gr.Column():
+            image_output = gr.Image(label="Generated Image")
+            review_output = gr.Markdown(value="Click Generate to see review")
     
-    with gr.Tab("🔍 Brand Tone Analyzer"):
-        gr.Markdown("### Upload an existing image to analyze its brand tone")
-        
-        with gr.Row():
-            with gr.Column(scale=1):
-                uploaded_img = gr.Image(
-                    type="pil", 
-                    label="Upload Marketing Image for Analysis",
-                    height=300
-                )
-            
-            with gr.Column(scale=1):
-                tone_analysis = gr.Markdown(
-                    value="*Upload an image to see brand tone analysis*"
-                )
-        
-        uploaded_img.change(
-            fn=detect_brand_tone, 
-            inputs=uploaded_img, 
-            outputs=tone_analysis
-        )
-    
-    # Connect main generation function
     generate_btn.click(
         fn=generate_and_review,
         inputs=[prompt, style],
         outputs=[image_output, review_output]
     )
-    
-    # Footer
-    gr.Markdown("""
-    ---
-    **🤖 AI Marketing Agent System** | Agent1: Image Generation | Agent2: Quality Review | Built with Gradio
-    """)
 
 if __name__ == "__main__":
     demo.launch()