#!/bin/bash
# Installation rapide des dépendances RPA Vision V3

set -e

echo "🔧 Installation des dépendances RPA Vision V3..."
echo ""

# Vérifier si venv_v3 existe
if [ ! -d "venv_v3" ]; then
    echo "📦 Création de l'environnement virtuel..."
    python3 -m venv venv_v3
    echo "✓ Environnement virtuel créé"
fi

# Activer venv
source venv_v3/bin/activate

echo "📦 Mise à jour de pip..."
venv_v3/bin/python3 -m pip install --upgrade pip -q

echo "📦 Installation des dépendances principales..."
venv_v3/bin/python3 -m pip install numpy>=1.24.0 Pillow>=10.0.0 scikit-learn>=1.3.0 requests>=2.31.0

echo "📦 Installation de PyTorch (CPU)..."
venv_v3/bin/python3 -m pip install torch>=2.0.0 torchvision>=0.15.0 --index-url https://download.pytorch.org/whl/cpu

echo "📦 Installation de FAISS..."
venv_v3/bin/python3 -m pip install faiss-cpu>=1.7.4

echo "📦 Installation de OpenCLIP..."
venv_v3/bin/python3 -m pip install open_clip_torch>=2.20.0 transformers>=4.30.0

echo "📦 Installation des outils de capture d'écran..."
venv_v3/bin/python3 -m pip install mss>=9.0.0 pyautogui>=0.9.54 pygetwindow>=0.0.9 opencv-python>=4.8.0

echo "📦 Installation de PyQt5 pour l'interface..."
venv_v3/bin/python3 -m pip install PyQt5>=5.15.0

echo "📦 Installation des outils de test..."
venv_v3/bin/python3 -m pip install pytest>=7.0.0 pytest-cov>=4.0.0

echo "📦 Installation de Flask pour le dashboard..."
venv_v3/bin/python3 -m pip install Flask==3.0.0

echo ""
echo "✅ Installation terminée !"
echo ""
echo "Pour utiliser l'environnement :"
echo "  source venv_v3/bin/activate"
echo ""
echo "Pour lancer l'application :"
echo "  ./run.sh"
echo ""

# Marquer comme installé
touch .deps_installed

deactivate