$reqBody = @{
    prompt = "stunning landscape with mountains, cinematic lighting, 8k, beautiful"
    negative_prompt = "blurry, low quality, white background"
    steps = 30
    provider = "localai"
} | ConvertTo-Json

Write-Host "Submitting generation request..."
$job = Invoke-RestMethod -Uri "http://127.0.0.1:8008/generate" -Method POST -Body $reqBody -ContentType "application/json"
Write-Host "Job submitted: $($job.job_id)"
$jobId = $job.job_id

# Poll for completion
for ($i = 0; $i -lt 240; $i++) {
    $status = Invoke-RestMethod -Uri "http://127.0.0.1:8008/jobs/$jobId" -Method GET
    Write-Host "[$([DateTime]::Now.ToString('HH:mm:ss'))] Status: $($status.status) | Progress: $($status.progress)%"
    
    if ($status.status -eq "done") {
        Write-Host "`nGenerated images: $($status.output_images.Count)"
        if ($status.output_images.Count -gt 0) {
            Write-Host "Image path: $($status.output_images[0])"
            
            # Check file size
            $file = Get-Item $status.output_images[0] -ErrorAction SilentlyContinue
            if ($file) {
                Write-Host "File size: $($file.Length) bytes"
            }
        }
        break
    }
    Start-Sleep -Seconds 3
}