Update app.py

app.py

@@ -790,29 +790,29 @@ 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):
-    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."
+    def apply_preview(path: str, last_path: str):
+        if path == last_path:
+            return gr.update(), gr.update(), ""
         
-        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."
+        if not path:
+            return gr.update(), gr.update(), ""
         
-        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])
+        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."
+            
+        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