@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
)

:CHOOSE_CLEAN
REM Ask user if they want to download FLUX SCHNELL Model
echo Do you want to do a clean install? (all currently present custom nodes are deleted)
echo A) Yes
echo B) No
set /p "CHOOSE_CLEAN=Enter your choice (A or B) and press Enter: "

if /i "%CHOOSE_CLEAN%"=="A" (
del /s /q "%comfyUIPath%\ComfyUI\custom_nodes\*.*"  >nul 2>&1
for /d %%i in ("%comfyUIPath%\ComfyUI\custom_nodes\*") do rmdir /s /q "%%i"  >nul 2>&1
) else if /i "%CHOOSE_CLEAN%"=="B" (
    set "CHOOSE_CLEANL=no"
) else (
echo Invalid choice. Please enter A or B.
    goto CHOOSE_CLEAN
)


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   - 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   - mxToolkit
git clone https://github.com/Smirnov75/ComfyUI-mxToolkit >nul 2>&1

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

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-VideoHelperSuite
git clone https://github.com/Kosinkadink/ComfyUI-VideoHelperSuite >nul 2>&1
cd ComfyUI-VideoHelperSuite
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location >nul 2>&1
cd ..

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

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   - ComfyUI-WanVideoWrapper
git clone https://github.com/kijai/ComfyUI-HunyuanVideoWrapper >nul 2>&1
cd ComfyUI-HunyuanVideoWrapper
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt --no-warn-script-location >nul 2>&1
cd ..

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

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

echo Installation complete
pause