#!/bin/bash
cd "$(dirname "$0")"

# Check for venv, create if needed
if [ -d "venv" ]; then
    source venv/bin/activate
else
    echo "No venv found. Creating one..."
    python3 -m venv venv
    source venv/bin/activate
    pip install -r requirements.txt
fi

echo "============================================================"
echo "  Spam XAI Project — Retrain Random Forest classifier"
echo "  Features: 3000 TF-IDF + 11 metadata features"
echo "  Data:     SpamAssassin + Enron + newsletters"
echo "  Time:     ~5-10 minutes"
echo "  Saves:    models/ (random_forest_spam.joblib, etc.)"
echo "============================================================"
echo ""
python3 retrain.py
echo ""
echo "Done! Models saved to models/"
echo ""
echo "Press any key to close..."
read -n 1