Upload INSTALLER.ps1 with huggingface_hub

INSTALLER.ps1

@@ -0,0 +1,248 @@
+# Portrait Generator V4 - INSTALLER.ps1
+# Installateur principal. Normalement lance via INSTALLER.bat.
+
+$ErrorActionPreference = "Continue"
+$Root = Split-Path -Parent $MyInvocation.MyCommand.Path
+Set-Location $Root
+$Log = Join-Path $Root "install_log.txt"
+"Portrait Generator V4 - Installation $(Get-Date)" | Out-File $Log -Encoding UTF8
+
+function Log($msg) {
+    $line = "[$(Get-Date -Format 'HH:mm:ss')] $msg"
+    Write-Host $line
+    Add-Content -Path $Log -Value $line -Encoding UTF8
+}
+
+function Ask-Choice($Title, $Choices, $Default) {
+    Write-Host ""
+    Write-Host $Title -ForegroundColor Cyan
+    foreach ($key in $Choices.Keys | Sort-Object) {
+        Write-Host "  $key - $($Choices[$key])"
+    }
+    $answer = Read-Host "Votre choix [$Default]"
+    if ([string]::IsNullOrWhiteSpace($answer)) { $answer = $Default }
+    if (-not $Choices.ContainsKey($answer)) {
+        Write-Host "Choix invalide, utilisation de $Default" -ForegroundColor Yellow
+        $answer = $Default
+    }
+    return $answer
+}
+
+function Test-ComfyAlive {
+    try {
+        Invoke-WebRequest -Uri 'http://127.0.0.1:8188/system_stats' -UseBasicParsing -TimeoutSec 2 | Out-Null
+        return $true
+    } catch { return $false }
+}
+
+Write-Host "============================================================" -ForegroundColor Magenta
+Write-Host " Portrait Generator V4 - Installation" -ForegroundColor Magenta
+Write-Host "============================================================" -ForegroundColor Magenta
+Write-Host "Ce programme va verifier Python, GPU, ComfyUI, PyTorch et le modele."
+Write-Host "Log detaille: $Log"
+Write-Host ""
+
+try {
+    # 1) Python
+    Log "Detection Python"
+    $PythonCmd = $null
+    $PythonVersion = $null
+
+    if (Get-Command python -ErrorAction SilentlyContinue) {
+        $PythonCmd = "python"
+        $PythonVersion = (& python --version 2>&1) -join " "
+    } elseif (Get-Command py -ErrorAction SilentlyContinue) {
+        $PythonCmd = "py -3"
+        $PythonVersion = (& py -3 --version 2>&1) -join " "
+    }
+
+    if ($PythonCmd) {
+        Write-Host "Python detecte: $PythonVersion via '$PythonCmd'" -ForegroundColor Green
+        Log "Python detecte: $PythonVersion via $PythonCmd"
+    } else {
+        Write-Host "Python n'a pas ete trouve." -ForegroundColor Yellow
+        Write-Host "Python 3.10+ est requis. L'installateur peut tenter de telecharger Python 3.12.8."
+        $installPy = Ask-Choice "Installer Python automatiquement ?" @{ "1"="Oui, installer Python 3.12.8"; "2"="Non, j'installerai Python moi-meme" } "1"
+        if ($installPy -eq "1") {
+            Log "Telechargement Python 3.12.8"
+            $pyInstaller = Join-Path $Root "python_installer.exe"
+            Invoke-WebRequest -Uri "https://www.python.org/ftp/python/3.12.8/python-3.12.8-amd64.exe" -OutFile $pyInstaller
+            Start-Process -FilePath $pyInstaller -ArgumentList "/quiet InstallAllUsers=0 PrependPath=1 Include_test=0" -Wait
+            if (Get-Command python -ErrorAction SilentlyContinue) {
+                $PythonCmd = "python"
+                $PythonVersion = (& python --version 2>&1) -join " "
+                Log "Python installe: $PythonVersion"
+            } else {
+                throw "Python installe mais pas encore detecte. Fermez cette fenetre et relancez INSTALLER.bat."
+            }
+        } else {
+            throw "Installation arretee. Installez Python 3.10+ puis relancez INSTALLER.bat."
+        }
+    }
+
+    # 2) Espace disque
+    Log "Verification espace disque"
+    $drive = (Get-Item -LiteralPath $Root).PSDrive
+    $freeGB = [math]::Round($drive.Free / 1GB, 1)
+    Write-Host "Espace libre sur le disque cible: $freeGB Go"
+    Log "Espace libre: $freeGB Go"
+    if ($freeGB -lt 20) {
+        Write-Host "ATTENTION: moins de 20 Go libres. Recommande: 20-30 Go." -ForegroundColor Yellow
+        Read-Host "Appuyez sur Entree pour continuer quand meme"
+    }
+
+    # 3) GPU / mode PyTorch
+    Log "Detection GPU"
+    $GpuDetected = $false
+    $GpuInfo = "CPU uniquement"
+    try {
+        $gpu = & nvidia-smi --query-gpu=name,memory.total,driver_version --format=csv,noheader 2>$null
+        if ($LASTEXITCODE -eq 0 -and $gpu) {
+            $GpuDetected = $true
+            $GpuInfo = $gpu
+            Write-Host "GPU NVIDIA detecte: $GpuInfo" -ForegroundColor Green
+            Log "GPU NVIDIA: $GpuInfo"
+        }
+    } catch {}
+    if (-not $GpuDetected) {
+        Write-Host "Aucun GPU NVIDIA detecte. Mode CPU possible mais tres lent." -ForegroundColor Yellow
+        Log "Aucun GPU NVIDIA detecte"
+    }
+
+    $torchChoices = @{
+        "1"="Auto recommande: CUDA 13.0 cu130 si NVIDIA, CPU sinon"
+        "2"="Forcer CUDA 13.0 cu130, recommande RTX 50 / RTX 40 recentes"
+        "3"="Forcer CPU, tres lent mais compatible sans NVIDIA"
+        "4"="Ne pas modifier PyTorch"
+    }
+    $torchChoice = Ask-Choice "Configuration PyTorch" $torchChoices "1"
+
+    # 4) ComfyUI
+    $comfyChoices = @{
+        "1"="Installer ComfyUI dans ce dossier, package autonome"
+        "2"="Utiliser ComfyUI existant: D:\IA\ComfyUI"
+        "3"="Entrer un chemin ComfyUI personnalise"
+        "4"="Ne pas installer ComfyUI, seulement installer l'application"
+    }
+    $comfyChoice = Ask-Choice "Configuration ComfyUI" $comfyChoices "1"
+
+    if ($comfyChoice -eq "2") { $ComfyDir = "D:\IA\ComfyUI" }
+    elseif ($comfyChoice -eq "3") { $ComfyDir = Read-Host "Chemin complet du dossier ComfyUI" }
+    else { $ComfyDir = Join-Path $Root "ComfyUI" }
+    $ComfyDir = [System.IO.Path]::GetFullPath($ComfyDir)
+    Log "COMFY_DIR=$ComfyDir"
+
+    "`$ComfyDir = '$ComfyDir'" | Out-File (Join-Path $Root "config_local.ps1") -Encoding UTF8
+    "set ""COMFY_DIR=$ComfyDir""" | Out-File (Join-Path $Root "config_local.bat") -Encoding ASCII
+
+    # 5) Venv + app deps
+    $Venv = Join-Path $Root "venv"
+    $PyExe = Join-Path $Venv "Scripts\python.exe"
+    $PipExe = Join-Path $Venv "Scripts\pip.exe"
+
+    if (-not (Test-Path $PyExe)) {
+        Log "Creation environnement virtuel"
+        if ($PythonCmd -eq "py -3") { & py -3 -m venv $Venv }
+        else { & python -m venv $Venv }
+        if ($LASTEXITCODE -ne 0) { throw "Creation de l'environnement virtuel echouee" }
+    }
+
+    Log "Mise a jour pip"
+    & $PyExe -m pip install --upgrade pip setuptools wheel | Tee-Object -FilePath $Log -Append
+
+    Log "Installation dependances application"
+    & $PipExe install gradio requests Pillow huggingface_hub | Tee-Object -FilePath $Log -Append
+    if ($LASTEXITCODE -ne 0) { throw "Installation dependances application echouee" }
+
+    # 6) ComfyUI download if requested
+    if ($comfyChoice -ne "4") {
+        $ComfyMain = Join-Path $ComfyDir "main.py"
+        if (-not (Test-Path $ComfyMain)) {
+            Write-Host "Telechargement ComfyUI..." -ForegroundColor Cyan
+            Log "Telechargement ComfyUI"
+            $zip = Join-Path $Root "ComfyUI.zip"
+            Invoke-WebRequest -Uri "https://github.com/comfyanonymous/ComfyUI/archive/refs/heads/master.zip" -OutFile $zip
+            $tmp = Join-Path $Root "_comfy_extract"
+            if (Test-Path $tmp) { Remove-Item $tmp -Recurse -Force }
+            Expand-Archive -Path $zip -DestinationPath $tmp -Force
+            $src = Join-Path $tmp "ComfyUI-master"
+            if (Test-Path $ComfyDir) { Remove-Item $ComfyDir -Recurse -Force }
+            New-Item -ItemType Directory -Path (Split-Path -Parent $ComfyDir) -Force | Out-Null
+            Move-Item -Path $src -Destination $ComfyDir -Force
+            Remove-Item $tmp -Recurse -Force
+            Remove-Item $zip -Force
+        } else {
+            Write-Host "ComfyUI trouve: $ComfyDir" -ForegroundColor Green
+            Log "ComfyUI deja present"
+        }
+    }
+
+    # 7) PyTorch
+    if ($torchChoice -ne "4") {
+        Log "Installation PyTorch"
+        & $PipExe uninstall -y torch torchvision torchaudio | Tee-Object -FilePath $Log -Append
+        $useCuda = $false
+        if ($torchChoice -eq "2") { $useCuda = $true }
+        elseif ($torchChoice -eq "1" -and $GpuDetected) { $useCuda = $true }
+
+        if ($useCuda) {
+            Write-Host "Installation PyTorch CUDA 13.0 cu130..." -ForegroundColor Cyan
+            & $PipExe install --upgrade torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu130 | Tee-Object -FilePath $Log -Append
+            if ($LASTEXITCODE -ne 0) {
+                Write-Host "cu130 stable echoue, tentative nightly cu130..." -ForegroundColor Yellow
+                & $PipExe install --pre --upgrade torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130 | Tee-Object -FilePath $Log -Append
+            }
+        } else {
+            Write-Host "Installation PyTorch CPU..." -ForegroundColor Cyan
+            & $PipExe install --upgrade torch torchvision torchaudio | Tee-Object -FilePath $Log -Append
+        }
+        if ($LASTEXITCODE -ne 0) { throw "Installation PyTorch echouee" }
+    }
+
+    Log "Verification PyTorch"
+    & $PyExe -c "import torch; print('TORCH', torch.__version__); print('CUDA', torch.version.cuda); print('CUDA_AVAILABLE', torch.cuda.is_available()); print('GPU', torch.cuda.get_device_name(0) if torch.cuda.is_available() else 'CPU'); print('ARCH', torch.cuda.get_arch_list() if torch.cuda.is_available() else 'CPU')" | Tee-Object -FilePath $Log -Append
+
+    # 8) Comfy requirements
+    $Req = Join-Path $ComfyDir "requirements.txt"
+    if (($comfyChoice -ne "4") -and (Test-Path $Req)) {
+        Log "Installation requirements ComfyUI"
+        & $PipExe install -r $Req | Tee-Object -FilePath $Log -Append
+    }
+
+    # 9) Model
+    if ($comfyChoice -ne "4") {
+        $modelChoice = Ask-Choice "Modele JuggernautXL Ragnarok" @{ "1"="Telecharger/verifier le modele, environ 6.6 Go"; "2"="Ne pas telecharger maintenant" } "1"
+        if ($modelChoice -eq "1") {
+            $ModelDir = Join-Path $ComfyDir "models\checkpoints"
+            $ModelFile = Join-Path $ModelDir "juggernautXL_ragnarokBy.safetensors"
+            New-Item -ItemType Directory -Path $ModelDir -Force | Out-Null
+            if (Test-Path $ModelFile) {
+                Write-Host "Modele deja present: $ModelFile" -ForegroundColor Green
+                Log "Modele deja present"
+            } else {
+                Write-Host "Telechargement du modele environ 6.6 Go..." -ForegroundColor Cyan
+                & $PyExe -c "from huggingface_hub import hf_hub_download; hf_hub_download(repo_id='maxborland/juggernautXL_ragnarokBy.safetensors', filename='juggernautXL_ragnarokBy.safetensors', local_dir=r'$ModelDir', local_dir_use_symlinks=False); print('Modele telecharge')" | Tee-Object -FilePath $Log -Append
+                if ($LASTEXITCODE -ne 0) { Write-Host "Modele non telecharge. Placez-le manuellement ici: $ModelFile" -ForegroundColor Yellow }
+            }
+        }
+    }
+
+    # 10) Summary
+    Log "Installation terminee"
+    Write-Host ""
+    Write-Host "============================================================" -ForegroundColor Green
+    Write-Host "Installation terminee" -ForegroundColor Green
+    Write-Host "Python : $PythonVersion"
+    Write-Host "GPU    : $GpuInfo"
+    Write-Host "ComfyUI: $ComfyDir"
+    Write-Host "Log    : $Log"
+    Write-Host "Lancement: LANCER.bat" -ForegroundColor Green
+    Write-Host "============================================================" -ForegroundColor Green
+}
+catch {
+    Log "ERREUR FATALE: $($_.Exception.Message)"
+    Write-Host "ERREUR: $($_.Exception.Message)" -ForegroundColor Red
+    Write-Host "Voir le log: $Log" -ForegroundColor Yellow
+}
+
+Read-Host "Appuyez sur Entree pour fermer"