start.ps1

#Requires -Version 5.1
<#
.SYNOPSIS
    ImageForge — Windows RTX launcher (CUDA), PowerShell edition.

.DESCRIPTION
    Creates/reuses a .venv, installs imageforge[console], and launches the app.
    Targets Windows 10/11 + RTX 2000 Ada (Ada Lovelace) or any CUDA GPU.
    No arch -arm64, no MPS — pure CUDA path.

.NOTES
    PyTorch + CUDA must be installed once before first run:
        pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
    Check https://pytorch.org/get-started/locally/ for the right CUDA version.

    Run from the repo root:
        .\start.ps1
    or from anywhere:
        & "C:\path\to\imageforge\start.ps1"
#>

Set-StrictMode -Version Latest
$ErrorActionPreference = 'Stop'

# ── locate repo root (directory containing this script) ─────────────────────
$RepoRoot = $PSScriptRoot

# ── venv path (override with $env:IMAGEFORGE_VENV) ──────────────────────────
if (-not $env:IMAGEFORGE_VENV) {
    $VenvPath = Join-Path $RepoRoot '.venv'
} else {
    $VenvPath = $env:IMAGEFORGE_VENV
}

Write-Host "[imageforge] Repo root : $RepoRoot"
Write-Host "[imageforge] Venv      : $VenvPath"

# ── create venv if absent ────────────────────────────────────────────────────
$VenvPython = Join-Path $VenvPath 'Scripts\python.exe'
if (-not (Test-Path $VenvPython)) {
    Write-Host "[imageforge] Creating venv ..."
    & python -m venv $VenvPath
    if ($LASTEXITCODE -ne 0) {
        Write-Error "python -m venv failed. Is Python 3.11+ on PATH?"
        exit 1
    }
}

# ── activate ─────────────────────────────────────────────────────────────────
$ActivateScript = Join-Path $VenvPath 'Scripts\Activate.ps1'
. $ActivateScript

# ── install / update the console extras ──────────────────────────────────────
Write-Host "[imageforge] Installing/updating imageforge[console] ..."
& pip install -e "$RepoRoot\.[console]" --quiet
if ($LASTEXITCODE -ne 0) {
    Write-Error "pip install failed."
    exit 1
}

# ── set CUDA device ───────────────────────────────────────────────────────────
$env:DEVICE = 'cuda'

# ── launch ────────────────────────────────────────────────────────────────────
Write-Host "[imageforge] Launching console (DEVICE=cuda) ..."
& python -m imageforge.console.app @args