# EmpathRAG - Python Dependencies
# Install order matters - read INSTALL NOTES below before running pip install

# ── INSTALL NOTES ──────────────────────────────────────────────────────────────
# 1. PyTorch (CUDA) - install FIRST before anything else:
#    pip install torch==2.5.1+cu121 --index-url https://download.pytorch.org/whl/cu121
#
# 2. llama-cpp-python (CUDA wheel) - install SECOND:
#    pip install llama_cpp_python==0.3.4 --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu121
#
# 3. Everything else:
#    pip install -r requirements.txt
#
# 4. Force correct numpy (faiss requires <2.0):
#    pip install "numpy==1.26.4" --force-reinstall
# ──────────────────────────────────────────────────────────────────────────────

# HuggingFace stack
transformers==4.57.6
peft==0.18.1
datasets==2.18.0
accelerate==1.13.0
evaluate==0.4.1

# Retrieval
faiss-cpu==1.9.0
sentence-transformers==3.4.1

# Explainability
captum==0.7.0

# Evaluation
ragas==0.1.21
bert-score==0.3.13
rank-bm25==0.2.2
scipy==1.17.1
scikit-learn==1.8.0

# Demo
gradio==4.44.1
fastapi==0.115.5
starlette==0.41.3
pydantic==2.10.6

# Data / Utilities
pandas==2.3.3
numpy==1.26.4
tqdm==4.67.3
kaggle==1.6.3