Update requirements.txt

requirements.txt

@@ -1,28 +1,28 @@
 # =============================================================================
 # Video Background Replacement - Enhanced Edition Requirements
-# Updated: 2025-09-10 - Fixed build errors and compatibility issues
+# Updated: 2025-09-11 - Fixed MatAnyone tensor compatibility issues
 # =============================================================================
 
 # =============================================================================
-# PyTorch ecosystem (CUDA 12.1) - Must be first to avoid conflicts
-# UPDATED: PyTorch 2.4+ for diffusers compatibility (custom_op support)
+# PyTorch ecosystem (CUDA 12.1) - PINNED VERSIONS for stability
 # =============================================================================
 --extra-index-url https://download.pytorch.org/whl/cu121
-torch>=2.4.0
-torchvision>=0.19.0
+torch==2.4.1
+torchvision==0.19.1
+torchaudio==2.4.1
 
 # =============================================================================
-# Core scientific computing - Foundation packages
+# Core scientific computing - UPDATED for PyTorch 2.4+ compatibility
 # =============================================================================
-numpy==1.24.3
+numpy>=1.25.0,<1.27.0
 scipy==1.11.1
 scikit-learn>=1.0.0
 py-cpuinfo
 
 # =============================================================================
-# Image and video processing
+# Image and video processing - FIXED OpenCV compatibility
 # =============================================================================
-opencv-python-headless==4.8.1.78
+opencv-python==4.8.1.78
 Pillow==10.1.0
 scikit-image==0.21.0
 matplotlib==3.7.2
@@ -36,24 +36,24 @@ imageio-ffmpeg==0.4.9
 av==10.0.0
 
 # =============================================================================
-# Core web interface - UPDATED for compatibility
+# Core web interface - PINNED for stability
 # =============================================================================
-gradio>=4.40.0,<5.0.0
+gradio==4.44.0
 
 # =============================================================================
-# System monitoring and memory management - CRITICAL DEPENDENCIES
+# System monitoring and memory management
 # =============================================================================
 psutil>=5.9.0
 GitPython>=3.1.40
 
 # =============================================================================
-# Enhanced SAM2 prompting dependencies - Add after PyTorch
+# Enhanced SAM2 prompting dependencies
 # =============================================================================
 ultralytics>=8.0.0
 mediapipe>=0.10.0
 
 # =============================================================================
-# SAM2 dependencies
+# SAM2 dependencies - COMPATIBLE VERSIONS
 # =============================================================================
 onnxruntime-gpu==1.16.0
 hydra-core>=1.3.2
@@ -62,12 +62,15 @@ iopath>=0.1.10
 submitit>=1.4.5
 
 # =============================================================================
-# MatAnyone dependencies - PyTorch-based packages
+# MatAnyone dependencies - FIXED COMPATIBILITY ISSUES
 # =============================================================================
-kornia>=0.7.0
+kornia==0.7.2
 einops>=0.7.0
 timm>=0.9.2
 safetensors>=0.4.2
+albumentations>=1.3.0
+torch-geometric>=2.3.0
+pytorch-lightning>=2.0.0
 
 # =============================================================================
 # Background removal
@@ -82,7 +85,7 @@ requests==2.31.0
 filelock==3.12.2
 
 # =============================================================================
-# HuggingFace ecosystem (for AI Backgrounds / Diffusers) - COMPATIBLE VERSIONS
+# HuggingFace ecosystem - COMPATIBLE VERSIONS
 # =============================================================================
 huggingface-hub>=0.19.3
 diffusers>=0.30.0,<0.32.0
@@ -91,80 +94,52 @@ accelerate>=0.33.0
 xformers>=0.0.22; sys_platform != "darwin"
 
 # =============================================================================
-# System monitoring (optional but recommended)
+# System monitoring and GPU utilities
 # =============================================================================
 nvidia-ml-py3==7.352.0
+pynvml>=11.5.0
+gpustat>=1.1.1
 
 # =============================================================================
-# Additional dependencies for enhanced features - FIXED VERSIONS
+# Enhanced features dependencies
 # =============================================================================
-# Memory management and process monitoring
 memory-profiler>=0.61.0
-
-# Enhanced logging
 colorlog>=6.7.0
-
-# JSON handling (enhanced)
 ujson>=5.8.0
-
-# Base64 utilities (enhanced)
 base64io>=1.0.3
-
-# UUID utilities (enhanced) 
 shortuuid>=1.0.11
-
-# Random number generation (enhanced)
 randomgen>=1.23.1
-
-# Progress bars and timing
 alive-progress>=3.1.4
-
-# HTTP utilities for OpenAI API
 httpx>=0.25.0
 aiohttp>=3.8.5
-
-# File format support
 h5py>=3.9.0
 hdf5storage>=0.1.19
-
-# Compression utilities
 lz4>=4.3.2
 zstandard>=0.21.0
-
-# Environment detection
 python-dotenv>=1.0.0
-
-# Enhanced error handling
 better-exceptions>=0.3.3
-
-# GPU monitoring utilities
-pynvml>=11.5.0
-gpustat>=1.1.1
-
-# Enhanced file operations
 send2trash>=1.8.2
 watchdog>=3.0.0
-
-# Configuration file formats
 toml>=0.10.2
-
-# Enhanced CLI utilities
 click>=8.1.7
 rich>=13.5.2
-
-# Data validation
 pydantic>=2.3.0
-
-# Type hints and annotations
 typing-extensions>=4.7.1
-
-# Enhanced containers and data structures
 more-itertools>=10.1.0
 toolz>=0.12.0
-
-# String processing
 regex>=2023.8.8
 chardet>=5.2.0
+netifaces>=0.11.0
 
-# Network utilities
-netifaces>=0.11.0
+# =============================================================================
+# CRITICAL ADDITIONS for MatAnyone tensor compatibility
+# =============================================================================
+# Image processing libraries MatAnyone expects
+imgaug>=0.4.0
+# OpenCV full version (not headless) for complete compatibility
+opencv-contrib-python==4.8.1.78
+# Additional tensor processing libraries
+tensorboard>=2.14.0
+# Ensure proper CUDA tensor operations
+nvidia-cublas-cu12>=12.1.0
+nvidia-cuda-runtime-cu12>=12.1.0