Update app.py

app.py

@@ -1,7 +1,6 @@
 import os
 import gradio as gr
 import json
-import time
 import traceback
 import io
 import base64
@@ -10,7 +9,6 @@ from PIL import Image, ImageEnhance, ImageFilter
 # --- Environment Configuration ---
 GEMINI_KEY = os.environ.get("GEMINI_KEY", "")
 DEFAULT_PORT = int(os.environ.get("PORT", 7860))
-API_TIMEOUT = 30  # seconds
 
 # --- Style Template Optimization ---
 BASE_TEMPLATE = """Describe this design as a concise Flux 1.1 Pro prompt focusing on:
@@ -52,11 +50,11 @@ def generate_prompt(image, api_key, style, creativity, neg_prompt, aspect, color
     try:
         # Step 1: Input Validation
         if not image:
-            return "⚠️ Please upload an image.", None, None
+            return "⚠️ Please upload an image.", None, None, "⚠️ No image uploaded."
 
         api_key = api_key or GEMINI_KEY
         if not api_key:
-            return "🔑 API key required - set in env (GEMINI_KEY) or input field.", None, None
+            return "🔑 API key required - set in env (GEMINI_KEY) or input field.", None, None, "🔑 Missing Gemini API key."
 
         # Step 2: Gemini Setup
         try:
@@ -64,12 +62,12 @@ def generate_prompt(image, api_key, style, creativity, neg_prompt, aspect, color
             genai.configure(api_key=api_key)
             model = genai.GenerativeModel("gemini-1.5-pro")
         except ImportError:
-            return "🚫 Failed to import google.generativeai. Install with: pip install google-generativeai", None, None
+            return "", None, None, "🚫 Failed to import google.generativeai. Install with: pip install google-generativeai"
         except Exception as e:
             if "authentication" in str(e).lower():
-                return "🔐 Invalid API key or authentication error", None, None
+                return "", None, None, "🔐 Invalid API key or authentication error."
             else:
-                return f"⚙️ API initialization error: {str(e)}", None, None
+                return "", None, None, f"⚙️ API initialization error: {str(e)}"
 
         # Step 3: Preprocess Image
         try:
@@ -78,14 +76,14 @@ def generate_prompt(image, api_key, style, creativity, neg_prompt, aspect, color
             img.save(img_bytes, format="PNG")
             img_b64 = base64.b64encode(img_bytes.getvalue()).decode()
         except Exception as e:
-            return f"🖼️ Image preparation failed: {str(e)}", None, None
+            return "", None, None, f"🖼️ Image preparation failed: {str(e)}"
 
         # Step 4: Build Instruction Prompt
         try:
             instruction = f"{STYLE_INSTRUCTIONS[style]}\nAVOID: {neg_prompt}\n"
             instruction += f"ASPECT: {aspect}, COLORS: {color_mode}, DPI: {dpi}\n"
         except KeyError:
-            return "🛠️ Invalid style selected. Please choose from available options.", None, None
+            return "", None, None, "🛠️ Invalid style selected. Please choose from available options."
 
         # Step 5: Call Gemini API
         try:
@@ -95,20 +93,22 @@ def generate_prompt(image, api_key, style, creativity, neg_prompt, aspect, color
             )
             raw_prompt = response.text
         except Exception as e:
-            return f"🤖 Prompt generation failed: {str(e)}", None, None
+            return "", None, None, f"🤖 Prompt generation failed: {str(e)}"
 
         # Step 6: Token Stats
         input_tokens = len(img_b64) // 4
         output_tokens = len(raw_prompt.split())
 
-        return raw_prompt, {"score": 8, "issues": [], "suggestions": []}, {
-            "Input Tokens": input_tokens,
-            "Output Tokens": output_tokens
-        }
+        return (
+            raw_prompt,
+            {"score": 8, "issues": [], "suggestions": []},
+            {"Input Tokens": input_tokens, "Output Tokens": output_tokens},
+            "✅ Prompt generated successfully!"
+        )
 
     except Exception as e:
         traceback.print_exc()
-        return f"💥 Unexpected error: {str(e)}", None, None
+        return "", None, None, f"💥 Unexpected error: {str(e)}"
 
 
 # --- Modern Copy Function ---
@@ -178,14 +178,8 @@ def build_interface():
                 img_input, api_key, style, creativity,
                 neg_prompt, aspect, color_mode, dpi
             ],
-            outputs=[prompt_output, quality_report, token_stats],
+            outputs=[prompt_output, quality_report, token_stats, status_msg],
             api_name="generate"
-        ).success(
-            fn=lambda: "✅ Prompt generated successfully!",
-            outputs=status_msg
-        ).error(
-            fn=lambda err: f"❌ Error: {err}",
-            outputs=status_msg
         )
 
         copy_btn.click(