From fcfe2e9ffd9b9ee8af63926a002602503f1ad71a Mon Sep 17 00:00:00 2001
From: Andrew Savchyn <dev@scorpil.com>
Date: Fri, 30 Sep 2022 10:51:05 +0200
Subject: [PATCH] Configurable launch script

---
 webui-user.sh | 3 +++
 webui.sh      | 7 ++++++-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/webui-user.sh b/webui-user.sh
index 0ce41d3f..11c316cb 100644
--- a/webui-user.sh
+++ b/webui-user.sh
@@ -21,6 +21,9 @@ export COMMANDLINE_ARGS=""
 # python3 venv without trailing slash (defaults to ${install_dir}/${clone_dir}/venv)
 #venv_dir="venv"
 
+# script to launch to start the app
+#export launch_script="launch.py"
+
 # install command for torch
 #export TORCH_COMMAND="pip install torch==1.12.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113"
 
diff --git a/webui.sh b/webui.sh
index 8355c9df..6e27a4a6 100755
--- a/webui.sh
+++ b/webui.sh
@@ -41,6 +41,11 @@ then
     venv_dir="venv"
 fi
 
+if [[ -z "${launch_script}" ]]
+then
+    launch_script="launch.py"
+fi
+
 # Disable sentry logging
 export ERROR_REPORTING=FALSE
 
@@ -133,4 +138,4 @@ fi
 printf "\n%s\n" "${delimiter}"
 printf "Launching launch.py..."
 printf "\n%s\n" "${delimiter}"
-"${python_cmd}" launch.py
+"${python_cmd}" "${launch_script}"