#!/bin/bash

# Script d'installation rapide de FAISS pour RPA Vision V2

echo "=================================================="
echo "📦 Installation de FAISS"
echo "=================================================="
echo ""

# Vérifier que l'environnement virtuel existe
if [ ! -d "venv" ]; then
    echo "❌ Environnement virtuel non trouvé"
    echo "   Exécutez d'abord: ./setup.sh"
    exit 1
fi

# Activer l'environnement virtuel
source venv/bin/activate

# Vérifier si FAISS est déjà installé
if python -c "import faiss" 2>/dev/null; then
    echo "✅ FAISS est déjà installé"
    python -c "import faiss; print(f'   Version: {faiss.__version__}')"
    exit 0
fi

echo "📥 Installation de FAISS..."
echo ""

# Détecter si GPU est disponible
if python -c "import torch; print(torch.cuda.is_available())" 2>/dev/null | grep -q "True"; then
    echo "🎮 GPU détecté - Installation de faiss-gpu"
    pip install faiss-gpu
else
    echo "💻 Pas de GPU - Installation de faiss-cpu"
    pip install faiss-cpu
fi

# Vérifier l'installation
if python -c "import faiss" 2>/dev/null; then
    echo ""
    echo "✅ FAISS installé avec succès!"
    python -c "import faiss; print(f'   Version: {faiss.__version__}')"
    
    echo ""
    echo "📊 Prochaine étape: Reconstruire l'index"
    echo "   cd .."
    echo "   python3 rebuild_faiss_simple.py"
else
    echo ""
    echo "❌ Erreur lors de l'installation de FAISS"
    exit 1
fi