Hugging Face's logo

Spaces:
Gregorfun
/
PixelForge
Running on Zero

App Files Community
PixelForge
File size: 4,633 Bytes
32c5da4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
param(
  [string]$BackendHost = "127.0.0.1",
  [int]$BackendPort = 8008,
  [int]$FrontendPort = 5173,
  [int]$HealthTimeoutSeconds = 90,
  [string]$PythonExe = "",
  [ValidateSet("web", "electron")][string]$FrontendMode = "web",
  [switch]$SkipFrontend,
  [switch]$RequireA1111
)

$ErrorActionPreference = "Stop"

$projectRoot = Split-Path -Parent $PSScriptRoot
Set-Location $projectRoot

function Resolve-PythonExe {
  param([string]$Override)
  if ($Override -and (Test-Path $Override)) {
    return (Resolve-Path $Override).Path
  }

  $candidates = @(
    (Join-Path $projectRoot ".venv\Scripts\python.exe"),
    (Join-Path (Split-Path -Parent $projectRoot) ".venv\Scripts\python.exe")
  )

  foreach ($candidate in $candidates) {
    if (Test-Path $candidate) {
      return (Resolve-Path $candidate).Path
    }
  }

  throw "Python executable not found. Use -PythonExe <path-to-python.exe>."
}

function Stop-PortListeners {
  param([int[]]$Ports)
  foreach ($port in $Ports) {
    $listeners = Get-NetTCPConnection -LocalPort $port -State Listen -ErrorAction SilentlyContinue
    foreach ($listener in $listeners) {
      try {
        Stop-Process -Id $listener.OwningProcess -Force -ErrorAction Stop
        Write-Host "Stopped PID $($listener.OwningProcess) on port $port"
      }
      catch {
        Write-Host "Could not stop PID $($listener.OwningProcess) on port $port"
      }
    }
  }
}

function Wait-Http {
  param(
    [Parameter(Mandatory = $true)][string]$Url,
    [Parameter(Mandatory = $true)][string]$Name,
    [Parameter(Mandatory = $true)][int]$TimeoutSeconds
  )

  $deadline = (Get-Date).AddSeconds($TimeoutSeconds)
  do {
    try {
      $response = Invoke-WebRequest -Uri $Url -UseBasicParsing -TimeoutSec 5
      if ($response.StatusCode -ge 200 -and $response.StatusCode -lt 500) {
        Write-Host "OK  $Name -> $Url"
        return
      }
    }
    catch {
      Start-Sleep -Milliseconds 800
    }
  } while ((Get-Date) -lt $deadline)

  throw "$Name not reachable within ${TimeoutSeconds}s: $Url"
}

$python = Resolve-PythonExe -Override $PythonExe
$logsDir = Join-Path $projectRoot "logs"
New-Item -ItemType Directory -Force -Path $logsDir | Out-Null

$backendLog = Join-Path $logsDir "backend.log"
$backendErrLog = Join-Path $logsDir "backend.err.log"
$frontendLog = Join-Path $logsDir "frontend.log"
$frontendErrLog = Join-Path $logsDir "frontend.err.log"

Stop-PortListeners -Ports @($BackendPort, $FrontendPort)

$backendCmd = @(
  "Set-Location '$projectRoot'",
  "`$env:IMAGEFORGE_HOST='$BackendHost'",
  "`$env:IMAGEFORGE_PORT='$BackendPort'",
  "`$env:IMAGEFORGE_CORS_ORIGINS='http://localhost:$FrontendPort,http://127.0.0.1:$FrontendPort'",
  "& '$python' -m backend.app.main"
) -join "; "

$backendProc = Start-Process -FilePath "powershell.exe" -ArgumentList @(
  "-NoProfile",
  "-ExecutionPolicy", "Bypass",
  "-Command", $backendCmd
) -RedirectStandardOutput $backendLog -RedirectStandardError $backendErrLog -PassThru

Write-Host "Backend started (PID $($backendProc.Id))."

$frontendProc = $null
if (-not $SkipFrontend.IsPresent) {
  $frontendScript = if ($FrontendMode -eq "electron") { "dev" } else { "dev:web" }
  $frontendCheckUrl = if ($FrontendMode -eq "electron") { "http://localhost:$FrontendPort" } else { "http://127.0.0.1:$FrontendPort" }
  $frontendCmd = @(
    "Set-Location '$projectRoot'",
    "npm --prefix frontend run $frontendScript"
  ) -join "; "

  $frontendProc = Start-Process -FilePath "powershell.exe" -ArgumentList @(
    "-NoProfile",
    "-ExecutionPolicy", "Bypass",
    "-Command", $frontendCmd
  ) -RedirectStandardOutput $frontendLog -RedirectStandardError $frontendErrLog -PassThru

  Write-Host "Frontend started (PID $($frontendProc.Id))."
  Wait-Http -Url $frontendCheckUrl -Name "Frontend" -TimeoutSeconds $HealthTimeoutSeconds
}

$healthArgs = @(
  "-BackendUrl", "http://$BackendHost`:$BackendPort",
  "-TimeoutSeconds", "$HealthTimeoutSeconds"
)
if ($RequireA1111.IsPresent) {
  $healthArgs += "-RequireA1111"
}

& (Join-Path $PSScriptRoot "healthcheck-stack.ps1") @healthArgs

Write-Host ""
Write-Host "Ready:"
Write-Host "- Backend: http://$BackendHost`:$BackendPort"
if (-not $SkipFrontend.IsPresent) {
  if ($FrontendMode -eq "electron") {
    Write-Host "- Frontend mode: electron (with embedded Vite on $FrontendPort)"
  }
  else {
    Write-Host "- Frontend (Web): http://localhost:$FrontendPort"
  }
}
Write-Host "- Backend log: $backendLog"
Write-Host "- Backend error log: $backendErrLog"
if (-not $SkipFrontend.IsPresent) {
  Write-Host "- Frontend log: $frontendLog"
  Write-Host "- Frontend error log: $frontendErrLog"
}