# Requirements for Fine-Tune LLM Project
# Includes MedCAT, Neo4j, LLM Fine-tuning, and Clinical Summarization dependencies

# ============================================================================
# Core Python Libraries
# ============================================================================
pandas>=1.5.0
numpy>=1.21.0
scikit-learn>=1.1.0
matplotlib>=3.5.0
seaborn>=0.11.0
plotly>=5.0.0

# ============================================================================
# Machine Learning and Deep Learning
# ============================================================================
torch>=2.0.0
torchvision>=0.15.0
accelerate>=0.20.0
transformers>=4.30.0
datasets>=2.10.0
tokenizers>=0.13.0

# ============================================================================
# LLM Fine-tuning and Training
# ============================================================================
unsloth @ git+https://github.com/unslothai/unsloth.git
peft>=0.4.0
trl>=0.7.0
bitsandbytes>=0.40.0
xformers>=0.0.20
einops>=0.6.0
packaging>=21.0

# ============================================================================
# NLP and Text Processing
# ============================================================================
spacy>=3.4.0
nltk>=3.8.0
bert-score>=0.3.13
openai>=1.0.0

# ============================================================================
# Medical Entity Extraction and Knowledge Graphs
# ============================================================================
medcat>=1.14.0
rdflib>=6.0.0
rdflib-neo4j>=0.3.0
neo4j>=5.0.0

# ============================================================================
# Experiment Tracking and Logging
# ============================================================================
wandb>=0.15.0
python-dotenv>=0.19.0
huggingface-hub>=0.16.0

# ============================================================================
# Development and Jupyter Support
# ============================================================================
jupyter>=1.0.0
ipykernel>=6.0.0
notebook>=6.4.0

# ============================================================================
# Utilities and Miscellaneous
# ============================================================================
tqdm>=4.60.0
requests>=2.28.0
urllib3>=1.26.0
certifi>=2022.0.0

# Note: 'ast' is a built-in Python module, no need to install separately