# ===============================
# BackgroundFX Pro — Requirements
# Python 3.10 + CUDA 12.1 + Torch 2.5.1
# SAM2 + MatAnyone Integration with Audio Support
# ===============================

# --- PyTorch Index (CUDA 12.1) ---
--extra-index-url https://download.pytorch.org/whl/cu121

# --- PyTorch (Must be first) ---
torch==2.5.1
torchvision==0.20.1

# --- Core deps (before models) ---
numpy>=1.24.4
einops==0.6.0
Pillow==10.0.0
protobuf==3.20.3

# --- Image / Video Processing ---
opencv-python-headless==4.7.0.72
imageio==2.25.0
imageio-ffmpeg==0.4.7
moviepy==1.0.3
decord==0.6.0
scikit-image==0.19.3

# --- SAM2 Dependencies ---
hydra-core==1.3.2
omegaconf==2.3.0
timm==0.9.0
pyyaml==6.0.0
matplotlib==3.5.0
iopath==0.1.10

# --- SAM2 Installation ---
git+https://github.com/facebookresearch/segment-anything-2.git

# --- MatAnyone Dependencies (BEFORE MatAnyone) ---
huggingface-hub>=0.20.0
kornia==0.7.2
tqdm==4.66.1
scipy==1.10.1
transformers==4.30.0
accelerate==0.20.0
gradio>=3.34
xformers>=0.0.28

# --- MatAnyone Installation ---
git+https://github.com/pq-yang/MatAnyone.git

# --- Streamlit UI ---
streamlit==1.32.0

# --- Web stack ---
fastapi==0.104.0
uvicorn==0.24.0

# --- Pydantic / Typing ---
pydantic==2.6.0
pydantic-core==2.16.1
annotated-types==0.6.0
typing-extensions==4.8.0

# --- Utilities ---
ffmpeg-python==0.2.0
psutil==5.9.4
requests==2.31.0
scikit-learn==1.3.0
loguru==0.6.0
python-multipart==0.0.5
ninja==1.10.0
aiofiles==23.1.0