From a76ded2a98abcd716dab1c4a5780d32ee6e283e4 Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sun, 30 Apr 2023 14:24:43 -0500
Subject: [PATCH] fix(exe): add build script for Windows bundle

---
 exe/.gitignore        |  4 ++++
 exe/build.bat         | 22 ++++++++++++++++++++++
 exe/onnx-web-full.bat |  2 +-
 exe/onnx-web-half.bat |  2 +-
 4 files changed, 28 insertions(+), 2 deletions(-)
 create mode 100644 exe/.gitignore
 create mode 100644 exe/build.bat

diff --git a/exe/.gitignore b/exe/.gitignore
new file mode 100644
index 00000000..a6ac0212
--- /dev/null
+++ b/exe/.gitignore
@@ -0,0 +1,4 @@
+build/
+dist/
+
+commit.txt
\ No newline at end of file
diff --git a/exe/build.bat b/exe/build.bat
new file mode 100644
index 00000000..cc36b417
--- /dev/null
+++ b/exe/build.bat
@@ -0,0 +1,22 @@
+REM activate venv
+..\api\onnx_env\Scripts\Activate.bat
+
+REM build bundle
+pyinstaller win10.directml.dir.spec
+
+REM add additional files
+xcopy \gfpgan \dist\server\gfpgan /t /e
+xcopy ..\api\gui \dist\client /t /e
+xcopy ..\api\schemas \dist\schemas /t /e
+xcopy ..\api\logging.yaml \dist\logging.yaml
+xcopy ..\api\params.json \dist\params.json
+xcopy ..\docs \dist\docs /t /e
+xcopy ..\models \dist\models /t /e
+xcopy ..\outputs \dist\outputs /t /e
+
+REM get commit info
+git rev-parse HEAD > commit.txt
+set /p GIT_SHA=<commit.txt
+
+REM create archive
+7za a ..\dist\onnx-web-v0.10.0-rc-%GIT_SHA%.zip "\dist\*"
diff --git a/exe/onnx-web-full.bat b/exe/onnx-web-full.bat
index 46d29127..b6b7296a 100644
--- a/exe/onnx-web-full.bat
+++ b/exe/onnx-web-full.bat
@@ -6,7 +6,7 @@ set ONNX_WEB_OUTPUT_PATH=%ONNX_WEB_BASE_PATH%\outputs
 
 REM customize these as needed
 set ONNX_WEB_CACHE_MODELS=0
-set ONNX_WEB_EXTRA_MODELS=%ONNX_WEB_BASE_PATH%\models\onnx-web-extras.json
+set ONNX_WEB_EXTRA_MODELS=%ONNX_WEB_BASE_PATH%\extras.json
 
 REM convert models and launch the server
 @echo Launching onnx-web in fp32 mode...
diff --git a/exe/onnx-web-half.bat b/exe/onnx-web-half.bat
index 9a966a87..c8816fca 100644
--- a/exe/onnx-web-half.bat
+++ b/exe/onnx-web-half.bat
@@ -7,7 +7,7 @@ set ONNX_WEB_OUTPUT_PATH=%ONNX_WEB_BASE_PATH%\outputs
 REM customize these as needed
 set ONNX_WEB_BLOCK_PLATFORMS=cpu
 set ONNX_WEB_CACHE_MODELS=0
-set ONNX_WEB_EXTRA_MODELS=%ONNX_WEB_BASE_PATH%\models\onnx-web-extras.json
+set ONNX_WEB_EXTRA_MODELS=%ONNX_WEB_BASE_PATH%\models\extras.json
 
 REM convert models and launch the server
 @echo Launching onnx-web in fp16 mode...