Update app.py

app.py

@@ -419,74 +419,58 @@ def create_demo():
                 preview_video = gr.Video(label="Preview Video", elem_classes="preview_media", visible=False)
             with gr.Column(scale=2):
                 url_input = gr.Textbox(label="Image / Video URL", placeholder="https://...", lines=1)
-                with gr.Accordion("Prompt (optional)", open=False):
-                    custom_prompt = gr.Textbox(label="Prompt", lines=4, value="")
-                with gr.Accordion("Mistral API Key (optional)", open=False):
-                    api_key = gr.Textbox(label="API Key", type="password", max_lines=1)
-                with gr.Row():  # Buttons in the same row
+                custom_prompt = gr.Textbox(label="Prompt (optional)", lines=4, value="")
+                api_key = gr.Textbox(label="Mistral API Key (optional)", type="password", max_lines=1)
+                with gr.Row():
                     submit_btn = gr.Button("Submit")
                     clear_btn = gr.Button("Clear")
                 output_md = gr.Markdown("")
                 status_state = gr.State("idle")
 
         def load_preview(url: str):
-            empty_img = gr.update(value=None, visible=False)
-            empty_vid = gr.update(value=None, visible=False)
-
-            if not url: return empty_img, empty_vid
-            
-            if is_remote(url):
-                head = safe_head(url)
-                if head and any(url.lower().endswith(ext) for ext in VIDEO_EXTS):
-                    return empty_img, gr.update(value=url, visible=True)
-                    
-                try:
+            if not url:
+                return gr.update(value=None, visible=False), gr.update(value=None, visible=False)
+            if is_remote(url) and any(url.lower().endswith(ext) for ext in VIDEO_EXTS):
+                return gr.update(value=None, visible=False), gr.update(value=url, visible=True)
+            try:
+                if is_remote(url):
                     r = safe_get(url)
-                    img = Image.open(BytesIO(r.content))
-                    if getattr(img, "is_animated", False): img.seek(0)
-                    return gr.update(value=img.convert("RGB"), visible=True), empty_vid
-                except Exception:
-                    return empty_img, empty_vid
-            return empty_img, empty_vid
+                    img_bytes = r.content
+                else:
+                    with open(url, "rb") as f:
+                        img_bytes = f.read()
+                img = Image.open(BytesIO(img_bytes))
+                if getattr(img, "is_animated", False):
+                    img.seek(0)
+                return gr.update(value=img.convert("RGB"), visible=True), gr.update(value=None, visible=False)
+            except Exception:
+                return gr.update(value=None, visible=False), gr.update(value=None, visible=False)
 
         url_input.change(fn=load_preview, inputs=[url_input], outputs=[preview_image, preview_video])
 
         def clear_all():
-            return "", None, None, "idle"
-        
-        clear_btn.click(fn=clear_all, inputs=[], outputs=[url_input, preview_image, preview_video, status_state])
+            return "", None, None, "idle", ""
+
+        clear_btn.click(fn=clear_all, inputs=[], outputs=[url_input, preview_image, preview_video, status_state, output_md])
 
         def worker(url: str, prompt: str, key: str, progress=gr.Progress()):
             try:
-                if not url: 
-                    raise ValueError("No URL provided.")
-                
-                src = url
+                if not url:
+                    return ("error", "**Error:** No URL provided.")
                 progress(0, desc="Starting processing...")
-                
-                result = run_blocking_in_thread(
-                    process_media, 
-                    src, 
-                    prompt or "", 
-                    key or "", 
-                    progress
-                )
-
-                if hasattr(result, "result"):
-                    result = result.result()
-                
-                return result if result else "Error: no result returned."
-    
+                fut = run_blocking_in_thread(process_media, url, prompt or "", key or "", progress)
+                res = fut.result()
+                status = "done" if not (isinstance(res, str) and res.lower().startswith("error")) else "error"
+                return (status, res if isinstance(res, str) else str(res))
             except Exception as e:
-                return f"Unexpected error in worker: {e}"
+                return ("error", f"Unexpected worker error: {e}")
 
-        submit_btn.click(fn=worker, inputs=[url_input, custom_prompt, api_key], outputs=[output_md], queue=True).then(
-            fn=lambda res: ("error", f"**Error:** {res}") if isinstance(res, str) and res.lower().startswith("error") else ("done", res),
-            inputs=[output_md],
-            outputs=[status_state, output_md],
-        )
+        submit_btn.click(fn=worker, inputs=[url_input, custom_prompt, api_key], outputs=[status_state, output_md], queue=True)
+
+        def btn_label_from_state(s):
+            return _btn_label_for_status(s)
 
-        status_state.change(fn=lambda s: _btn_label_for_status(s), inputs=[status_state], outputs=[submit_btn])
+        status_state.change(fn=btn_label_from_state, inputs=[status_state], outputs=[submit_btn])
 
     return demo