requirements.txt

##########################################################
# TORCH + TORCH GEOMETRIC (CPU BUILD)
##########################################################

# Install torch (CPU version)
-f https://download.pytorch.org/whl/cpu
torch==2.3.1
torchvision==0.18.1
torchaudio==2.3.1

# Install PyTorch Geometric dependencies (CPU wheels)
-f https://data.pyg.org/whl/torch-2.3.1+cpu.html
torch-scatter==2.1.2
torch-sparse==0.6.18
torch-cluster==1.6.3
torch-spline-conv==1.2.2

# Main PyTorch Geometric library
torch-geometric==2.6.0

##########################################################
# RESET TO DEFAULT PYPI INDEX FOR REMAINING PACKAGES
##########################################################

--index-url https://pypi.org/simple

# PyTorch ecosystem
pytorch-lightning>=2.0.0
torchmetrics>=0.11.0

# Data processing
numpy>=1.21.0
pandas>=1.5.0

# Music processing
partitura>=1.4.0
music21>=8.0.0

# Gradio and Hugging Face Spaces
gradio>=4.0.0

# Experiment tracking
wandb>=0.13.0

# PDF rendering and image processing
reportlab>=4.0.0
pdf2image>=1.16.0
Pillow>=9.0.0

# ML utilities
scikit-learn>=1.1.0
tqdm>=4.64.0

# Graph music representation library (required by AnalysisGNN)
git+https://github.com/manoskary/graphmuse.git

# Install AnalysisGNN
git+https://github.com/manoskary/analysisgnn.git