Update app.py

app.py

@@ -151,17 +151,19 @@ def build_interface():
                 token_stats = gr.JSON(label="🧮 Token Usage", visible=True)
 
         # Event bindings
-        gen_btn.click(
+       gen_btn.click(
             fn=generate_prompt,
             inputs=[img_input, api_key, style, creativity, neg_prompt, aspect, color_mode, dpi],
-            outputs=[prompt_output, quality_report, token_stats],
-            api_name="generate"
+            outputs=None
         ).then(
-            fn=lambda r: update_status({"prompt": r}) if isinstance(r, str) else update_status(r),
+            fn=format_generation_response,
+            inputs=None,
+            outputs=[prompt_output, quality_report, token_stats]
+        ).then(
+            fn=update_status,
             inputs=[prompt_output],
-            outputs=status_msg
+            outputs=[status_msg]
         )
-
     return app
 
 # --- Launch App ---