#!/bin/bash

# Script d'installation des dépendances pour RPA Vision V3

echo "=========================================================="
echo "  Installation des dépendances - RPA Vision V3"
echo "=========================================================="
echo ""

# Vérifier Python
echo "1. Vérification de Python..."
python3 --version
if [ $? -ne 0 ]; then
    echo "❌ Python 3 n'est pas installé"
    exit 1
fi
echo "✓ Python 3 détecté"
echo ""

# Créer un environnement virtuel (optionnel)
read -p "Créer un environnement virtuel ? (y/n) " -n 1 -r
echo
if [[ $REPLY =~ ^[Yy]$ ]]; then
    echo "2. Création de l'environnement virtuel..."
    python3 -m venv venv
    source venv/bin/activate
    echo "✓ Environnement virtuel créé et activé"
else
    echo "2. Installation dans l'environnement global..."
fi
echo ""

# Mettre à jour pip
echo "3. Mise à jour de pip..."
pip install --upgrade pip
echo ""

# Installer les dépendances
echo "4. Installation des dépendances depuis requirements.txt..."
pip install -r requirements.txt
echo ""

# Vérifier les installations critiques
echo "5. Vérification des installations..."

echo "   - PyTorch..."
python3 -c "import torch; print('     ✓ PyTorch version:', torch.__version__)"

echo "   - OpenCLIP..."
python3 -c "import open_clip; print('     ✓ OpenCLIP installé')"

echo "   - FAISS..."
python3 -c "import faiss; print('     ✓ FAISS installé')"

echo "   - OpenCV..."
python3 -c "import cv2; print('     ✓ OpenCV version:', cv2.__version__)"

echo "   - NumPy..."
python3 -c "import numpy; print('     ✓ NumPy version:', numpy.__version__)"

echo ""
echo "=========================================================="
echo "  ✅ Installation terminée avec succès !"
echo "=========================================================="
echo ""
echo "Prochaines étapes:"
echo "  1. Tester les embedders CLIP:"
echo "     python3 examples/test_clip_embedder.py"
echo ""
echo "  2. Tester le pipeline complet:"
echo "     python3 examples/test_embedding_pipeline.py"
echo ""
echo "  3. Lancer les tests unitaires:"
echo "     pytest tests/"
echo ""