# Core Deep Learning & Computation
# Note: torch>=1.13.0 is supported, but torch<2.0.0 recommended for DeepFloyd IF compatibility
torch>=1.13.0
torchvision
numpy<2.0  # PyTorch 2.1.0 requires numpy 1.x
# triton  # Required for PyTorch 2.x with CUDA, not needed for 1.x

# Note: nvidia-cuda-* packages are automatically installed as torch dependencies
# They are required for CUDA extension builds (setup.py, setup_fp16.py)
# No need to list them explicitly - torch will pull the correct versions (cu11/cu12)

# Image & Video Processing
Pillow
opencv-python
matplotlib

# SVG & Vector Graphics Processing (Core functionality)
fonttools
svgwrite
svgpathtools
scour
cairosvg  # Requires cairocffi, cffi (auto-installed)

# PSD Export (Required when export_psd=true in config)
psd-tools

# GPU Monitoring
nvidia-ml-py

# Progress Bars & Utilities
tqdm

# Optional: Image Quality Metrics (only if compute_psnr=true)
piq

# Optional: Experiment Tracking (currently not used)
# wandb

# Optional: Animation Scripts
moviepy  # For psd_to_mp4.py and animation scripts
pyvista  # For 3D animations (psd_3d_*.py scripts)

# Optional: Text Processing
pylrc  # For LRC file support

# Optional: CLIP Loss (advanced loss function)
git+https://github.com/openai/CLIP.git

# Optional: Data Analysis
pandas  # For compare_methods.py

# Optional: Advanced Image Processing
scikit-image
scipy
