Upload Install-ComfyUI.ps1

scripts/ComfyUI/Install-ComfyUI.ps1

@@ -0,0 +1,384 @@
+#Requires -RunAsAdministrator
+
+<#
+.SYNOPSIS
+    A complete, all-in-one installer for ComfyUI and its environment.
+.DESCRIPTION
+    This single, robust script handles everything:
+    - Sets the required TLS 1.2 security protocol.
+    - Directly installs the Aria2 binary for accelerated downloads.
+    - Installs Git and 7-Zip if they are missing.
+    - Downloads and extracts the latest portable version of ComfyUI.
+    - Installs a comprehensive list of custom nodes and their Python dependencies.
+    It is designed to be fully compatible with PowerShell 5.1 (standard on Windows 10).
+.NOTES
+    Author: Code Partner (based on original scripts and user feedback)
+    Version: 4.1 (Compatibility Fix)
+#>
+
+#===========================================================================
+# SECTION 1: SCRIPT CONFIGURATION & HELPER FUNCTIONS
+#===========================================================================
+
+# --- Force TLS 1.2 for all web requests (Crucial Fix) ---
+[Net.ServicePointManager]::SecurityProtocol = [Net.SecurityProtocolType]::Tls12
+
+# --- Path & Log Variables ---
+$installPath = $PSScriptRoot
+$basePath = Join-Path $installPath "ComfyUI_windows_portable"
+$comfyPath = Join-Path $basePath "ComfyUI"
+$pythonExecutable = Join-Path $basePath "python_embeded\python.exe"
+$customNodesPath = Join-Path $comfyPath "custom_nodes"
+$logPath = Join-Path $installPath "logs"
+$logFile = Join-Path $logPath "install_log.txt"
+$sevenZipPath = "C:\Program Files\7-Zip\7z.exe" # Default path
+
+# --- Create Log Directory ---
+if (-not (Test-Path $logPath)) {
+    New-Item -ItemType Directory -Force -Path $logPath | Out-Null
+}
+
+# --- Helper function to write to console and log file ---
+function Write-Log {
+    param([string]$Message, [string]$Color = "White")
+    $timestamp = Get-Date -Format "yyyy-MM-dd HH:mm:ss"
+    $formattedMessage = "[$timestamp] $Message"
+    Write-Host $Message -ForegroundColor $Color
+    Add-Content -Path $logFile -Value $formattedMessage
+}
+
+# --- Helper function to execute a command and log its output (VERSION-COMPATIBLE) ---
+function Invoke-AndLog {
+    param(
+        [string]$File,
+        [string]$Arguments
+    )
+    # This method is compatible with all PowerShell versions, including 5.1.
+    # It uses cmd.exe to run the command and handle output redirection.
+    # The structure is: cmd /c "full command >> "path/to/log.txt" 2>&1"
+    $commandToRun = "`"$File`" $Arguments"
+    $cmdArguments = "/C `"$commandToRun >> `"`"$logFile`"`" 2>&1`""
+    
+    try {
+        Start-Process -FilePath "cmd.exe" -ArgumentList $cmdArguments -Wait -WindowStyle Hidden
+    } catch {
+        Write-Log "FATAL ERROR trying to execute command: $commandToRun" -Color Red
+    }
+}
+
+# --- Function to install the aria2 binary directly ---
+function Install-Aria2-Binary {
+    Write-Log "--- Starting Aria2 binary installation ---" -Color Magenta
+    $destFolder = "C:\Tools\aria2"
+    if (-not (Test-Path $destFolder)) { New-Item -ItemType Directory -Force -Path $destFolder | Out-Null }
+
+    $aria2Url = "https://github.com/aria2/aria2/releases/download/release-1.36.0/aria2-1.36.0-win-64bit-build1.zip"
+    $zipPath  = Join-Path $env:TEMP "aria2_temp.zip"
+    Write-Log "Downloading aria2 from $aria2Url..."
+    Invoke-WebRequest -Uri $aria2Url -OutFile $zipPath
+
+    Write-Log "Extracting zip file to $destFolder..."
+    Expand-Archive -Path $zipPath -DestinationPath $destFolder -Force
+
+    $extractedSubfolder = Join-Path $destFolder "aria2-1.36.0-win-64bit-build1"
+    if (Test-Path $extractedSubfolder) {
+        Move-Item -Path (Join-Path $extractedSubfolder "*") -Destination $destFolder -Force
+        Remove-Item -Path $extractedSubfolder -Recurse -Force
+    }
+
+    $configFile = Join-Path $destFolder "aria2.conf"
+    $configContent = "continue=true`nmax-connection-per-server=16`nsplit=16`nmin-split-size=1M`nfile-allocation=none"
+    $configContent | Out-File $configFile -Encoding UTF8
+
+    $envScope = "User"
+    $oldPath = [System.Environment]::GetEnvironmentVariable("Path", $envScope)
+    if ($oldPath -notlike "*$destFolder*") {
+        Write-Log "Adding '$destFolder' to user PATH..."
+        $newPath = $oldPath + ";$destFolder"
+        [System.Environment]::SetEnvironmentVariable("Path", $newPath, $envScope)
+        $env:Path = $newPath
+        Write-Log "PATH updated. Aria2 will be available immediately."
+    }
+    Write-Log "--- Aria2 binary installation complete ---" -Color Magenta
+}
+
+
+# --- Smart Download Function with fallback ---
+function Download-File {
+    param(
+        [string]$Uri,
+        [string]$OutFile
+    )
+    # Extrait juste le nom du fichier de l'URL pour un affichage plus propre
+    $fileName = Split-Path -Path $Uri -Leaf
+
+    if (Get-Command 'aria2c' -ErrorAction SilentlyContinue) {
+        Write-Log "  - Downloading $fileName"
+        $aria_args = "-c -x 16 -s 16 -k 1M --dir=`"$((Split-Path $OutFile -Parent))`" --out=`"$((Split-Path $OutFile -Leaf))`" `"$Uri`""
+        Invoke-AndLog "aria2c" $aria_args
+    } else {
+        Write-Log "Aria2 not found. Falling back to standard download: $fileName" -Color Yellow
+        Invoke-WebRequest -Uri $Uri -OutFile $OutFile
+    }
+}
+
+
+#===========================================================================
+# SECTION 2: MAIN SCRIPT EXECUTION
+#===========================================================================
+
+Clear-Host
+# --- Banner ---
+# The ASCII art is now embedded directly in the script for simplicity and reliability.
+Write-Log "-------------------------------------------------------------------------------"
+$asciiBanner = @'
+                      __  __               ___    _ ____  ______
+                     / / / /___ ___  ___  /   |  (_) __ \/_  __/
+                    / / / / __ `__ \/ _ \/ /| | / / /_/ / / / 
+                   / /_/ / / / / / /  __/ ___ |/ / _, _/ / /
+                   \____/_/ /_/ /_/\___/_/  |_/_/_/ |_| /_/
+'@
+Write-Host $asciiBanner -ForegroundColor Cyan
+Write-Log "-------------------------------------------------------------------------------"
+Write-Log "                          ComfyUI - All-in-One Installer                       " -Color Yellow
+Write-Log "                                  Version 3.0                                  " -Color White
+Write-Log "-------------------------------------------------------------------------------"
+
+# --- Step 1: Install Dependencies (Aria2, 7-Zip, Git) ---
+Write-Log "Step 1: Checking for required tools..." -Color Yellow
+
+# Aria2
+if (-not (Get-Command 'aria2c' -ErrorAction SilentlyContinue)) {
+    Write-Log "Aria2 not found. Proceeding with direct installation."
+    Install-Aria2-Binary
+} else {
+    Write-Log "Aria2 is already installed." -Color Green
+}
+
+# 7-Zip
+if (-not (Test-Path $sevenZipPath)) {
+    Write-Log "7-Zip not found. Downloading and installing..."
+    $sevenZipInstaller = Join-Path $env:TEMP "7z-installer.exe"
+    Download-File -Uri "https://www.7-zip.org/a/7z2201-x64.exe" -OutFile $sevenZipInstaller
+    Start-Process -FilePath $sevenZipInstaller -ArgumentList "/S" -Wait
+    Remove-Item $sevenZipInstaller
+} else {
+    Write-Log "7-Zip is already installed." -Color Green
+}
+
+# Git
+if (-not (Get-Command git.exe -ErrorAction SilentlyContinue)) {
+    Write-Log "Git not found. Downloading and installing..."
+    $gitInstaller = Join-Path $env:TEMP "Git-Installer.exe"
+    Download-File -Uri "https://github.com/git-for-windows/git/releases/download/v2.41.0.windows.3/Git-2.41.0.3-64-bit.exe" -OutFile $gitInstaller
+    Start-Process -FilePath $gitInstaller -ArgumentList "/VERYSILENT" -Wait
+    Remove-Item $gitInstaller
+}
+Invoke-AndLog "git" "config --system core.longpaths true"
+Write-Log "Git is ready." -Color Green
+
+# --- Step 2: Download and Extract ComfyUI ---
+Write-Log "`nStep 2: Downloading and setting up ComfyUI..." -Color Yellow
+if (-not (Test-Path $basePath)) {
+    $comfyArchive = Join-Path $installPath "ComfyUI_windows_portable_nvidia.7z"
+    Download-File -Uri "https://github.com/comfyanonymous/ComfyUI/releases/download/v0.3.43/ComfyUI_windows_portable_nvidia.7z" -OutFile $comfyArchive
+    Write-Log "  - Extracting ComfyUI..."
+    Invoke-AndLog "$sevenZipPath" "x `"$comfyArchive`" -o`"$installPath`" -y"
+    Remove-Item $comfyArchive
+} else {
+    Write-Log "ComfyUI folder already exists. Skipping download." -Color Green
+}
+
+# --- Step 3: Update ComfyUI and Install Custom Nodes ---
+Write-Log "`nStep 3: Updating ComfyUI and installing custom nodes..." -Color Yellow
+# Update pip and ComfyUI itself
+Invoke-AndLog "$pythonExecutable" "-m pip install --upgrade pip"
+Invoke-AndLog "$pythonExecutable" "`"$basePath\update\update.py`" `"$comfyPath`""
+Invoke-AndLog "$pythonExecutable" "-s -m pip install -r `"$comfyPath\requirements.txt`""
+# Install special Python dependencies
+Write-Log "  - Installing special Python dependencies" -Color White
+Invoke-AndLog "$pythonExecutable" "-s -m pip install ultralytics --no-warn-script-location"
+$insightfaceWheel = Join-Path $basePath "python_embeded\insightface-0.7.3-cp312-cp312-win_amd64.whl"
+Download-File -Uri "https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/whl/insightface-0.7.3-cp312-cp312-win_amd64.whl" -OutFile $insightfaceWheel
+Invoke-AndLog "$pythonExecutable" "-m pip install --use-pep517 facexlib"
+Invoke-AndLog "$pythonExecutable" "-m pip install git+https://github.com/rodjjo/filterpy.git"
+Invoke-AndLog "$pythonExecutable" "-m pip install onnxruntime==1.19.2 onnxruntime-gpu==1.17.1"
+Invoke-AndLog "$pythonExecutable" "-s -m pip install $insightfaceWheel --no-warn-script-location"
+
+# Define all custom nodes in one place
+$customNodes = @(
+    @{Name="ComfyUI-Manager"; Repo="https://github.com/ltdrdata/ComfyUI-Manager.git"; HasRequirements=$false},
+    @{Name="ComfyUI-Impact-Pack"; Repo="https://github.com/ltdrdata/ComfyUI-Impact-Pack"; HasRequirements=$true},
+    @{Name="ComfyUI-Impact-Subpack"; Repo="https://github.com/ltdrdata/ComfyUI-Impact-Subpack"; Subfolder="ComfyUI-Impact-Pack/impact_subpack"; HasRequirements=$true},
+    @{Name="ComfyUI-GGUF"; Repo="https://github.com/city96/ComfyUI-GGUF"; HasRequirements=$true},
+    @{Name="ComfyUI-mxToolkit"; Repo="https://github.com/Smirnov75/ComfyUI-mxToolkit"; HasRequirements=$false},
+    @{Name="ComfyUI-Custom-Scripts"; Repo="https://github.com/pythongosssss/ComfyUI-Custom-Scripts"; HasRequirements=$false},
+    @{Name="ComfyUI-KJNodes"; Repo="https://github.com/kijai/ComfyUI-KJNodes"; HasRequirements=$true},
+    @{Name="ComfyUI-WanVideoWrapper"; Repo="https://github.com/kijai/ComfyUI-WanVideoWrapper"; HasRequirements=$true},
+    @{Name="ComfyUI-VideoHelperSuite"; Repo="https://github.com/Kosinkadink/ComfyUI-VideoHelperSuite"; HasRequirements=$true},
+    @{Name="ComfyUI-Frame-Interpolation"; Repo="https://github.com/Fannovel16/ComfyUI-Frame-Interpolation"; RequirementsFile="requirements-with-cupy.txt"; HasRequirements=$true},
+    @{Name="rgthree-comfy"; Repo="https://github.com/rgthree/rgthree-comfy"; HasRequirements=$true},
+    @{Name="ComfyUI-Easy-Use"; Repo="https://github.com/yolain/ComfyUI-Easy-Use"; HasRequirements=$true},
+    @{Name="ComfyUI_PuLID_Flux_ll"; Repo="https://github.com/lldacing/ComfyUI_PuLID_Flux_ll"; HasRequirements=$true},
+    @{Name="ComfyUI-HunyuanVideoMultiLora"; Repo="https://github.com/facok/ComfyUI-HunyuanVideoMultiLora"; HasRequirements=$false},
+    @{Name="was-node-suite-comfyui"; Repo="https://github.com/WASasquatch/was-node-suite-comfyui"; HasRequirements=$true},
+    @{Name="ComfyUI-Florence2"; Repo="https://github.com/kijai/ComfyUI-Florence2"; HasRequirements=$true},
+    @{Name="ComfyUI-MultiGPU"; Repo="https://github.com/pollockjj/ComfyUI-MultiGPU"; HasRequirements=$false},
+    @{Name="ComfyUI-WanStartEndFramesNative"; Repo="https://github.com/Flow-two/ComfyUI-WanStartEndFramesNative"; HasRequirements=$false},
+    @{Name="ComfyUI-Image-Saver"; Repo="https://github.com/alexopus/ComfyUI-Image-Saver"; HasRequirements=$true},
+    @{Name="ComfyUI_UltimateSDUpscale"; Repo="https://github.com/ssitu/ComfyUI_UltimateSDUpscale"; HasRequirements=$false},
+    @{Name="comfyui_controlnet_aux"; Repo="https://github.com/Fannovel16/comfyui_controlnet_aux"; HasRequirements=$true},
+    @{Name="x-flux-comfyui"; Repo="https://github.com/XLabs-AI/x-flux-comfyui"; HasRequirements=$true},
+    @{Name="ComfyUI-RMBG"; Repo="https://github.com/1038lab/ComfyUI-RMBG"; HasRequirements=$true},
+    @{Name="ComfyUI-Detail-Daemon"; Repo="https://github.com/Jonseed/ComfyUI-Detail-Daemon"; HasRequirements=$true},
+    @{Name="ComfyUI-TeaCache"; Repo="https://github.com/welltop-cn/ComfyUI-TeaCache"; HasRequirements=$true},
+    @{Name="ComfyUI-Crystools"; Repo="https://github.com/crystian/ComfyUI-Crystools"; HasRequirements=$true}
+)
+
+# Loop through and install each custom node
+foreach ($node in $customNodes) {
+    $nodePath = if ($node.Subfolder) { Join-Path $customNodesPath $node.Subfolder } else { Join-Path $customNodesPath $node.Name }
+    
+    if (-not (Test-Path $nodePath)) {
+        Write-Log "  - Installing $($node.Name)..."
+        # For sub-repositories, clone into the parent directory
+        $cloneTargetPath = if ($node.Subfolder) { (Split-Path $nodePath -Parent) } else { $nodePath }
+        # Special case for subpack which needs a specific final directory name
+        $finalDirName = if ($node.Name -eq 'ComfyUI-Impact-Subpack') { 'impact_subpack' } else { $node.Name }
+        $cloneCommand = "clone $($node.Repo) `"$((Join-Path (Split-Path $cloneTargetPath -Parent) $finalDirName))`""
+        
+        Invoke-AndLog "git" "clone $($node.Repo) `"$cloneTargetPath`""
+
+        if ($node.HasRequirements) {
+            $requirementsFile = if ($node.RequirementsFile) { $node.RequirementsFile } else { "requirements.txt" }
+            $reqPath = Join-Path $nodePath $requirementsFile
+            if (Test-Path $reqPath) {
+                Invoke-AndLog "$pythonExecutable" "-s -m pip install -r `"$reqPath`" --no-warn-script-location"
+            }
+        }
+    } else {
+        Write-Log "  - $($node.Name) (already exists, skipping)" -Color Green
+    }
+}
+
+# Install supplementary performance modules
+Write-Log "`nStep 4: Installing supplementary performance modules..." -Color Yellow
+
+# Triton
+Write-Log "  - Installing Triton..."
+$tritonWheel = Join-Path $basePath "python_embeded\triton-3.3.0-py3-none-any.whl"
+Download-File -Uri "https://github.com/woct0rdho/triton-windows/releases/download/empty/triton-3.3.0-py3-none-any.whl" -OutFile $tritonWheel
+Invoke-AndLog "$pythonExecutable" "-m pip install `"$tritonWheel`""
+Invoke-AndLog "$pythonExecutable" "-s -m pip install triton-windows"
+
+# xformers
+Write-Log "  - Installing xformers..."
+Invoke-AndLog "$pythonExecutable" "-m pip install -U xformers --index-url https://download.pytorch.org/whl/cu128"
+Write-Log "  - Checking xformers files for compatibility (looking for file named 'pyd')..."
+
+$xformersBaseDir = Join-Path $basePath "python_embeded\Lib\site-packages\xformers"
+$dirsToProcess = @(
+    $xformersBaseDir,
+    (Join-Path $xformersBaseDir "flash_attn_3")
+)
+
+foreach ($dir in $dirsToProcess) {
+    Write-Log "      - Checking directory: $dir"
+    if (Test-Path $dir) {
+        # On vérifie l'existence d'un fichier nommé EXACTEMENT "pyd"
+        $exactFilePath = Join-Path $dir "pyd"
+        if (Test-Path $exactFilePath) {
+            # Le fichier existe, on le renomme.
+            Write-Log "        - Found file named 'pyd'. Renaming to '_C.pyd'..." -Color Yellow
+            $newName = "_C.pyd"
+            try {
+                Rename-Item -Path $exactFilePath -NewName $newName -Force -ErrorAction Stop
+                Write-Log "        - SUCCESS: Renamed 'pyd' to '$newName'" -Color Green
+            } catch {
+                Write-Log "        - FAILED to rename. Error: $($_.Exception.Message)" -Color Red
+            }
+        } else {
+            # Le fichier 'pyd' n'a pas été trouvé. On vérifie si '_C.pyd' existe déjà.
+            $finalFilePath = Join-Path $dir "_C.pyd"
+            if (Test-Path $finalFilePath) {
+                Write-Log "        - File '_C.pyd' already exists. No action needed." -Color Green
+            } else {
+                Write-Log "        - No file named 'pyd' or '_C.pyd' found in this directory." -Color Gray
+            }
+        }
+    } else {
+        # Le répertoire lui-même n'existe pas.
+        Write-Log "      - Directory not found. Skipping." -Color Gray
+    }
+}
+
+# SageAttention
+Write-Log "  - Installing SageAttention..."
+$sageWheel = Join-Path $basePath "python_embeded\sageattention-2.1.1+cu128torch2.7.0-cp312-cp312-win_amd64.whl"
+Download-File -Uri "https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/whl/sageattention-2.1.1+cu128torch2.7.0-cp312-cp312-win_amd64.whl" -OutFile $sageWheel
+Invoke-AndLog "$pythonExecutable" "-m pip install `"$sageWheel`""
+
+# --- Step 5: Download Custom Workflows ---
+Write-Log "`nStep 5: Downloading UmeAiRT Workflows..." -Color Yellow
+
+$workflowPath = Join-Path $comfyPath "user\default\workflows"
+$cloneDest = Join-Path $workflowPath "UmeAiRT-Workflow"
+
+# Create the target directory if it doesn't exist
+if (-not (Test-Path $workflowPath)) {
+    Write-Log "  - Creating workflow directory: $workflowPath"
+    New-Item -Path $workflowPath -ItemType Directory -Force | Out-Null
+}
+
+# Clone the workflow repository if it doesn't already exist
+if (-not (Test-Path $cloneDest)) {
+    Write-Log "  - Cloning workflows from GitHub into '$cloneDest'..."
+    Invoke-AndLog "git" "clone https://github.com/UmeAiRT/ComfyUI-Workflows `"$cloneDest`""
+    Write-Log "  - Workflows downloaded successfully." -Color Green
+} else {
+    Write-Log "  - Workflow directory already exists. Skipping download." -Color Green
+}
+
+# --- Step 6: Optional Model Pack Downloads ---
+Write-Log "`nStep 6: Optional Model Pack Downloads..." -Color Yellow
+
+# Définition des packs de modèles et de leurs scripts PowerShell correspondants
+$modelPacks = @(
+    @{Name="FLUX"; ScriptName="Download-FLUX-Models.ps1"}
+    # Ajoutez ici les autres packs de modèles (WAN, HIDREAM, etc.) quand vous les aurez convertis
+)
+# (Vous devrez créer les scripts Download-WAN-Models.ps1, etc., sur le même modèle)
+
+# Boucle pour chaque pack de modèles
+foreach ($pack in $modelPacks) {
+    # On pourrait télécharger le script ici si nécessaire, mais pour l'instant on assume qu'il est local
+    $scriptPath = Join-Path $installPath $pack.ScriptName
+    if (-not (Test-Path $scriptPath)) {
+        Write-Log "Model downloader script not found: $($pack.ScriptName). Skipping." -Color Red
+        continue # Passe au pack suivant
+    }
+
+    $validInput = $false
+    while (-not $validInput) {
+        Write-Host "`nWould you like to download $($pack.Name) models? (Y/N)"
+        $choice = Read-Host
+
+        if ($choice -eq 'Y' -or $choice -eq 'y') {
+            Write-Log "  - Launching downloader for $($pack.Name) models..." -Color Green
+            # Exécute le script PowerShell de téléchargement
+            & $scriptPath
+            $validInput = $true
+        } elseif ($choice -eq 'N' -or $choice -eq 'n') {
+            Write-Log "  - Skipping download for $($pack.Name) models." -Color Gray
+            $validInput = $true
+        } else {
+            Write-Host "  Invalid choice. Please enter Y or N."
+        }
+    }
+}
+
+#===========================================================================
+# SECTION 7: FINALIZATION
+#===========================================================================
+Write-Log "-------------------------------------------------------------------------------" -Color Green
+Write-Log "Installation of ComfyUI and all nodes is complete!" -Color Green