Update app.py

app.py

@@ -195,8 +195,8 @@ def create_overlay_thumbnail(rgb_frame, depth_frame):
 
 def create_rounded_corner_mask(width, height):
     """Erstellt Anti-Aliased Maske mit abgerundeter oberer linker Ecke"""
-    # Radius für die Rundung (etwa 8% der kleineren Dimension)
-    radius = min(width, height) // 12
+    # Radius für die Rundung (40% der kleineren Dimension)
+    radius = int(min(width, height) * 0.40)
     radius = max(radius, 5)  # Minimum 5 Pixel
     
     # Erstelle Maske (weiß = sichtbar, schwarz = transparent)
@@ -364,16 +364,68 @@ def validate_url(url):
     except:
         return False
 
+def download_video_with_ytdlp(url):
+    """Universal video download using yt-dlp for maximum compatibility"""
+    try:
+        import time
+        temp_path = f"temp_ytdlp_{int(time.time())}.%(ext)s"
+        
+        # yt-dlp command with best quality settings
+        cmd = [
+            "yt-dlp",
+            "--format", "best[ext=mp4]/best",  # Prefer MP4, fallback to best available
+            "--output", temp_path,
+            "--no-playlist",  # Only download single video
+            "--no-part",  # Don't create .part files
+            url
+        ]
+        
+        print(f"DEBUG: Downloading with yt-dlp: {url}")
+        result = subprocess.run(cmd, capture_output=True, text=True)
+        
+        if result.returncode != 0:
+            raise RuntimeError(f"yt-dlp failed: {result.stderr}")
+        
+        # Find the actual downloaded file (yt-dlp replaces %(ext)s)
+        import glob
+        temp_base = temp_path.replace(".%(ext)s", "")
+        downloaded_files = glob.glob(f"{temp_base}.*")
+        
+        if not downloaded_files:
+            raise RuntimeError("yt-dlp completed but no file found")
+        
+        actual_path = downloaded_files[0]
+        print(f"DEBUG: yt-dlp downloaded: {actual_path}")
+        return actual_path
+        
+    except Exception as e:
+        raise RuntimeError(f"Failed to download with yt-dlp: {e}")
+
 def detect_video_source(url):
-    """Detect video source (MidJourney, Civitai, Kling, or unknown)"""
+    """Detect video source and determine download method"""
+    # Direct video file URLs (priority check first)
+    if any(ext in url.lower() for ext in ['.mp4', '.webm', '.mov', '.avi', '.mkv']):
+        return "direct_video"
+    
+    # Known platforms with direct download support
     if "image.civitai.com" in url:
         return "civitai"
-    elif "v21-kling.klingai.com" in url or "kling.ai" in url:
-        return "kling"
     elif "cdn.midjourney.com" in url or "midjourney" in url.lower():
         return "midjourney"
+    elif "v21-kling.klingai.com" in url or "kling.ai" in url:
+        return "kling"
+    
+    # Popular video platforms (use yt-dlp)
+    elif any(platform in url.lower() for platform in [
+        'youtube.com', 'youtu.be', 'vimeo.com', 'dailymotion.com', 
+        'tiktok.com', 'instagram.com', 'twitter.com', 'x.com',
+        'facebook.com', 'reddit.com', 'twitch.tv'
+    ]):
+        return "ytdlp_platform"
+    
+    # Unknown URL - try yt-dlp first, fallback to direct
     else:
-        return "unknown"
+        return "ytdlp_fallback"
 
 def optimize_civitai_url(url):
     """Convert gallery Civitai URLs to original quality to avoid dimension issues"""
@@ -438,7 +490,7 @@ def download_civitai_video(civitai_url):
         raise RuntimeError(f"Failed to download Civitai video: {e}")
 
 def download_video_from_url(original_url):
-    """Universal video downloader for MJ, Civitai, and others"""
+    """Universal video downloader with yt-dlp integration"""
     try:
         if not validate_url(original_url):
             raise ValueError("Invalid URL format")
@@ -447,10 +499,23 @@ def download_video_from_url(original_url):
         source = detect_video_source(original_url)
         print(f"DEBUG: Detected video source: {source}")
         
-        if source == "civitai":
+        if source == "direct_video":
+            return download_generic_video(original_url)
+        elif source == "civitai":
             return download_civitai_video(original_url)
         elif source == "midjourney":
             return download_midjourney_video(original_url)
+        elif source == "kling":
+            return download_generic_video(original_url)  # Kling usually works with direct download
+        elif source == "ytdlp_platform":
+            return download_video_with_ytdlp(original_url)
+        elif source == "ytdlp_fallback":
+            # Try yt-dlp first, fallback to direct download
+            try:
+                return download_video_with_ytdlp(original_url)
+            except Exception as ytdlp_error:
+                print(f"DEBUG: yt-dlp failed, trying direct download: {ytdlp_error}")
+                return download_generic_video(original_url)
         else:
             return download_generic_video(original_url)
             
@@ -775,6 +840,22 @@ def on_video_url_change(url, use_blip):
                     
             elif source == "midjourney":
                 original_filename = "midjourney_video"
+            elif source == "kling":
+                original_filename = "kling_video"
+            elif source == "direct_video":
+                # Extract filename from direct video URL
+                parsed_url = urlparse(url)
+                url_filename = os.path.splitext(os.path.basename(parsed_url.path))[0]
+                cleaned = "".join(c for c in url_filename if c.isalnum() or c in "_-")[:20]
+                original_filename = cleaned if cleaned else "direct_video"
+            elif source in ["ytdlp_platform", "ytdlp_fallback"]:
+                # Extract domain name for yt-dlp downloads
+                parsed_url = urlparse(url)
+                domain = parsed_url.netloc.lower()
+                # Remove www. and common prefixes
+                domain = domain.replace('www.', '').replace('m.', '')
+                domain_name = domain.split('.')[0]  # Get main domain part
+                original_filename = f"{domain_name}_video"
             else:
                 original_filename = "downloaded_video"
                 
@@ -853,7 +934,7 @@ with gr.Blocks(analytics_enabled=False, title="Video Depth Anything") as demo:
     # 🎥 Video Depth Anything + RGBD Output
     
     Generate depth maps from videos and watch RGBD videos on holographic displays like Looking Glass Go. 
-    Upload a video or paste a video URL (Midjourney, Civitai, or Kling).
+    Upload a video or paste a video URL from **YouTube, TikTok, Instagram, MidJourney, Civitai**, or any platform.
     
     [🔗 Project Page](https://videodepthanything.github.io/) | [📖 Paper](https://arxiv.org/abs/2401.01884)
     """)
@@ -888,8 +969,8 @@ with gr.Blocks(analytics_enabled=False, title="Video Depth Anything") as demo:
     # Single row with all input controls and thumbnail preview
     with gr.Row():
         video_url = gr.Textbox(
-            label="Video URL (MJ, Civitai, or Kling)", 
-            placeholder="Paste MidJourney, Civitai, or Kling video URL here...",
+            label="Video URL (YouTube, TikTok, Instagram, Civitai, MidJourney, etc.)", 
+            placeholder="Paste video URL from YouTube, TikTok, Instagram, MidJourney, Civitai, or any platform...",
             scale=3
         )
         use_blip = gr.Checkbox(