Update app.py

app.py

@@ -717,16 +717,15 @@ def create_demo():
                 return "error", f"Unexpected worker error: {exc}", ""
 
         def _start_processing(url, prompt, key):
-            # set state to busy and launch the worker in the same call
-            return "busy", None, None   # temporary values; the real result will replace them
-        
+            return "busy", None, None
+
         submit_btn.click(
             fn=_start_processing,
             inputs=[url_input, custom_prompt, api_key],
             outputs=[status_state, output_md, preview_path_state],
-            queue=False,               # this tiny wrapper runs instantly
+            queue=False,
         )
-        
+
         submit_btn.click(
             fn=worker,
             inputs=[url_input, custom_prompt, api_key],
@@ -743,16 +742,14 @@ def create_demo():
                 "done": "Done!",
                 "error": "Retry",
             }
-            return labels.get(s, "Submit")  # Default case
+            return labels.get(s, "Submit")
 
-        # after submit_btn / clear_btn definitions
         cancel_btn = gr.Button("Cancel", variant="secondary")
-        
+
         def _cancel():
-            # abort any queued jobs
             demo.cancel_all()
             return "idle", "", "", "idle", "Idle", "", ""
-        
+
         cancel_btn.click(
             fn=_cancel,
             inputs=[],
@@ -762,60 +759,58 @@ def create_demo():
             ],
             queue=False,
         )
-            
+
         status_state.change(fn=lambda s: _btn_label_for_status(s), inputs=[status_state], outputs=[submit_btn])
 
-        # map status to progress text
         def status_to_progress_text(s):
-            return {"idle":"Idle","busy":"Processing…","done":"Completed","error":"Error — see output"}.get(s, s)
+            return {"idle": "Idle", "busy": "Processing…", "done": "Completed", "error": "Error — see output"}.get(s, s)
         status_state.change(fn=status_to_progress_text, inputs=[status_state], outputs=[progress_md])
 
-        preview_cache = {}   # module‑level dict: url → (preview_path, is_video)
+        preview_cache = {}
 
         def _make_preview(url: str, raw: bytes) -> str:
-            # reuse existing preview if we already have one for this URL
             if url in preview_cache:
                 return preview_cache[url][0]
-        
-            if determine_media_type(url)[1]:          # video
+
+            if determine_media_type(url)[1]:
                 tmp = _temp_file(raw, suffix=ext_from_src(url) or ".mp4")
                 preview = _convert_video_for_preview_if_needed(tmp)
-                # keep the raw temp file for later clean‑up
                 preview_cache[url] = (preview, True)
-            else:                                      # image
+            else:
                 preview = _temp_file(convert_to_jpeg_bytes(raw, base_h=1024), suffix=".jpg")
                 preview_cache[url] = (preview, False)
             return preview
 
-        preview_path_state   = gr.State("")          # current preview file
-        prev_preview_state   = gr.State("")          # remembers the last file we showed
-        
-    def apply_preview(path: str, last_path: str):
-        if path == last_path:
-            return gr.update(), gr.update(), ""
-        
-        if not path:
-            return gr.update(), gr.update(), ""
-        
-        try:
-            if any(path.lower().endswith(ext) for ext in IMAGE_EXTS):
+        preview_path_state = gr.State("")
+        prev_preview_state = gr.State("")
+
+        def apply_preview(path: str, last_path: str):
+            if path == last_path:
+                return gr.update(), gr.update(), ""
+
+            if not path:
+                return gr.update(), gr.update(), ""
+
+            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."
+                
+                img = Image.open(path)
+                img.verify()
                 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."
-            
-            img = Image.open(path)
-            img.verify()
-            return gr.update(value=path, visible=True), gr.update(value=None, visible=False), "Preview updated."
-            
-        except Exception as e:
-            print(f"Failed to update preview: {e}")
-            return gr.update(value=None, visible=False), gr.update(value=None, visible=False), ""
-    
-    preview_path_state.change(fn=apply_preview, inputs=[preview_path_state], outputs=[preview_image, preview_video, preview_status])
+                
+            except Exception as e:
+                print(f"Failed to update preview: {e}")
+                return gr.update(value=None, visible=False), gr.update(value=None, visible=False), ""
+
+        preview_path_state.change(fn=apply_preview, inputs=[preview_path_state], outputs=[preview_image, preview_video, preview_status])
 
     demo.queue()
-return demo
+    return demo
 
 if __name__ == "__main__":
     create_demo().launch(share=False, server_name="0.0.0.0", server_port=7860, max_threads=8)
+