# ╔═══════════════════════════════════════════════════════════════════════╗
# ║  MedSight AI — Python Dependencies                                  ║
# ║  Python 3.10+  |  PyTorch 2.2                                       ║
# ║  Install: pip install -r requirements.txt                            ║
# ╚═══════════════════════════════════════════════════════════════════════╝

# ── PyTorch (install FIRST — version must match torchvision) ──────────
# GPU (CUDA 11.8):
#   pip install torch==2.2.0+cu118 torchvision==0.17.0+cu118 --index-url https://download.pytorch.org/whl/cu118
# CPU-only:
#   pip install torch==2.2.0+cpu torchvision==0.17.0+cpu --index-url https://download.pytorch.org/whl/cpu

# ── [ML Core] ─────────────────────────────────────────────────────────
torchvision==0.17.0             # Pairs exactly with PyTorch 2.2
transformers==4.38.2            # Stable version for BioGPT and BioBERT
datasets==2.18.0                # Required for NCBI dataset parsing
accelerate==0.27.2              # Handles memory-efficient model loading
peft==0.9.0                     # Lightweight fine-tuning parameter handling
sentence-transformers==2.6.0    # Powers the all-MiniLM-L6-v2 embedding model
timm==0.9.16                    # Required backend for DINOv2 architecture
scikit-learn==1.4.1.post1       # Used for metric calculations (F1, precision, recall)
numpy==1.26.4                   # Last stable 1.x version before 2.0 breaking changes
Pillow==10.2.0                  # Safe version for PIL image processing
opencv-python-headless==4.9.0.80  # Headless avoids Qt DLL conflicts on servers
onnxruntime==1.17.1             # Fast CPU inference for ConvAE ONNX backend
scispacy==0.5.4                 # Lightweight medical NLP (NER pipeline)
# Install scispaCy model manually:
#   pip install https://s3-us-west-2.amazonaws.com/ai2-s3-scispacy/releases/v0.5.1/en_core_sci_sm-0.5.1.tar.gz

# ── [Backend — FastAPI] ───────────────────────────────────────────────
fastapi==0.110.0                # Base API framework
uvicorn[standard]==0.27.1       # ASGI server for FastAPI
python-multipart==0.0.9         # Required for FastAPI file uploads
pydantic==2.6.3                 # Pydantic v2 exact version for type safety
pydantic-settings==2.2.1        # Required for v2 configuration management
email-validator==2.1.1          # Required by Pydantic EmailStr schemas
slowapi==0.1.9                  # Rate limiting without Redis
python-jose[cryptography]==3.3.0  # JWT creation and verification
passlib[bcrypt]==1.7.4          # Password hashing algorithms
httpx==0.27.0                   # Async HTTP client for OAuth
aiofiles==23.2.1                # Async file handling for image uploads
authlib==1.3.0                  # Handles Google OAuth 2.0 flow
itsdangerous==2.1.2             # Required by Authlib sessions
python-magic-bin==0.4.14        # Windows-compatible MIME type checker (critical for file validation)

# ── [Database] ────────────────────────────────────────────────────────
sqlalchemy==2.0.28              # SQLAlchemy 2.0 async ORM
alembic==1.13.1                 # Database migration manager
aiosqlite==0.20.0               # Async SQLite driver for local development
asyncpg==0.29.0                 # Async PostgreSQL driver for production
greenlet==3.0.3                 # Required for SQLAlchemy async execution

# ── [RAG + Vector DB] ─────────────────────────────────────────────────
chromadb==0.4.24                # Local vector database for RAG retrieval

# ── [Generative AI] ───────────────────────────────────────────────────
google-generativeai>=0.8.0      # Gemini 2.0 Flash API for clinical Q&A chat
huggingface_hub>=0.20.0         # HuggingFace Hub for model downloads

# ── [Utilities] ───────────────────────────────────────────────────────
reportlab==4.1.0                # PDF report generation
fpdf2==2.7.9                    # Fallback lightweight PDF generation
python-dotenv==1.0.1            # Environment variable parsing
tqdm==4.66.2                    # Progress bars for dataset processing
ffmpeg-python==0.2.0            # Audio conversion wrapper for Whisper
apscheduler==3.10.4             # In-memory background task scheduling
boto3==1.34.69                  # Optional Cloudflare R2/S3-compatible upload storage

# ── [Audio — Optional] ────────────────────────────────────────────────
openai-whisper==20231117        # Whisper CPU transcription for voice input

# ── [Dev + Testing] ───────────────────────────────────────────────────
pytest==8.1.0                   # Core testing framework
pytest-asyncio==0.23.5          # Testing async FastAPI routes
black==24.3.0                   # Code formatting
ruff==0.3.4                     # Fast Python linter
mypy==1.9.0                     # Static type checking
locust==2.24.0                  # Load testing framework