107 lines
3.5 KiB
Batchfile
107 lines
3.5 KiB
Batchfile
@echo off
|
|
title Local AI Image Generator
|
|
cd /d "%~dp0"
|
|
|
|
set APP=%~dp0app
|
|
set NODE=%APP%\tools\node\node.exe
|
|
set NPM=%APP%\tools\node\npm.cmd
|
|
set DIST=%APP%\dist\index.html
|
|
set SETUP=%~dp0scripts\setup.ps1
|
|
set CUDA_BACKEND=%APP%\backend\win\cuda\sd-cuda.exe
|
|
set VULKAN_BACKEND=%APP%\backend\win\vulkan\sd-vulkan.exe
|
|
set SERVE=%~dp0scripts\serve.cjs
|
|
set SETUP_REASON=
|
|
set SETUP_MODE=Repair
|
|
|
|
:: ── First-time setup check ────────────────────────────────────────────────────
|
|
if not exist "%APP%\tools\node" set SETUP_MODE=First-Time Setup
|
|
if not exist "%NODE%" (
|
|
set SETUP_REASON=Portable Node.js is missing.
|
|
goto :run_setup
|
|
)
|
|
if not exist "%NPM%" (
|
|
set SETUP_REASON=Portable npm is missing.
|
|
goto :run_setup
|
|
)
|
|
if not exist "%DIST%" (
|
|
set SETUP_REASON=Frontend build is missing.
|
|
goto :run_setup
|
|
)
|
|
if exist "%CUDA_BACKEND%" goto :launch
|
|
if exist "%VULKAN_BACKEND%" goto :launch
|
|
set SETUP_REASON=No backend binary is installed.
|
|
goto :run_setup
|
|
|
|
:run_setup
|
|
echo.
|
|
echo ============================================================
|
|
echo LOCAL AI IMAGE GENERATOR ^| %SETUP_MODE%
|
|
echo ============================================================
|
|
echo.
|
|
if "%SETUP_MODE%"=="First-Time Setup" (
|
|
echo This looks like your first run. Setting up automatically...
|
|
) else (
|
|
echo Local AI Image Generator needs a quick repair before launch.
|
|
)
|
|
if not "%SETUP_REASON%"=="" echo Reason: %SETUP_REASON%
|
|
echo Models are not downloaded during setup. Download or import them in the app.
|
|
echo.
|
|
echo Press any key to continue, or Ctrl+C to cancel.
|
|
pause >nul
|
|
|
|
:: Clear old local server processes before setup so app/tools/node can be replaced
|
|
for /f "tokens=5" %%a in ('netstat -aon 2^>nul ^| findstr ":8080 "') do taskkill /f /pid %%a >nul 2>nul
|
|
for /f "tokens=5" %%a in ('netstat -aon 2^>nul ^| findstr ":1420 "') do taskkill /f /pid %%a >nul 2>nul
|
|
|
|
powershell -ExecutionPolicy Bypass -File "%SETUP%"
|
|
if errorlevel 1 (
|
|
echo.
|
|
echo [ERROR] Setup failed. Please check the output above.
|
|
pause
|
|
exit /b 1
|
|
)
|
|
|
|
:: After setup, continue to launch
|
|
goto :launch
|
|
|
|
:: ── Launch ────────────────────────────────────────────────────────────────────
|
|
:launch
|
|
cls
|
|
echo.
|
|
echo ============================================================
|
|
echo LOCAL AI IMAGE GENERATOR ^| Launching...
|
|
echo ============================================================
|
|
echo.
|
|
|
|
:: Kill anything on our ports
|
|
echo Clearing ports 8080 and 1420...
|
|
for /f "tokens=5" %%a in ('netstat -aon 2^>nul ^| findstr ":8080 "') do taskkill /f /pid %%a >nul 2>nul
|
|
for /f "tokens=5" %%a in ('netstat -aon 2^>nul ^| findstr ":1420 "') do taskkill /f /pid %%a >nul 2>nul
|
|
|
|
:: Start frontend server + backend manager (serve.cjs manages sd-vulkan.exe)
|
|
echo Starting Local AI Image Generator...
|
|
start "SD-Server" /min "%NODE%" "%SERVE%"
|
|
|
|
:: Wait for server to bind
|
|
timeout /t 2 >nul
|
|
|
|
:: Open browser
|
|
echo Opening browser at http://localhost:1420
|
|
start http://localhost:1420
|
|
|
|
echo.
|
|
echo ============================================================
|
|
echo Running!
|
|
echo Web UI: http://localhost:1420
|
|
echo GPU API: http://127.0.0.1:8080
|
|
echo.
|
|
echo Close this window to stop all services.
|
|
echo ============================================================
|
|
echo.
|
|
pause >nul
|
|
|
|
:: Cleanup on exit
|
|
echo Shutting down...
|
|
taskkill /fi "WINDOWTITLE eq SD-Server*" /f >nul 2>nul
|
|
echo Done. Goodbye!
|