Update app.py

app.py

@@ -215,57 +215,134 @@ def infer_video_depth_from_source(upload_video, video_url, filename, use_blip, *
 # --- UI event handlers ---
 def on_video_url_change(url, use_blip):
     """Handle URL input change with safe error handling"""
+    global current_video_file, current_video_url, blip_generated_name, original_filename
+    
     if not url or url.strip() == "":
+        current_video_file = None
+        current_video_url = None
+        blip_generated_name = ""
+        original_filename = ""
         return None, "", "Enter a valid video URL"
     
     try:
         print(f"Downloading video from URL: {url}")
         video_path = download_video_from_url(url)
-        name = ""
         
+        # Store the current video info
+        current_video_file = None  # Clear file when using URL
+        current_video_url = video_path
+        original_filename = "downloaded_video"
+        blip_generated_name = ""
+        
+        # Generate BLIP name if requested
         if use_blip and video_path:
             try:
                 frames, _ = read_video_frames(video_path, 999, -1, 480)
                 if len(frames) > 0:
                     frame = frames[len(frames) // 2]
-                    name = generate_blip_name(frame)
+                    blip_generated_name = generate_blip_name(frame)
             except Exception as e:
                 print(f"BLIP naming failed: {e}")
-                name = "downloaded_video"
+                blip_generated_name = ""
+        
+        # Return appropriate name
+        final_name = blip_generated_name if (use_blip and blip_generated_name) else original_filename
+        return video_path, final_name, "Video downloaded successfully!"
         
-        return video_path, name, "Video downloaded successfully!"
     except Exception as e:
         error_msg = f"Download failed: {str(e)}"
         print(error_msg)
         return None, "", error_msg
 
+# Global variables to store video info for toggling
+current_video_file = None
+current_video_url = None
+blip_generated_name = ""
+original_filename = ""
+
 def on_video_upload_change(video_file, use_blip):
-    """Handle video upload with BLIP naming"""
+    """Handle video upload and store video info for toggling"""
+    global current_video_file, blip_generated_name, original_filename, current_video_url
+    
     if not video_file:
+        current_video_file = None
+        blip_generated_name = ""
+        original_filename = ""
         return "", "Upload a video file"
     
     try:
-        name = ""
+        # Store the current video
+        current_video_file = video_file
+        current_video_url = None  # Clear URL when uploading file
+        
+        # Generate original filename
+        if hasattr(video_file, 'name') and video_file.name:
+            original_name = os.path.splitext(os.path.basename(video_file.name))[0]
+            original_filename = "".join(c for c in original_name if c.isalnum() or c in "_-")[:30]
+        else:
+            original_filename = "uploaded_video"
+        
+        # Generate BLIP name
+        blip_generated_name = ""
         if use_blip:
             frames, _ = read_video_frames(video_file, 999, -1, 480)
             if len(frames) > 0:
                 frame = frames[len(frames) // 2]
-                name = generate_blip_name(frame)
+                blip_generated_name = generate_blip_name(frame)
+        
+        # Return appropriate name based on BLIP setting
+        final_name = blip_generated_name if (use_blip and blip_generated_name) else original_filename
+        return final_name, "Video uploaded successfully!"
         
-        return name or "uploaded_video", "Video uploaded successfully!"
     except Exception as e:
         error_msg = f"Upload processing failed: {str(e)}"
         print(error_msg)
         return "uploaded_video", error_msg
 
+def on_blip_toggle(use_blip):
+    """Handle BLIP checkbox toggle - switch between BLIP and original name"""
+    global current_video_file, current_video_url, blip_generated_name, original_filename
+    
+    # Only react if we have a video loaded
+    if current_video_file is None and current_video_url is None:
+        return "", "No video loaded"
+    
+    try:
+        # If toggling BLIP on and we don't have a BLIP name yet, generate it
+        if use_blip and not blip_generated_name:
+            if current_video_file:
+                frames, _ = read_video_frames(current_video_file, 999, -1, 480)
+                if len(frames) > 0:
+                    frame = frames[len(frames) // 2]
+                    blip_generated_name = generate_blip_name(frame)
+            elif current_video_url:
+                # For URL videos, we might need to re-read frames
+                frames, _ = read_video_frames(current_video_url, 999, -1, 480)
+                if len(frames) > 0:
+                    frame = frames[len(frames) // 2]
+                    blip_generated_name = generate_blip_name(frame)
+        
+        # Return appropriate name based on toggle
+        if use_blip and blip_generated_name:
+            return blip_generated_name, "Using BLIP generated name"
+        else:
+            name = original_filename if original_filename else "video"
+            return name, "Using original filename"
+            
+    except Exception as e:
+        error_msg = f"Name generation failed: {str(e)}"
+        print(error_msg)
+        fallback = original_filename if original_filename else "video"
+        return fallback, error_msg
+
 # --- Gradio Interface ---
 with gr.Blocks(analytics_enabled=False, title="Video Depth Anything") as demo:
     gr.Markdown("""
     # 🎥 Video Depth Anything + RGBD Output
     
-    Transform your videos into depth maps! Upload a video or paste a URL to generate:
+    Transform your videos into depth maps! Upload a video or paste a URL (Midjourney video for now) to generate:
     - **Depth visualization** (colorized depth map)
-    - **RGBD side-by-side** (original + depth combined)
+    - **RGBD side-by-side** (original + depth combined) for viewing on holographic displays like Looking Glass Go.
     
     [🔗 Project Page](https://videodepthanything.github.io/) | [📖 Paper](https://arxiv.org/abs/2401.01884)
     """)
@@ -299,7 +376,7 @@ with gr.Blocks(analytics_enabled=False, title="Video Depth Anything") as demo:
 
     with gr.Row():
         video_url = gr.Textbox(
-            label="🔗 Video URL", 
+            label="🔗 Midjourney Video URL (Experimental)", 
             placeholder="Paste video URL here...",
             scale=3
         )
@@ -328,6 +405,13 @@ with gr.Blocks(analytics_enabled=False, title="Video Depth Anything") as demo:
         outputs=[filename, status_display]
     )
 
+    # New: Toggle BLIP checkbox to switch between names
+    use_blip.change(
+        fn=on_blip_toggle,
+        inputs=[use_blip],
+        outputs=[filename, status_display]
+    )
+
     with gr.Accordion("⚙️ Advanced Settings", open=False):
         with gr.Row():
             max_len = gr.Slider(