# Script PowerShell pour lancer le serveur API LLM Lab
Write-Host "=== Lancement du serveur API LLM Lab pour Cursor et Obsidian ===" -ForegroundColor Green

# Vérifier si l'environnement virtuel existe
if (-not (Test-Path -Path ".\llmlab")) {
    Write-Host "L'environnement virtuel n'existe pas. Veuillez exécuter setup_env.bat pour le créer." -ForegroundColor Red
    exit 1
}

# Activer l'environnement virtuel
Write-Host "Activation de l'environnement virtuel..." -ForegroundColor Cyan
try {
    & .\llmlab\Scripts\Activate.ps1
} catch {
    Write-Host "Erreur lors de l'activation de l'environnement virtuel: $_" -ForegroundColor Red
    Write-Host "Tentative alternative d'activation..." -ForegroundColor Yellow
    & cmd /c ".\llmlab\Scripts\activate.bat && powershell -NoExit"
    exit 1
}

# Vérifier que Flask et Flask-CORS sont installés
Write-Host "Vérification des dépendances..." -ForegroundColor Cyan
$flaskInstalled = $false
$flaskCorsInstalled = $false

try {
    $modules = pip list
    $flaskInstalled = $modules -match "flask" -and $modules -notmatch "flask-cors"
    $flaskCorsInstalled = $modules -match "flask-cors"
} catch {
    Write-Host "Erreur lors de la vérification des modules: $_" -ForegroundColor Red
}

# Installer les dépendances manquantes
if (-not $flaskInstalled) {
    Write-Host "Installation de Flask..." -ForegroundColor Yellow
    pip install flask
}

if (-not $flaskCorsInstalled) {
    Write-Host "Installation de Flask-CORS..." -ForegroundColor Yellow
    pip install flask-cors
}

# Vérifier si Ollama est en cours d'exécution
Write-Host "Vérification qu'Ollama est en cours d'exécution..." -ForegroundColor Cyan
try {
    $ollamaResponse = Invoke-WebRequest -Uri "http://localhost:11434/api/tags" -UseBasicParsing -ErrorAction SilentlyContinue
    if ($ollamaResponse.StatusCode -eq 200) {
        Write-Host "Ollama est en cours d'exécution." -ForegroundColor Green
    } else {
        Write-Host "Ollama semble ne pas fonctionner correctement." -ForegroundColor Yellow
    }
} catch {
    Write-Host "Impossible de se connecter à Ollama. Assurez-vous qu'il est en cours d'exécution." -ForegroundColor Red
    Write-Host "Vous pouvez le télécharger depuis https://ollama.com/download/windows" -ForegroundColor Yellow
}

# Lancer le serveur API
Write-Host "`nLancement du serveur API..." -ForegroundColor Green
Write-Host "Utilisez Ctrl+C pour arrêter le serveur`n" -ForegroundColor Yellow

# Exécution du serveur
python api_server.py

# Ce code ne sera exécuté qu'après l'arrêt du serveur
Write-Host "`nServeur API arrêté." -ForegroundColor Cyan
Write-Host "Désactivation de l'environnement virtuel..." -ForegroundColor Cyan
deactivate