Update app.py

app.py

@@ -52,11 +52,11 @@ def generate_prompt(image, api_key, style, creativity, neg_prompt, aspect, color
     try:
         # Step 1: Input Validation
         if not image:
-            return {"error": "⚠️ Please upload an image."}, None, None
+            return "⚠️ Please upload an image.", None, None
 
         api_key = api_key or GEMINI_KEY
         if not api_key:
-            return {"error": "🔑 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
 
         # Step 2: Gemini Setup
         try:
@@ -64,12 +64,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 {"error": "🚫 Failed to import google.generativeai. Install with: pip install google-generativeai"}, None, None
+            return "🚫 Failed to import google.generativeai. Install with: pip install google-generativeai", None, None
         except Exception as e:
             if "authentication" in str(e).lower():
-                return {"error": "🔐 Invalid API key or authentication error"}, None, None
+                return "🔐 Invalid API key or authentication error", None, None
             else:
-                return {"error": f"⚙️ API initialization error: {str(e)}"}, None, None
+                return f"⚙️ API initialization error: {str(e)}", None, None
 
         # Step 3: Preprocess Image
         try:
@@ -78,14 +78,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 {"error": f"🖼️ Image preparation failed: {str(e)}"}, None, None
+            return f"🖼️ Image preparation failed: {str(e)}", None, None
 
         # 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 {"error": "🛠️ Invalid style selected. Please choose from available options."}, None, None
+            return "🛠️ Invalid style selected. Please choose from available options.", None, None
 
         # Step 5: Call Gemini API
         try:
@@ -95,22 +95,20 @@ def generate_prompt(image, api_key, style, creativity, neg_prompt, aspect, color
             )
             raw_prompt = response.text
         except Exception as e:
-            return {"error": f"🤖 Prompt generation failed: {str(e)}"}, None, None
+            return f"🤖 Prompt generation failed: {str(e)}", None, None
 
-        # Step 6: Quality & Token Stats
-        validation = {"score": 8, "issues": [], "suggestions": []}
+        # Step 6: Token Stats
         input_tokens = len(img_b64) // 4
         output_tokens = len(raw_prompt.split())
 
-        return {
-            "prompt": raw_prompt,
-            "validation": validation,
-            "stats": {"input": input_tokens, "output": output_tokens}
-        }, validation, {"tokens": f"Input: {input_tokens}, Output: {output_tokens}"}
+        return raw_prompt, {"score": 8, "issues": [], "suggestions": []}, {
+            "Input Tokens": input_tokens,
+            "Output Tokens": output_tokens
+        }
 
     except Exception as e:
         traceback.print_exc()
-        return {"error": f"💥 Unexpected error: {str(e)}"}, None, None
+        return f"💥 Unexpected error: {str(e)}", None, None
 
 
 # --- Modern Copy Function ---
@@ -182,9 +180,12 @@ def build_interface():
             ],
             outputs=[prompt_output, quality_report, token_stats],
             api_name="generate"
-        ).then(
+        ).success(
             fn=lambda: "✅ Prompt generated successfully!",
             outputs=status_msg
+        ).error(
+            fn=lambda err: f"❌ Error: {err}",
+            outputs=status_msg
         )
 
         copy_btn.click(