Update app.py

app.py

@@ -790,95 +790,32 @@ def create_demo():
         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):
-            """
-            Returns:
-                (image_update, video_update, status_msg)
-        
-            * If the path is the same as the previous one we do nothing – this stops
-              the UI from flickering on every job completion.
-            * Otherwise we decide whether the file is an image or a video and show the
-              appropriate component.
-            """
-            # ------------------------------------------------------------------
-            # 1️⃣  No‑op when the preview hasn’t changed
-            # ------------------------------------------------------------------
-            if path == last_path:
-                # keep the current UI state unchanged
-                return gr.update(), gr.update(), ""
-        
-            # ------------------------------------------------------------------
-            # 2️⃣  Nothing to show
-            # ------------------------------------------------------------------
-            if not path:
-                return (
-                    gr.update(value=None, visible=False),
-                    gr.update(value=None, visible=False),
-                    "",
-                )
-        
-            try:
-                # ------------------------------------------------------------------
-                # 3️⃣  Image preview
-                # ------------------------------------------------------------------
-                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.",
-                    )
-        
-                # ------------------------------------------------------------------
-                # 4️⃣  Video preview
-                # ------------------------------------------------------------------
-                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.",
-                    )
-        
-                # ------------------------------------------------------------------
-                # 5️⃣  Fallback – try to open as an image (covers odd extensions)
-                # ------------------------------------------------------------------
-                img = Image.open(path)
-                img.verify()                     # raises if not a valid image
-                return (
-                    gr.update(value=path, visible=True),
-                    gr.update(value=None, visible=False),
-                    "Preview updated.",
-                )
-            except Exception as e:
-                # Log the problem – the UI will simply hide both previews
-                print(f"Failed to update preview for {path!r}: {e}")
-                return (
-                    gr.update(value=None, visible=False),
-                    gr.update(value=None, visible=False),
-                    "",
-                )
-        
-        # ----------------------------------------------------------------------
-        #  Bind the function – we feed both the current preview path and the
-        #  previously‑shown path, then store the new path back into both states.
-        # ----------------------------------------------------------------------
-        preview_path_state.change(
-            fn=apply_preview,
-            inputs=[preview_path_state, prev_preview_state],
-            outputs=[preview_image, preview_video, preview_status],
-        )
+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."
         
-        # After the preview is updated we also need to remember the new path.
-        # This tiny helper copies the new path into the “previous” state.
-        def _store_new_path(new_path: str):
-            return new_path, new_path   # (prev_state, preview_state)
+        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."
         
-        preview_path_state.change(
-            fn=_store_new_path,
-            inputs=[preview_path_state],
-            outputs=[prev_preview_state, preview_path_state],
-            queue=False,
-        )
+        img = Image.open(path)
+        img.verify()
+        return gr.update(value=path, visible=True), gr.update(value=None, visible=False), "Preview updated."
         
-        # ----------------------------------------------------------------------
-        demo.queue()
-        return demo
+    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
+
+if __name__ == "__main__":
+    create_demo().launch(share=False, server_name="0.0.0.0", server_port=7860, max_threads=8)