Update app.py

app.py

@@ -16,6 +16,7 @@ import cv2
 import timeout_decorator
 import spaces
 import tempfile
+import shutil
 
 from registry import get_model
 from core.describe_scene import describe_scene
@@ -141,25 +142,23 @@ def handle(mode, media_upload, url, run_det, det_model, det_confidence, run_seg,
 def show_preview_from_upload(files):
     if not files:
         return gr.update(visible=False), gr.update(visible=False)
-
+    
     file = files[0]
     filename = file.name.lower()
 
-    # Handle image
     if filename.endswith((".png", ".jpg", ".jpeg", ".webp")):
         img = Image.open(file).convert("RGB")
         return gr.update(value=img, visible=True), gr.update(visible=False)
 
-    # Handle video
     elif filename.endswith((".mp4", ".mov", ".avi")):
-        tmp_dir = tempfile.mkdtemp()
+        # Copy uploaded video to a known temp location
+        temp_dir = tempfile.mkdtemp()
         ext = os.path.splitext(filename)[-1]
-        save_path = os.path.join(tmp_dir, f"preview_video{ext}")
-        with open(save_path, "wb") as f:
+        safe_path = os.path.join(temp_dir, f"uploaded_video{ext}")
+        with open(safe_path, "wb") as f:
             f.write(file.read())
 
-        # ✅ Gradio needs a proper path for video preview
-        return gr.update(visible=False), gr.update(value=save_path, visible=True)
+        return gr.update(visible=False), gr.update(value=safe_path, visible=True)
 
     return gr.update(visible=False), gr.update(visible=False)
 
@@ -261,7 +260,7 @@ with gr.Blocks() as demo:
             # video_preview = gr.Video(label="Preview (Video)", visible=False)        
             # Only one is shown at a time — image or video
             img_out = gr.Image(label="Preview / Processed Output", visible=False)
-            vid_out = gr.Video(label="Preview / Processed Output", visible=False)
+            vid_out = gr.Video(label="Preview / Processed Video", visible=False, streaming=True, autoplay=True)
             json_out = gr.JSON(label="Scene JSON")
             zip_out = gr.File(label="Download Results")