$ErrorActionPreference = "Stop"
$Root = Split-Path -Parent $PSScriptRoot
Set-Location $Root
if (-not (Test-Path ".venv")) { py -3.12 -m venv .venv }
$Py = Join-Path $Root ".venv\Scripts\python.exe"
& $Py -m pip install --upgrade pip
# Keep ORT on CUDA 12.x. ORT >=1.27 defaults to CUDA 13.
& $Py -m pip install "onnxruntime-gpu[cuda,cudnn]>=1.24,<1.27" torch huggingface_hub numpy pyzmq
if (-not (Test-Path ".vendor\NitroGen")) {
  New-Item -ItemType Directory -Force ".vendor" | Out-Null
  git clone https://github.com/MineDojo/NitroGen.git .vendor/NitroGen
  git -C .vendor/NitroGen checkout 32608444660950ffda95e1e57c79632ad65bea10
}
& $Py -m pip install -e .vendor/NitroGen
Write-Host "`nReady. Benchmark:" -ForegroundColor Green
Write-Host ".\.venv\Scripts\python.exe scripts\benchmark_dit.py"
Write-Host "Gameplay server (fast profile):"
Write-Host ".\.venv\Scripts\python.exe scripts\serve_onnx.py --steps 4"