Update app.py

app.py

@@ -241,7 +241,6 @@ def chat_complete(client, model: str, messages, timeout: int = 120) -> str:
     except Exception as e:
         return f"Error during model call: {e}"
 
-
 def upload_file_to_mistral(
     client,
     path: str,
@@ -399,9 +398,8 @@ def process_media(src: str, custom_prompt: str, api_key: str, progress=gr.Progre
     except Exception as e:
         return f"Unable to determine media type or fetch file: {e}"
 
-
 # ----------------------------------------------------------------------
-# Gradio UI helpers
+# Gradio UI helpers (continued/fixed)
 # ----------------------------------------------------------------------
 css = ".preview_media img, .preview_media video { max-width: 100%; height: auto; }"
 
@@ -509,5 +507,46 @@ def create_demo():
         def worker(url: str, prompt: str, key: str, progress=gr.Progress()):
             return process_media(url or "", prompt or "", key or "", progress=progress)
 
-        def finish(result: str) -> tuple[str, str]:
-            if not result or result.lower().startswith(("error", "unhandled"))
+        def finish(result: str) -> tuple[str, dict]:
+            s = "done"
+            if not result:
+                md = "Error: no result returned."
+                s = "error"
+            elif isinstance(result, str) and result.lower().startswith("error"):
+                md = f"**Error:** {result}"
+                s = "error"
+            else:
+                md = result
+
+            return s, md
+
+        # Wire submit button: set status to busy, run worker, then finish
+        submit_btn.click(
+            fn=start_busy,
+            inputs=[],
+            outputs=[status, submit_btn],
+        )
+
+        submit_btn.click(
+            fn=worker,
+            inputs=[url_input, custom_prompt, api_key],
+            outputs=[final_md],
+            queue=True,
+        ).then(
+            fn=finish,
+            inputs=[final_md],
+            outputs=[status, final_md],
+        )
+
+        # Update button label whenever status changes
+        def btn_label_for_state(s: str) -> str:
+            return _btn_label_for_status(s)
+
+        status.change(fn=btn_label_for_state, inputs=[status], outputs=[submit_btn])
+
+    return demo
+
+
+if __name__ == "__main__":
+    demo = create_demo()
+    demo.launch(server_name="0.0.0.0", share=False)