run_bench.ps1

# Phase 7.1 AAP Benchmark Runner
# Usage:
#   .\scripts\run_benchmark.ps1
#   .\scripts\run_benchmark.ps1 -Quick
#   .\scripts\run_benchmark.ps1 -Url http://localhost:7860
#   .\scripts\run_benchmark.ps1 -Suite phase7   # run original phase7 benchmark instead

param(
    [string]$Url    = "http://localhost:7860",
    [switch]$Quick,
    [ValidateSet("aap","phase7","both")]
    [string]$Suite  = "aap"
)

$ErrorActionPreference = "Stop"
Set-Location $PSScriptRoot\..

Write-Host ""
Write-Host "================================================" -ForegroundColor Cyan
Write-Host "  Codette Phase 7.1 Benchmark" -ForegroundColor Cyan
Write-Host "================================================" -ForegroundColor Cyan
Write-Host "  Server : $Url"
Write-Host "  Suite  : $Suite"
Write-Host "  Quick  : $Quick"
Write-Host ""

# Check server is up
Write-Host "Checking server..." -NoNewline
try {
    $resp = Invoke-WebRequest -Uri "$Url/health" -TimeoutSec 3 -UseBasicParsing -ErrorAction SilentlyContinue
    if ($resp.StatusCode -eq 200) { Write-Host " OK" -ForegroundColor Green }
} catch {
    try {
        $resp = Invoke-WebRequest -Uri "$Url/api/status" -TimeoutSec 3 -UseBasicParsing -ErrorAction SilentlyContinue
        Write-Host " OK (via /api/status)" -ForegroundColor Green
    } catch {
        Write-Host " UNREACHABLE" -ForegroundColor Red
        Write-Host ""
        Write-Host "Server not responding at $Url" -ForegroundColor Red
        Write-Host "Start it with:  python inference\codette_server.py" -ForegroundColor Yellow
        exit 1
    }
}

# Run selected suite(s)
$exitCode = 0

if ($Suite -eq "aap" -or $Suite -eq "both") {
    Write-Host ""
    Write-Host "Running Phase 7.1 AAP benchmark..." -ForegroundColor Cyan
    $args = @("benchmarks\phase71_aap_benchmark.py", "--url", $Url)
    if ($Quick) { $args += "--quick" }
    python @args
    if ($LASTEXITCODE -ne 0) { $exitCode = $LASTEXITCODE }
}

if ($Suite -eq "phase7" -or $Suite -eq "both") {
    Write-Host ""
    Write-Host "Running Phase 7 benchmark..." -ForegroundColor Cyan
    python benchmarks\phase7_benchmark.py
    if ($LASTEXITCODE -ne 0) { $exitCode = $LASTEXITCODE }
}

Write-Host ""
if ($exitCode -eq 0) {
    Write-Host "Benchmark complete." -ForegroundColor Green
} else {
    Write-Host "Benchmark finished with errors (exit $exitCode)." -ForegroundColor Yellow
}

exit $exitCode