#!/bin/bash

# Script pour exécuter les commandes dans l'environnement virtuel LLM Lab

# Vérification si l'environnement virtuel existe
if [ ! -d "llmlab" ]; then
    echo "L'environnement virtuel n'existe pas. Veuillez exécuter ./setup_env.sh pour le créer."
    exit 1
fi

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

# Fonction d'aide
show_help() {
    echo "Usage: ./run.sh [commande] [arguments...]"
    echo ""
    echo "Commandes disponibles:"
    echo "  chat [agent]        Lance le chat en ligne de commande avec l'agent spécifié"
    echo "  gui                 Lance l'interface graphique de chat"
    echo "  monitor             Lance le moniteur système"
    echo "  list                Liste les agents disponibles"
    echo "  test                Exécute les tests unitaires"
    echo "  help                Affiche cette aide"
    echo ""
    echo "Exemples:"
    echo "  ./run.sh chat cursor         Lance le chat en ligne de commande avec l'agent Cursor (CodeLlama)"
    echo "  ./run.sh gui                 Lance l'interface graphique de chat"
    echo "  ./run.sh monitor             Lance le moniteur système"
    echo "  ./run.sh list                Liste tous les agents disponibles"
}

# Vérification des arguments
if [ $# -eq 0 ]; then
    show_help
    exit 0
fi

# Traitement des commandes
case "$1" in
    chat)
        if [ $# -eq 1 ]; then
            python chat.py --list
        else
            python chat.py "$2"
        fi
        ;;
    gui)
        python chat_gui.py
        ;;
    monitor)
        python monitor.py
        ;;
    list)
        python chat.py --list
        ;;
    test)
        python -m unittest discover tests
        ;;
    help)
        show_help
        ;;
    *)
        echo "Commande inconnue: $1"
        show_help
        exit 1
        ;;
esac

# Désactivation de l'environnement virtuel
deactivate