Update requirements.txt

requirements.txt

@@ -1,55 +1,84 @@
-# Core Dependencies (pinned for stability)
-numpy>=1.24.3,<1.25
-Pillow==10.0.1
-tqdm==4.66.1
+# ============================================================================
+# CORE PYTHON DEPENDENCIES
+# ============================================================================
+numpy>=1.24.3,<1.26
+Pillow>=10.0.1,<11.0
+setuptools>=65.7.0,<69.0
+wheel>=0.40.0,<1.0
+typing-extensions>=4.12.2,<5.0
 
-# Web & UI
+# ============================================================================
+# WEB FRAMEWORK & UI
+# ============================================================================
 gradio==4.44.1
-gradio_client==1.3.0  # Added to match gradio version for schema patch
+gradio_client==1.3.0
 
-# Core AI and ML (compatible with Hugging Face Spaces CUDA 12.3)
-torch>=2.0.1
-torchvision>=0.15.2
-torchaudio>=0.0.2
-transformers==4.30.2
-huggingface_hub>=0.19.3
-accelerate==0.20.3
+# ============================================================================
+# DEEP LEARNING & AI MODELS
+# ============================================================================
+# PyTorch ecosystem (CUDA 12.1 compatible)
+torch>=2.0.1,<2.4
+torchvision>=0.15.2,<0.19
+torchaudio>=2.0.1,<2.4
 
-# Image and Video Processing
+# Hugging Face ecosystem
+transformers>=4.30.2,<5.0
+huggingface_hub>=0.19.3,<1.0
+accelerate>=0.20.3,<1.0
+safetensors>=0.3.3,<1.0
+
+# Model utilities
+einops>=0.7.0,<1.0
+timm>=0.9.5,<1.0
+
+# ============================================================================
+# COMPUTER VISION & VIDEO PROCESSING
+# ============================================================================
 opencv-python-headless==4.8.0.76
-moviepy==1.0.3
-imageio-ffmpeg==0.4.8
-ffmpeg-python==0.2.0
-imageio==2.25.0  # Added for MatAnyone compatibility
-
-# Model Dependencies
-einops==0.7.0
-safetensors==0.3.3
-timm==0.9.5
-av>=0.5.2  # Added for MatAnyone
-cchardet>=2.1.7  # Added for MatAnyone
-cython  # Added for MatAnyone
-easydict  # Added for MatAnyone
-gdown>=4.7.1  # Added for MatAnyone
-gitpython>=3.1  # Added for MatAnyone
-hickle>=5.0  # Added for MatAnyone
-netifaces>=0.11.0  # Added for MatAnyone
-pycocotools>=2.0.7  # Added for MatAnyone
-tensorboard>=2.11  # Added for MatAnyone
-git+https://github.com/cheind/py-thin-plate-spline@f6995795397118b7d0ac01aecd3f39ffbfad9dee  # Added for MatAnyone
-
-# Configuration
-hydra-core==1.3.2
-omegaconf==2.3.0
-
-# Utilities
-diskcache==5.6.3
-scipy==1.10.1
-typing-extensions>=4.12.2,<5
-setuptools==65.7.0
-wheel==0.40.0
-psutil>=5.9.0
-
-# Git Dependencies (pinned to specific commits for stability)
+
+# Video processing
+moviepy>=1.0.3,<2.0
+imageio>=2.25.0,<3.0
+imageio-ffmpeg>=0.4.8,<1.0
+ffmpeg-python>=0.2.0,<1.0
+av>=0.5.2,<1.0
+
+# ============================================================================
+# SCIENTIFIC COMPUTING
+# ============================================================================
+scipy>=1.10.1,<2.0
+tqdm>=4.66.1,<5.0
+
+# ============================================================================
+# CONFIGURATION & UTILITIES
+# ============================================================================
+hydra-core>=1.3.2,<2.0
+omegaconf>=2.3.0,<3.0
+diskcache>=5.6.3,<6.0
+psutil>=5.9.0,<6.0
+
+# ============================================================================
+# MATANYONE DEPENDENCIES
+# ============================================================================
+# Core dependencies for MatAnyone
+cchardet>=2.1.7
+cython
+easydict
+gdown>=4.7.1
+gitpython>=3.1
+hickle>=5.0
+netifaces>=0.11.0
+pycocotools>=2.0.7
+tensorboard>=2.11
+
+# ============================================================================
+# GIT DEPENDENCIES (Model Repositories)
+# ============================================================================
+# SAM2 - Segment Anything 2
 git+https://github.com/facebookresearch/segment-anything-2.git@2b90b9f5ceec907a1c18123530e92e794ad901a4
-git+https://github.com/pq-yang/MatAnyone.git@2234ce5cdc487749515518bd035b5e18bccea3da
+
+# MatAnyone - High-quality matting
+git+https://github.com/pq-yang/MatAnyone.git@2234ce5cdc487749515518bd035b5e18bccea3da
+
+# Thin Plate Spline (MatAnyone dependency)
+git+https://github.com/cheind/py-thin-plate-spline@f6995795397118b7d0ac01aecd3f39ffbfad9dee