From 8edc4399526c98fc8d75c364f171f118ed187568 Mon Sep 17 00:00:00 2001
From: Sanster <cwq1913@gmail.com>
Date: Tue, 25 Oct 2022 14:41:01 +0800
Subject: [PATCH] add win installer

---
 scripts/pack.bat                    | 40 +++++++++++++++++------------
 scripts/user_scripts/win_config.bat | 12 +++++++++
 scripts/user_scripts/win_start.bat  |  7 +++++
 3 files changed, 43 insertions(+), 16 deletions(-)
 create mode 100644 scripts/user_scripts/win_config.bat
 create mode 100644 scripts/user_scripts/win_start.bat

diff --git a/scripts/pack.bat b/scripts/pack.bat
index 47ad058..22aca01 100644
--- a/scripts/pack.bat
+++ b/scripts/pack.bat
@@ -1,22 +1,30 @@
 @echo off
 
-export "PYTHONNOUSERSITE=1"
+set "PYTHONNOUSERSITE=1"
+
 SET BUILD_DIST=lama-cleaner
 SET BUILD_ENV=installer
 SET USER_SCRIPTS=user_scripts
-echo "Creating a distributable package.."
-source "~\miniconda3\etc\profile.d\conda.sh"
-conda "install" "-c" "conda-forge" "-y" "conda-pack"
-conda "env" "create" "--prefix" "%BUILD_ENV%" "-f" "environment.yaml"
-conda "activate" "%CD%\%BUILD_ENV%"
-conda "pack" "--n-threads" "-1" "--prefix" "%BUILD_ENV%" "--format" "tar"
-mkdir "-p" "%BUILD_DIST%/%BUILD_ENV%"
+
+
+echo Creating a distributable package..
+@call conda env create --prefix %BUILD_ENV% -f environment.yaml
+
+echo Finish creating environment
+@call conda activate .\%BUILD_ENV%
+@call conda install -c conda-forge -y conda-pack
+
+@call conda pack --n-threads -1 --prefix %BUILD_ENV% --format tar
+
+mkdir %BUILD_DIST%\%BUILD_ENV%
+
 echo "Copy user scripts file %USER_SCRIPTS%"
-COPY  "%USER_SCRIPTS%/*" "%BUILD_DIST%"
-cd "%BUILD_DIST%"
-tar "-xf" "%CD%.\%BUILD_ENV%%CD%tar" "-C" "%BUILD_ENV%"
-cd "%CD%."
-DEL /S "%BUILD_ENV%"
-DEL  "%BUILD_ENV%%CD%tar"
-echo "zip %BUILD_DIST%%CD%zip"
-zip "-q" "-r" "%BUILD_DIST%%CD%zip" "%BUILD_DIST%"
\ No newline at end of file
+copy  %USER_SCRIPTS%\* %BUILD_DIST%
+
+cd %BUILD_DIST%
+@call tar -xf ..\%BUILD_ENV%.tar -C %BUILD_ENV%
+
+cd ..
+@call conda deactivate
+rmdir /s /q %BUILD_ENV%
+del  %BUILD_ENV%.tar
diff --git a/scripts/user_scripts/win_config.bat b/scripts/user_scripts/win_config.bat
new file mode 100644
index 0000000..51a72b9
--- /dev/null
+++ b/scripts/user_scripts/win_config.bat
@@ -0,0 +1,12 @@
+@echo off
+
+set PATH=C:\Windows\System32;%PATH%
+
+@call installer\Scripts\activate.bat
+
+@call conda-unpack
+
+@call pip3 install -U torch==1.12.1 --extra-index-url https://download.pytorch.org/whl/cu116
+@call pip3 install -U lama-cleaner
+
+@call invoke config
diff --git a/scripts/user_scripts/win_start.bat b/scripts/user_scripts/win_start.bat
new file mode 100644
index 0000000..c9b1648
--- /dev/null
+++ b/scripts/user_scripts/win_start.bat
@@ -0,0 +1,7 @@
+@echo off
+
+set PATH=C:\Windows\System32;%PATH%
+
+@call installer\Scripts\activate.bat
+
+@call invoke start