@echo off
setlocal
chcp 65001 > nul
set target_dir=%~dp0
set custom_n=%target_dir%ComfyUI\custom_nodes
set py=%target_dir%python_embeded\

call :ColorText "=============================================================" "Yellow"
call :ColorText ".bat file for installing good-ComfyUI by Skiffbox" "Green"

call :ColorText "=============================================================" "Yellow"
echo Installing wget
curl --ssl-no-revoke -L -o wget.exe https://huggingface.co/iRedHat/iBat-ComfyUI/resolve/main/wget.exe
call :ColorText "Install wget complete" "Green"

call :ColorText "=============================================================" "Yellow"
call :ColorText "Downloading and unzip python+git" "Green"
cd %target_dir%
%target_dir%wget.exe -c -q -O Git-Python_embeded.bat https://huggingface.co/iRedHat/iBat-ComfyUI/resolve/main/Git-Python_embeded.bat
call %target_dir%Git-Python_embeded.bat
del /f /q %target_dir%Git-Python_embeded.bat

cd %target_dir%
%target_dir%wget.exe -c -q -O install_modules.py https://huggingface.co/iRedHat/iBat-ComfyUI/resolve/main/install_modules.py
%py%python.exe install_modules.py
del /f /q %target_dir%install_modules.py
del /f /q %target_dir%insightface-0.7.3-cp311-cp311-win_amd64.whl

%target_dir%wget.exe -c -q -O all_nodes.py https://huggingface.co/iRedHat/iBat-ComfyUI/resolve/main/all_nodes-good.py
%py%python.exe all_nodes.py
del /f /q %target_dir%all_nodes.py

call :ColorText "=============================================================" "Yellow"
call :ColorText "Download update.bat file" "Green"
%target_dir%wget.exe -c -q -O update.bat https://huggingface.co/iRedHat/iBat-ComfyUI/resolve/main/update.bat
%target_dir%wget.exe -c -q -O models_links.bat https://huggingface.co/iRedHat/iBat-ComfyUI/resolve/main/models_links.bat
%target_dir%wget.exe -c -q -O rename_list.txt https://huggingface.co/iRedHat/iBat-ComfyUI/resolve/main/rename_list.txt
%target_dir%wget.exe -c -q -O update_git_pull.bat https://huggingface.co/iRedHat/iBat-ComfyUI/resolve/main/update_git_pull.bat
%target_dir%wget.exe -c -q -O run_nvidia_gpu.bat https://huggingface.co/iRedHat/iBat-ComfyUI/resolve/main/run_nvidia_gpu.bat
call %target_dir%update.bat
rem call %target_dir%models_links.bat

call :ColorText "=============================================================" "Yellow"
call :ColorText "Building run.bat file" "Green"
echo @echo off> run.bat
echo setlocal>> run.bat
echo chcp 65001 > nul>> run.bat
echo set target_dir=%~dp0>> run.bat
echo set PATH=%target_dir%git;%target_dir%git\bin;%target_dir%git\libexec\;%target_dir%python_embeded;%target_dir%python_embeded\Library\bin;%target_dir%python_embeded\Scripts>> run.bat
echo set COMFYUI_PATH=%target_dir%ComfyUI>> run.bat
echo set COMFYUI_MODEL_PATH=%target_dir%ComfyUI\models>> run.bat
echo %target_dir%python_embeded\python %target_dir%ComfyUI\main.py --auto-launch --lowvram>> run.bat
echo pause>> run.bat

call :ColorText "=============================================================" "Yellow"
call :ColorText "Starting ComfyUI - run.bat" "Green"
start "" "%target_dir%run.bat"

call :ColorText "=============================================================" "Yellow"
call :ColorText "Installation of ComfyUI and Nodes complete by Skiffbox" "Green"

pause
exit /b

:ColorText
setlocal
set text=%~1
set color=%~2
powershell -NoProfile -ExecutionPolicy Bypass -Command "Write-Host '%text%' -ForegroundColor %color%"
endlocal
exit /b