#!/bin/bash

# Script de configuration de l'environnement virtuel LLM Lab
# Ce script crée un nouvel environnement virtuel et installe les dépendances requises

echo "=== Configuration de l'environnement LLM Lab ==="

# Vérification si Python 3 est installé
if ! command -v python3 &> /dev/null; then
    echo "Erreur: Python 3 n'est pas installé. Veuillez l'installer avant de continuer."
    exit 1
fi

# Vérification si pip est installé
if ! command -v pip3 &> /dev/null; then
    echo "Erreur: pip3 n'est pas installé. Veuillez l'installer avant de continuer."
    exit 1
fi

# Vérification si venv est disponible
python3 -c "import venv" &> /dev/null
if [ $? -ne 0 ]; then
    echo "Le module venv n'est pas disponible. Installation en cours..."
    sudo apt-get update
    sudo apt-get install -y python3-venv
fi

# Vérification de l'installation de tkinter
python3 -c "import tkinter" &> /dev/null
if [ $? -ne 0 ]; then
    echo "Tkinter n'est pas installé. Installation en cours..."
    sudo apt-get update
    sudo apt-get install -y python3-tk
fi

# Suppression de l'ancien environnement s'il existe
if [ -d "llmlab" ]; then
    echo "Suppression de l'ancien environnement virtuel..."
    rm -rf llmlab
fi

# Création du nouvel environnement virtuel
echo "Création d'un nouvel environnement virtuel..."
python3 -m venv llmlab

# Activation de l'environnement virtuel
echo "Activation de l'environnement virtuel..."
source llmlab/bin/activate

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

# Installation des dépendances
echo "Installation des dépendances requises..."
pip install -r requirements.txt

# Installation optionnelle de dépendances pour NVIDIA GPU
if command -v nvidia-smi &> /dev/null; then
    echo "NVIDIA GPU détecté, installation des dépendances NVIDIA..."
    pip install nvidia-ml-py
fi

# Création de liens symboliques pour faciliter l'utilisation
echo "Création de liens symboliques pour les scripts..."
chmod +x chat.py
chmod +x monitor.py

# Affichage du résumé
echo ""
echo "=== Configuration terminée ==="
echo "Pour activer l'environnement virtuel, exécutez:"
echo "  source llmlab/bin/activate"
echo ""
echo "Pour lancer le moniteur système:"
echo "  ./monitor.py"
echo ""
echo "Pour utiliser un agent LLM:"
echo "  ./chat.py [nom de l'agent]"
echo ""
echo "Agents disponibles:"
echo "  cursor - CodeLlama 13B Python (pour le code)"
echo "  obsidian - Llama2 13B (pour la gestion des connaissances)"
echo "  test - Mistral 7B (pour les tests rapides)"
echo ""