install.ps1

# PowerShell script for Windows installation

# Find a Python version that supports gradio 3.x (Python 3.8-3.11)
$PYTHON_CMD = $null
$pythonCandidates = @("python3.11", "python3.10", "python3.9", "python3.8", "py -3.11", "py -3.10", "py -3.9", "py -3.8", "python")

foreach ($cmd in $pythonCandidates) {
    try {
        $cmdParts = $cmd -split ' '
        if ($cmdParts.Length -eq 2) {
            $version = & $cmdParts[0] $cmdParts[1] -c "import sys; print(f'{sys.version_info.major}.{sys.version_info.minor}')" 2>$null
        } else {
            $version = & $cmd -c "import sys; print(f'{sys.version_info.major}.{sys.version_info.minor}')" 2>$null
        }
        if ($version) {
            $major, $minor = $version -split '\.'
            if ([int]$major -eq 3 -and [int]$minor -ge 8 -and [int]$minor -le 11) {
                $PYTHON_CMD = $cmd
                Write-Host "Found compatible Python: $cmd ($version)"
                break
            }
        }
    } catch {
        continue
    }
}

if (-not $PYTHON_CMD) {
    Write-Host "ERROR: No compatible Python found (need 3.8-3.11 for gradio 3.x)"
    Write-Host ""
    $downloadPython = Read-Host "Would you like to download a portable Python 3.11? (y/n)"
    if ($downloadPython -eq "y" -or $downloadPython -eq "Y") {
        Write-Host "Downloading portable Python 3.11..."
        $pyUrl = "https://github.com/indygreg/python-build-standalone/releases/download/20240224/cpython-3.11.8+20240224-x86_64-pc-windows-msvc-install_only.tar.gz"
        Invoke-WebRequest -Uri $pyUrl -OutFile "python.tar.gz"
        New-Item -ItemType Directory -Force -Path "portable_python" | Out-Null
        tar -xzf python.tar.gz -C portable_python --strip-components=1
        Remove-Item "python.tar.gz"
        $PYTHON_CMD = "$PWD\portable_python\python.exe"
        Write-Host "Portable Python installed: $PYTHON_CMD"
    } else {
        Write-Host "Please install Python 3.10, 3.11, or 3.12 from https://www.python.org/downloads/"
        exit 1
    }
}

# Try to detect GPU type
try {
    $gpu = Get-CimInstance -ClassName Win32_VideoController | Select-Object -First 1 -ExpandProperty Name
    if ($gpu -match "AMD") {
        $GPU_TYPE = "AMD"
        Write-Host "Detected GPU: AMD"
    } elseif ($gpu -match "NVIDIA") {
        $GPU_TYPE = "NVIDIA"
        Write-Host "Detected GPU: NVIDIA"
    } else {
        throw "No match"
    }
} catch {
    Write-Host "Could not auto-detect GPU type."
    Write-Host "Select your GPU type:"
    Write-Host "1. AMD"
    Write-Host "2. NVIDIA"
    $choice = Read-Host "Enter choice (1 or 2)"

    switch ($choice) {
        1 { $GPU_TYPE = "AMD" }
        2 { $GPU_TYPE = "NVIDIA" }
        default {
            Write-Host "Invalid choice. Exiting."
            exit 1
        }
    }
}

# Fetch asset options from GitHub releases
Write-Host "Fetching asset options from GitHub..."
$RELEASE_URL = "https://api.github.com/repos/leejet/stable-diffusion.cpp/releases/latest"
try {
    $release = Invoke-RestMethod -Uri $RELEASE_URL
    $assets = $release.assets | Select-Object -ExpandProperty name
} catch {
    Write-Host "Failed to fetch assets. Please ensure internet connection."
    exit 1
}

if ($GPU_TYPE -eq "AMD") {
    $ASSET_NAME = $assets | Where-Object { $_ -match "win" -and $_ -match "rocm" } | Select-Object -First 1
} elseif ($GPU_TYPE -eq "NVIDIA") {
    $ASSET_NAME = $assets | Where-Object { $_ -match "win" -and $_ -match "cuda" } | Select-Object -First 1
}

if (-not $ASSET_NAME) {
    Write-Host "No suitable asset found for $GPU_TYPE."
    exit 1
}

$DOWLOAD_URL = "https://github.com/leejet/stable-diffusion.cpp/releases/latest/download/$ASSET_NAME"

if (Test-Path "stable-diffusion-cpp") {
    Write-Host "stable-diffusion-cpp already exists, skipping download."
} else {
    Write-Host "Downloading $GPU_TYPE release..."
    Invoke-WebRequest -Uri $DOWLOAD_URL -OutFile $ASSET_NAME
    if ($?) {
        Write-Host "Download complete. Unzipping to stable-diffusion-cpp..."
        Expand-Archive -Path $ASSET_NAME -DestinationPath "stable-diffusion-cpp"
        Write-Host "Done."
        Remove-Item $ASSET_NAME
    } else {
        Write-Host "Download failed."
        exit 1
    }
}

Write-Host "Creating virtual environment in .venv..."
if (Test-Path ".venv") {
    Write-Host "Virtual environment already exists, skipping."
} else {
    $cmdParts = $PYTHON_CMD -split ' '
    if ($cmdParts.Length -eq 2) {
        & $cmdParts[0] $cmdParts[1] -m venv .venv
    } else {
        & $PYTHON_CMD -m venv .venv
    }
}
& ".venv\Scripts\Activate.ps1"
Write-Host "Installing required packages: pillow, gradio, requests..."
pip install pillow gradio==3.41.2 requests
deactivate

Write-Host "Creating model directories..."
New-Item -ItemType Directory -Force -Path checkpoints, vae, lora

@'
if (-not (Test-Path ".venv")) {
    Write-Host "ERROR: Virtual environment not found. Run install.ps1 first."
    exit 1
}
& ".venv\Scripts\Activate.ps1"
pip install --upgrade pillow gradio==3.41.2 requests
python -m sd_ui.app
'@ | Out-File -FilePath run.ps1 -Encoding UTF8

Write-Host "Setup complete, run with './run.ps1'"