Update app.py

app.py

@@ -15,6 +15,7 @@ from PIL import Image
 import cv2
 import timeout_decorator
 import spaces
+import tempfile
 
 from registry import get_model
 from core.describe_scene import describe_scene
@@ -140,17 +141,25 @@ 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]
-    path = file.name.lower()
-    if path.endswith((".png", ".jpg", ".jpeg", ".webp")):
+    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)
-    elif path.endswith((".mp4", ".mov", ".avi")):
-        temp_video_dir = tempfile.mkdtemp()
-        safe_path = os.path.join(temp_video_dir, os.path.basename(path))
-        with open(safe_path, "wb") as f:
+
+    # Handle video
+    elif filename.endswith((".mp4", ".mov", ".avi")):
+        tmp_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:
             f.write(file.read())
-        return gr.update(visible=False), gr.update(value=safe_path, visible=True)
+
+        # ✅ 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(visible=False)