@echo off
setlocal enabledelayedexpansion

set "comfyUIPath=%CD%"

:check_folder
if exist "%comfyUIPath%\ComfyUI" (
    echo ComfyUI folder detected
) else if exist "%comfyUIPath%\ComfyUI_windows_portable" (
	cd ComfyUI_windows_portable
    echo ComfyUI folder detected
) else (
	echo ComfyUI folder not detected
	pause
	exit /b
)

echo Update ComfyUI
cd update
..\python_embeded\python.exe -m pip install --upgrade pip
..\python_embeded\python.exe .\update.py ..\ComfyUI\
if exist update_new.py (
  move /y update_new.py update.py
  echo Running updater again since it got updated.
  ..\python_embeded\python.exe .\update.py ..\ComfyUI\ --skip_self_update
)

cd ..
cd ComfyUI\custom_nodes

REM Clone ComfyUI-Manager
echo Installing ComfyUI-Manager...
git clone https://github.com/ltdrdata/ComfyUI-Manager.git >nul 2>&1
echo Installing additional nodes...
echo   - Impact-Pack
git clone https://github.com/ltdrdata/ComfyUI-Impact-Pack >nul 2>&1
cd ComfyUI-Impact-Pack
git clone https://github.com/ltdrdata/ComfyUI-Impact-Subpack impact_subpack >nul 2>&1
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location >nul 2>&1
..\..\..\python_embeded\python.exe -s -m pip install ultralytics --no-warn-script-location >nul 2>&1
cd ..

echo   - WAS-Suite
git clone https://github.com/WASasquatch/was-node-suite-comfyui >nul 2>&1
cd was-node-suite-comfyui
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location >nul 2>&1
cd ..

echo   - GGUF
git clone https://github.com/city96/ComfyUI-GGUF >nul 2>&1
cd ComfyUI-GGUF
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location >nul 2>&1
cd ..

echo   - Custom-Scripts
git clone https://github.com/pythongosssss/ComfyUI-Custom-Scripts >nul 2>&1

echo   - UltimateSDUpscale
git clone https://github.com/ssitu/ComfyUI_UltimateSDUpscale --recursive >nul 2>&1

echo   - rgthree
git clone https://github.com/rgthree/rgthree-comfy >nul 2>&1
cd rgthree-comfy
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location >nul 2>&1
cd ..

echo   - Florence2
git clone https://github.com/kijai/ComfyUI-Florence2 >nul 2>&1
cd ComfyUI-Florence2
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location >nul 2>&1
cd ..

echo   - KJNodes
git clone https://github.com/kijai/ComfyUI-KJNodes >nul 2>&1
cd ComfyUI-KJNodes
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location >nul 2>&1
cd ..

echo   - ComfyUI-Image-Saver
git clone https://github.com/alexopus/ComfyUI-Image-Saver >nul 2>&1
cd ComfyUI-Image-Saver
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location >nul 2>&1
cd ..

echo   - ComfyUI-RMBG
git clone https://github.com/1038lab/ComfyUI-RMBG >nul 2>&1
cd ComfyUI-RMBG
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location >nul 2>&1
cd ..

echo   - X-Flux
git clone https://github.com/XLabs-AI/x-flux-comfyui >nul 2>&1
cd x-flux-comfyui
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location >nul 2>&1
cd ..

echo   - mxToolkit
git clone https://github.com/Smirnov75/ComfyUI-mxToolkit >nul 2>&1

echo   - Comfyroll
git clone https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes >nul 2>&1

echo   - PulID
git clone https://github.com/sipie800/ComfyUI-PuLID-Flux-Enhanced >nul 2>&1
cd ..\..\python_embeded
curl -L -o "insightface-0.7.3-cp311-cp311-win_amd64.whl" https://github.com/Gourieff/Assets/raw/main/Insightface/insightface-0.7.3-cp311-cp311-win_amd64.whl  >nul 2>&1
.\python.exe -m pip install --use-pep517 facexlib  >nul 2>&1
.\python.exe -m pip install git+https://github.com/rodjjo/filterpy.git  >nul 2>&1
.\python.exe -m pip install onnxruntime==1.19.2 onnxruntime-gpu==1.15.1 insightface-0.7.3-cp311-cp311-win_amd64.whl  >nul 2>&1
cd ..
cd ComfyUI\custom_nodes\ComfyUI-PuLID-Flux-Enhanced
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location  >nul 2>&1
cd ..