Update app.py

app.py

@@ -127,7 +127,7 @@ def enhance_prompt_with_gemini(prompt: str, style: str) -> str:
         Return only the enhanced prompt without explanation.
         """
         
-        model = genai.GenerativeModel('gemini-1.5-flash')
+        model = genai.GenerativeModel('gemini-2.0-flash-exp')
         response = model.generate_content(enhancement_prompt)
         enhanced = response.text.strip()
         
@@ -321,7 +321,7 @@ def analyze_marketing_image_with_gemini(image_url: str, prompt: str, review_guid
             return _fallback_image_analysis(prompt, review_guidelines)
         
         # Generate analysis using Gemini Vision
-        model = genai.GenerativeModel('gemini-1.5-flash')
+        model = genai.GenerativeModel('gemini-2.0-flash-exp')
         response = model.generate_content([analysis_prompt, image])
         analysis_text = response.text
         
@@ -698,6 +698,13 @@ def gradio_generate_marketing_image(prompt: str, style: str, max_retries: int, r
         return None, "⚠️ Please enter a prompt to generate an image."
     
     try:
+        # Ensure parameters are correct types
+        max_retries = int(max_retries) if max_retries is not None else 3
+        review_threshold = float(review_threshold) if review_threshold is not None else 0.8
+        review_guidelines = str(review_guidelines) if review_guidelines is not None else ""
+        
+        logger.info(f"🎯 Starting generation with max_retries={max_retries}, threshold={review_threshold}")
+        
         # Use the complete workflow function with iteration parameters
         result_json = generate_and_review_marketing_image(
             prompt=prompt, 
@@ -759,6 +766,26 @@ with gr.Blocks(title="Marketing Image Generator MCP", theme=gr.themes.Soft()) as
                 info="Provide specific marketing guidelines for review"
             )
             
+            # Advanced settings
+            with gr.Accordion("🔧 Advanced Settings", open=False):
+                max_retries = gr.Slider(
+                    minimum=1,
+                    maximum=5,
+                    value=3,
+                    step=1,
+                    label="Max Retries",
+                    info="Maximum number of retry attempts if quality threshold not met"
+                )
+                
+                review_threshold = gr.Slider(
+                    minimum=0.0,
+                    maximum=1.0,
+                    value=0.8,
+                    step=0.1,
+                    label="Quality Threshold",
+                    info="Minimum quality score required for auto-approval"
+                )
+            
             # Generate button
             generate_btn = gr.Button("🚀 Generate Marketing Image", variant="primary", size="lg")
             
@@ -810,7 +837,7 @@ with gr.Blocks(title="Marketing Image Generator MCP", theme=gr.themes.Soft()) as
     # Event handlers
     generate_btn.click(
         fn=gradio_generate_marketing_image,
-        inputs=[prompt, style, review_guidelines],
+        inputs=[prompt, style, max_retries, review_threshold, review_guidelines],
         outputs=[image_output, review_output],
         show_progress=True
     )