Update app.py

app.py

@@ -450,12 +450,11 @@ def create_demo():
                                 delay = float(ra) if ra is not None else delay
                             except Exception:
                                 pass
-                        # for other 4xx/5xx raise to outer except to allow retry logic
                         r.raise_for_status()
                     else:
                         with open(src, "rb") as fh:
                             return fh.read()
-                except requests.exceptions.RequestException as e:
+                except requests.exceptions.RequestException:
                     if attempt >= max_retries:
                         raise
                     time.sleep(delay)
@@ -489,20 +488,17 @@ def create_demo():
             # returns (preview_image_path, preview_video_path, status_msg)
             if not url:
                 return gr.update(value=None, visible=False), gr.update(value=None, visible=False), gr.update(value="")
-            # If remote, check head to avoid unnecessary download when it's clearly video
             try:
                 if is_remote(url):
                     head = safe_head(url)
                     if head:
                         ctype = (head.headers.get("content-type") or "").lower()
                         if ctype.startswith("video/") or any(url.lower().endswith(ext) for ext in VIDEO_EXTS):
-                            # save remote to temp to avoid Gradio re-download later
                             local = _save_preview_local(url)
                             if local:
                                 return gr.update(value=None, visible=False), gr.update(value=local, visible=True), gr.update(value=f"Remote video detected (content-type={ctype}). Showing preview if browser-playable.")
                             else:
                                 return gr.update(value=None, visible=False), gr.update(value=None, visible=False), gr.update(value=f"Remote video detected but preview download failed (content-type={ctype}).")
-                # Try image fetch and load
                 local = _save_preview_local(url)
                 if not local:
                     return gr.update(value=None, visible=False), gr.update(value=None, visible=False), gr.update(value="Preview load failed: could not fetch resource.")
@@ -510,10 +506,8 @@ def create_demo():
                     img = Image.open(local)
                     if getattr(img, "is_animated", False):
                         img.seek(0)
-                    # return local path for gr.Image (type=filepath)
                     return gr.update(value=local, visible=True), gr.update(value=None, visible=False), gr.update(value="Image preview loaded.")
                 except UnidentifiedImageError:
-                    # Not an image — maybe a video; return video preview if available
                     if any(local.lower().endswith(ext) for ext in VIDEO_EXTS) or True:
                         return gr.update(value=None, visible=False), gr.update(value=local, visible=True), gr.update(value="Non-image file — showing as video preview if playable.")
                     return gr.update(value=None, visible=False), gr.update(value=None, visible=False), gr.update(value="Preview load failed: file is not a valid image.")
@@ -528,7 +522,6 @@ def create_demo():
         clear_btn.click(fn=clear_all, inputs=[], outputs=[url_input, preview_image, preview_video, status_state, progress_md, output_md, preview_path_state])
 
         def _convert_video_for_preview(path: str) -> str:
-            # fix: check FFMPEG_BIN path, not 'path'
             if not FFMPEG_BIN or not os.path.exists(FFMPEG_BIN):
                 return path
             out_fd, out_path = tempfile.mkstemp(suffix=".mp4")
@@ -546,7 +539,7 @@ def create_demo():
                 except Exception: pass
                 return path
 
-        # Worker now returns (status_state, output_md, preview_path)
+        # Worker now returns (status_state, output_md, preview_path_state)
         def worker(url: str, prompt: str, key: str, progress=gr.Progress()):
             try:
                 if not url:
@@ -563,14 +556,12 @@ def create_demo():
                     tmp = save_bytes_to_temp(raw, suffix=ext_from_src(url) or ".mp4")
                     progress(0.18, desc="Saved video to temp; converting for preview if needed...")
                     preview_tmp = _convert_video_for_preview(tmp)
-                    # ensure preview_local is a local path to avoid Gradio re-download
                     preview_local = preview_tmp if os.path.exists(preview_tmp) else tmp
                     progress(0.25, desc="Starting video analysis...")
                     res = analyze_video_cohesive(client, tmp, prompt or "", progress=progress)
                     progress(0.98, desc="Finalizing result...")
                     try:
                         if preview_tmp != tmp and os.path.exists(preview_tmp):
-                            # leave preview_tmp for frontend; worker will clean tmp
                             pass
                     finally:
                         try: os.remove(tmp)
@@ -580,7 +571,6 @@ def create_demo():
                 elif is_img:
                     progress(0.08, desc="Fetching image bytes...")
                     raw = fetch_bytes(url, progress=progress)
-                    # save preview locally so Gradio won't re-download later
                     try:
                         preview_fd, preview_path = tempfile.mkstemp(suffix=".jpg")
                         os.close(preview_fd)
@@ -606,7 +596,6 @@ def create_demo():
                         progress(0.2, desc="Image detected — analyzing...")
                         res = analyze_image_structured(client, raw, prompt or "", progress=progress)
                         status = "done" if not (isinstance(res, str) and res.lower().startswith("error")) else "error"
-                        # save preview as image
                         try:
                             preview_fd, preview_path = tempfile.mkstemp(suffix=".jpg")
                             os.close(preview_fd)
@@ -634,8 +623,18 @@ def create_demo():
             except Exception as e:
                 return ("error", f"Unexpected worker error: {e}", "")
 
-        # submit returns (status_state, output_md, preview_path_state)
-        submit_btn.click(fn=worker, inputs=[url_input, custom_prompt, api_key], outputs=[status_state, output_md, preview_path_state], queue=True)
+        # immediate UI flip to "busy" so user sees work started
+        submit_btn.click(fn=lambda: "busy", inputs=[], outputs=[status_state])
+
+        # actual heavy work runs in the queue and shows progress (attach to progress_md)
+        submit_btn.click(
+            fn=worker,
+            inputs=[url_input, custom_prompt, api_key],
+            outputs=[status_state, output_md, preview_path_state],
+            queue=True,
+            show_progress="full",
+            show_progress_on=progress_md,
+        )
 
         # update submit button label from status
         def btn_label_from_state(s):
@@ -652,13 +651,11 @@ def create_demo():
         def apply_preview(path: str):
             if not path:
                 return gr.update(value=None, visible=False), gr.update(value=None, visible=False), ""
-            # choose whether it's image or video by extension/content
             try:
                 if any(path.lower().endswith(ext) for ext in IMAGE_EXTS):
                     return gr.update(value=path, visible=True), gr.update(value=None, visible=False), "Preview updated."
                 if any(path.lower().endswith(ext) for ext in VIDEO_EXTS):
                     return gr.update(value=None, visible=False), gr.update(value=path, visible=True), "Preview updated."
-                # try load as image
                 try:
                     Image.open(path)
                     return gr.update(value=path, visible=True), gr.update(value=None, visible=False), "Preview updated."
@@ -669,8 +666,9 @@ def create_demo():
 
         preview_path_state.change(fn=apply_preview, inputs=[preview_path_state], outputs=[preview_image, preview_video, preview_status])
 
+    # ensure global queue behavior
+    demo.queue()
     return demo
 
-
 if __name__ == "__main__":
     create_demo().launch(share=False, server_name="0.0.0.0", server_port=7860, max_threads=8)