Upload scripts\add_vr180_metadata.py with huggingface_hub

scripts//add_vr180_metadata.py

@@ -0,0 +1,151 @@
+#!/usr/bin/env python3
+"""
+VR180 Metadata Injector
+Adds proper VR180 metadata to side-by-side stereoscopic videos for VR headset compatibility.
+"""
+
+import os
+import sys
+import subprocess
+import json
+import tempfile
+from pathlib import Path
+
+def check_spatial_media_installer():
+    """Check if Google's Spatial Media Metadata Injector is installed."""
+    try:
+        # Try to find the spatial media installer
+        result = subprocess.run(['spatial-media', '--help'], 
+                              capture_output=True, text=True, timeout=10)
+        return True
+    except (subprocess.TimeoutExpired, FileNotFoundError, subprocess.CalledProcessError):
+        return False
+
+def install_spatial_media():
+    """Install Google's Spatial Media Metadata Injector."""
+    print("Installing Google's Spatial Media Metadata Injector...")
+    try:
+        # Install via pip
+        subprocess.run([sys.executable, '-m', 'pip', 'install', 'spatial-media'], 
+                      check=True)
+        print("✅ Spatial Media Metadata Injector installed successfully!")
+        return True
+    except subprocess.CalledProcessError as e:
+        print(f"❌ Failed to install spatial-media: {e}")
+        return False
+
+def create_metadata_json():
+    """Create VR180 metadata JSON configuration."""
+    metadata = {
+        "spatial_audio": False,
+        "stereo_mode": "left-right",
+        "projection": "180"
+    }
+    
+    # Create temporary JSON file
+    temp_file = tempfile.NamedTemporaryFile(mode='w', suffix='.json', delete=False)
+    json.dump(metadata, temp_file, indent=2)
+    temp_file.close()
+    return temp_file.name
+
+def inject_vr180_metadata(input_file, output_file):
+    """Inject VR180 metadata into the video file."""
+    print(f"Processing: {input_file}")
+    print(f"Output: {output_file}")
+    
+    # Create metadata JSON
+    metadata_file = create_metadata_json()
+    
+    try:
+        # Use spatial-media to inject metadata
+        cmd = [
+            'spatial-media',
+            'inject',
+            '--input', input_file,
+            '--output', output_file,
+            '--metadata', metadata_file
+        ]
+        
+        print("Injecting VR180 metadata...")
+        result = subprocess.run(cmd, capture_output=True, text=True, timeout=300)
+        
+        if result.returncode == 0:
+            print("✅ VR180 metadata injected successfully!")
+            print(f"✅ Output saved as: {output_file}")
+            return True
+        else:
+            print(f"❌ Error injecting metadata: {result.stderr}")
+            return False
+            
+    except subprocess.TimeoutExpired:
+        print("❌ Process timed out. Video might be too large.")
+        return False
+    except Exception as e:
+        print(f"❌ Error: {e}")
+        return False
+    finally:
+        # Clean up temporary metadata file
+        try:
+            os.unlink(metadata_file)
+        except:
+            pass
+
+def validate_input_file(input_file):
+    """Validate that the input file exists and is a video."""
+    if not os.path.exists(input_file):
+        print(f"❌ Input file not found: {input_file}")
+        return False
+    
+    # Check if it's a video file
+    video_extensions = ['.mp4', '.avi', '.mov', '.mkv', '.webm']
+    if not any(input_file.lower().endswith(ext) for ext in video_extensions):
+        print(f"❌ Input file doesn't appear to be a video: {input_file}")
+        return False
+    
+    return True
+
+def main():
+    """Main function."""
+    if len(sys.argv) != 3:
+        print("Usage: python add_vr180_metadata.py <input_video> <output_vr180_video>")
+        print("\nExample:")
+        print("python add_vr180_metadata.py input.mp4 output_vr180.mp4")
+        sys.exit(1)
+    
+    input_file = sys.argv[1]
+    output_file = sys.argv[2]
+    
+    print("🥽 VR180 Metadata Injector")
+    print("=" * 40)
+    
+    # Validate input file
+    if not validate_input_file(input_file):
+        sys.exit(1)
+    
+    # Check if spatial media installer is available
+    if not check_spatial_media_installer():
+        print("📦 Spatial Media Metadata Injector not found.")
+        if not install_spatial_media():
+            print("❌ Cannot proceed without Spatial Media Metadata Injector.")
+            sys.exit(1)
+    
+    # Create output directory if it doesn't exist
+    output_dir = os.path.dirname(output_file)
+    if output_dir and not os.path.exists(output_dir):
+        os.makedirs(output_dir)
+    
+    # Inject VR180 metadata
+    if inject_vr180_metadata(input_file, output_file):
+        print("\n🎉 Success! Your video is now VR180-compliant!")
+        print(f"📱 You can now play '{output_file}' in VR headsets like:")
+        print("   • Oculus Quest/Quest 2/Quest Pro")
+        print("   • HTC Vive")
+        print("   • PlayStation VR")
+        print("   • Windows Mixed Reality")
+        print("\n💡 Make sure to select 'Side-by-Side' or 'SBS' mode in your VR player.")
+    else:
+        print("\n❌ Failed to inject VR180 metadata.")
+        sys.exit(1)
+
+if __name__ == "__main__":
+    main()