requirements.txt

# ============================================================================
# BrainGemma3D Dashboard - Requirements
# ============================================================================
# Installa con: pip install -r requirements.txt

# --- Web Interface ---
gradio>=4.0.0

# --- Deep Learning Core ---
torch>=2.0.0
transformers>=4.35.0
peft>=0.7.0
accelerate>=0.24.0
safetensors>=0.4.0
bitsandbytes>=0.41.0              # For 4-bit quantization (BitsAndBytesConfig)

# --- Medical Imaging ---
nibabel>=5.0.0                    # NIfTI file handling
SimpleITK>=2.3.0                  # Advanced medical image processing (opzionale)

# --- Vision Models ---
timm>=0.9.0                       # Per SigLIP vision encoder
einops>=0.7.0                     # Tensor operations

# --- Visualization ---
matplotlib>=3.7.0
Pillow>=10.0.0
numpy>=1.24.0

# --- Interpretability (LIME) ---
lime>=0.2.0.1                     # LIME explainability (requires scikit-image)
scikit-image>=0.21.0              # For SLIC segmentation, morphology operations
scipy>=1.10.0                     # For ndimage operations (binary_closing, binary_opening, etc.)

# --- NLG Evaluation (Opzionale per valutazione) ---
rouge-score>=0.1.2
nltk>=3.8.1
bert-score>=0.3.13
# pycocoevalcap                   # Per CIDEr (richiede installazione manuale)

# --- Utilities ---
pandas>=2.0.0
tqdm>=4.65.0
scikit-learn>=1.3.0

# --- HuggingFace Hub ---
huggingface-hub>=0.19.0
sentencepiece>=0.1.99             # Per tokenizer

# --- Optional Performance ---
# flash-attn>=2.3.0               # CUDA Flash Attention (solo se GPU compatibile)
# xformers>=0.0.22                # Memory efficient attention