@echo off
setlocal enabledelayedexpansion

set "basePath=%CD%"
set "comfyPath=%basePath%\ComfyUI"
set "pythonPath=%basePath%\python_embeded"
set "customNodesPath=%comfyPath%\custom_nodes"

if not exist "%basePath%\logs" mkdir "%basePath%\logs"

:check_folder
if exist "%basePath%\ComfyUI" (
    echo ComfyUI folder detected
	set "comfyPath=%basePath%\ComfyUI"
	if exist "%basePath%\python_embeded" (
		echo Python folder detected
		set "pythonPath=%basePath%\python_embeded"
	) else (
		echo Python folder not detected, give Python folder path :
	pause
	)
) else if exist "%basePath%\ComfyUI_windows_portable" (
	set "comfyPath=%basePath%\ComfyUI_windows_portable\ComfyUI"
	set "pythonPath=%basePath%\ComfyUI_windows_portable\python_embeded"
    echo ComfyUI folder detected
) else (
	echo ComfyUI folder not detected, give ComfyUi folder path :
	set /p "comfyPath=Path: "
	echo Python folder not detected, give Python folder path :
	set /p "pythonPath=Path: "
)

set "customNodesPath=%comfyPath%\custom_nodes"

echo Update ComfyUI
if exist "%basePath%\update" (
	"%pythonPath%\python.exe" -m pip install --upgrade pip  >> "%basePath%\logs\Missing_nodes.txt" 2>&1
	"%pythonPath%\python.exe" "%basePath%\update\update.py" "%basePath%\ComfyUI"  >> "%basePath%\logs\Missing_nodes.txt" 2>&1
	if exist update_new.py (
		move /y update_new.py update.py
		echo Running updater again since it got updated.
		"%pythonPath%\python.exe" "%basePath%\update\update.py" "%basePath%\ComfyUI" --skip_self_update  >> "%basePath%\logs\Missing_nodes.txt" 2>&1
	)
) else if exist "%basePath%\ComfyUI_windows_portable\update" (
	"%pythonPath%\python.exe" -m pip install --upgrade pip  >> "%basePath%\logs\Missing_nodes.txt" 2>&1
	"%pythonPath%\python.exe" "%basePath%\ComfyUI_windows_portable\update\update.py" "%comfyPath%"  >> "%basePath%\logs\Missing_nodes.txt" 2>&1
	if exist update_new.py (
		move /y update_new.py update.py
		echo Running updater again since it got updated.
		"%pythonPath%\python.exe" "%basePath%\ComfyUI_windows_portable\update\update.py" "%comfyPath%" --skip_self_update  >> "%basePath%\logs\Missing_nodes.txt" 2>&1
	)
) else (
	git --git-dir="%comfyPath%" --work-tree="%comfyPath%" pull >> "%basePath%\logs\Missing_nodes.txt" 2>&1
	"%pythonPath%\python.exe" -s -m pip install -r "%comfyPath%/requirements.txt" --no-warn-script-location >> "%basePath%\logs\Missing_nodes.txt" 2>&1
)

: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 "%customNodesPath%\*.*"  >> "%basePath%\logs\Missing_nodes.txt" 2>&1
for /d %%i in ("%customNodesPath%\*") do rmdir /s /q "%%i"  >> "%basePath%\logs\Missing_nodes.txt" 2>&1
) else if /i "%CHOOSE_CLEAN%"=="B" (
    set "CHOOSE_CLEANL=no"
) else (
echo Invalid choice. Please enter A or B.
    goto CHOOSE_CLEAN
)

REM Clone ComfyUI-Manager
echo Installing ComfyUI-Manager...
git clone https://github.com/ltdrdata/ComfyUI-Manager.git "%customNodesPath%/ComfyUI-Manager" >> "%basePath%\logs\Missing_nodes.txt" 2>&1

echo Installing additional nodes...

echo   - Impact-Pack
git clone https://github.com/ltdrdata/ComfyUI-Impact-Pack "%customNodesPath%/ComfyUI-Impact-Pack" >> "%basePath%\logs\Missing_nodes.txt" 2>&1
"%pythonPath%\python.exe" -s -m pip install -r "%customNodesPath%/ComfyUI-Impact-Pack/requirements.txt" --no-warn-script-location >> "%basePath%\logs\Missing_nodes.txt" 2>&1
git clone https://github.com/ltdrdata/ComfyUI-Impact-Subpack "%customNodesPath%/ComfyUI-Impact-Pack/impact_subpack" >> "%basePath%\logs\Missing_nodes.txt" 2>&1
"%pythonPath%\python.exe" -s -m pip install -r "%customNodesPath%/ComfyUI-Impact-Pack/impact_subpack/requirements.txt" --no-warn-script-location >> "%basePath%\logs\Missing_nodes.txt" 2>&1
"%pythonPath%\python.exe" -s -m pip install ultralytics --no-warn-script-location >> "%basePath%\logs\Missing_nodes.txt" 2>&1

echo   - GGUF
git clone https://github.com/city96/ComfyUI-GGUF "%customNodesPath%/ComfyUI-GGUF" >> "%basePath%\logs\Missing_nodes.txt" 2>&1
"%pythonPath%\python.exe" -s -m pip install -r "%customNodesPath%/ComfyUI-GGUF/requirements.txt" --no-warn-script-location >> "%basePath%\logs\Missing_nodes.txt" 2>&1

echo   - mxToolkit
git clone https://github.com/Smirnov75/ComfyUI-mxToolkit "%customNodesPath%/ComfyUI-mxToolkit" >> "%basePath%\logs\Missing_nodes.txt" 2>&1

echo   - Custom-Scripts
git clone https://github.com/pythongosssss/ComfyUI-Custom-Scripts "%customNodesPath%/ComfyUI-Custom-Scripts" >> "%basePath%\logs\Missing_nodes.txt" 2>&1

echo   - KJNodes
git clone https://github.com/kijai/ComfyUI-KJNodes "%customNodesPath%/ComfyUI-KJNodes" >> "%basePath%\logs\Missing_nodes.txt" 2>&1
"%pythonPath%\python.exe" -s -m pip install -r "%customNodesPath%/ComfyUI-KJNodes/requirements.txt" --no-warn-script-location >> "%basePath%\logs\Missing_nodes.txt" 2>&1

echo   - VideoHelperSuite
git clone https://github.com/Kosinkadink/ComfyUI-VideoHelperSuite "%customNodesPath%/ComfyUI-VideoHelperSuite" >> "%basePath%\logs\Missing_nodes.txt" 2>&1
"%pythonPath%\python.exe" -s -m pip install -r "%customNodesPath%/ComfyUI-VideoHelperSuite/requirements.txt" --no-warn-script-location >> "%basePath%\logs\Missing_nodes.txt" 2>&1

echo   - Frame-Interpolation
git clone https://github.com/Fannovel16/ComfyUI-Frame-Interpolation "%customNodesPath%/ComfyUI-Frame-Interpolation" >> "%basePath%\logs\Missing_nodes.txt" 2>&1
"%pythonPath%\python.exe" -s -m pip install -r "%customNodesPath%/ComfyUI-Frame-Interpolation/requirements-with-cupy.txt" --no-warn-script-location >> "%basePath%\logs\Missing_nodes.txt" 2>&1

echo   - rgthree
git clone https://github.com/rgthree/rgthree-comfy "%customNodesPath%/rgthree-comfy" >> "%basePath%\logs\Missing_nodes.txt" 2>&1
"%pythonPath%\python.exe" -s -m pip install -r "%customNodesPath%/rgthree-comfy/requirements.txt" --no-warn-script-location >> "%basePath%\logs\Missing_nodes.txt" 2>&1

echo   - Easy-Use
git clone https://github.com/yolain/ComfyUI-Easy-Use "%customNodesPath%/ComfyUI-Easy-Use" >> "%basePath%\logs\Missing_nodes.txt" 2>&1
"%pythonPath%\python.exe" -s -m pip install -r "%customNodesPath%/ComfyUI-Easy-Use/requirements.txt" --no-warn-script-location >> "%basePath%\logs\Missing_nodes.txt" 2>&1

echo   - PuLID_Flux_ll
git clone https://github.com/lldacing/ComfyUI_PuLID_Flux_ll "%customNodesPath%/ComfyUI_PuLID_Flux_ll" >> "%basePath%\logs\Missing_nodes.txt" 2>&1
"%pythonPath%\python.exe" -s -m pip install -r "%customNodesPath%/ComfyUI_PuLID_Flux_ll/requirements.txt" --no-warn-script-location >> "%basePath%\logs\Missing_nodes.txt" 2>&1
curl -L -o "%pythonPath%\insightface-0.7.3-cp310-cp310-win_amd64.whl" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/others/insightface-0.7.3-cp310-cp310-win_amd64.whl?download=true  >> "%basePath%\logs\Missing_nodes.txt" 2>&1
curl -L -o "%pythonPath%\insightface-0.7.3-cp311-cp311-win_amd64.whl" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/others/insightface-0.7.3-cp311-cp311-win_amd64.whl?download=true  >> "%basePath%\logs\Missing_nodes.txt" 2>&1
curl -L -o "%pythonPath%\insightface-0.7.3-cp312-cp312-win_amd64.whl" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/others/insightface-0.7.3-cp312-cp312-win_amd64.whl?download=true  >> "%basePath%\logs\Missing_nodes.txt" 2>&1
"%pythonPath%\python.exe" -m pip install --use-pep517 facexlib  >> "%basePath%\logs\Missing_nodes.txt" 2>&1
"%pythonPath%\python.exe" -m pip install git+https://github.com/rodjjo/filterpy.git  >> "%basePath%\logs\Missing_nodes.txt" 2>&1
"%pythonPath%\python.exe" -m pip install onnxruntime==1.19.2 onnxruntime-gpu==1.15.1 "%pythonPath%\insightface-0.7.3-cp310-cp310-win_amd64.whl"  >> "%basePath%\logs\Missing_nodes.txt" 2>&1
"%pythonPath%\python.exe" -m pip install onnxruntime==1.19.2 onnxruntime-gpu==1.15.1 "%pythonPath%\insightface-0.7.3-cp311-cp311-win_amd64.whl"  >> "%basePath%\logs\Missing_nodes.txt" 2>&1
"%pythonPath%\python.exe" -m pip install onnxruntime==1.19.2 onnxruntime-gpu==1.17.1 "%pythonPath%\insightface-0.7.3-cp312-cp312-win_amd64.whl"  >> "%basePath%\logs\Missing_nodes.txt" 2>&1

echo   - HunyuanVideoMultiLora
git clone https://github.com/facok/ComfyUI-HunyuanVideoMultiLora "%customNodesPath%/ComfyUI-HunyuanVideoMultiLora" >> "%basePath%\logs\Missing_nodes.txt" 2>&1

echo   - was-node-suite-comfyui
git clone https://github.com/WASasquatch/was-node-suite-comfyui "%customNodesPath%/was-node-suite-comfyui" >> "%basePath%\logs\Missing_nodes.txt" 2>&1
"%pythonPath%\python.exe" -s -m pip install -r "%customNodesPath%/was-node-suite-comfyui/requirements.txt" --no-warn-script-location >> "%basePath%\logs\Missing_nodes.txt" 2>&1

echo   - Florence2
git clone https://github.com/kijai/ComfyUI-Florence2  "%customNodesPath%/ComfyUI-Florence2">> "%basePath%\logs\Missing_nodes.txt" 2>&1
"%pythonPath%\python.exe" -m pip install transformers==4.49.0 --upgrade >> "%basePath%\logs\Missing_nodes.txt" 2>&1
"%pythonPath%\python.exe" -s -m pip install -r "%customNodesPath%/ComfyUI-Florence2/requirements.txt" --no-warn-script-location >> "%basePath%\logs\Missing_nodes.txt" 2>&1

echo   - Upscaler-Tensorrt
git clone https://github.com/yuvraj108c/ComfyUI-Upscaler-Tensorrt "%customNodesPath%/ComfyUI-Upscaler-Tensorrt" >> "%basePath%\logs\Missing_nodes.txt" 2>&1
"%pythonPath%\python.exe" -s -m pip install wheel-stub >> "%basePath%\logs\Missing_nodes.txt" 2>&1
"%pythonPath%\python.exe" -s -m pip install -r "%customNodesPath%/ComfyUI-Upscaler-Tensorrt/requirements.txt" --no-warn-script-location >> "%basePath%\logs\Missing_nodes.txt" 2>&1

echo   - MultiGPU
git clone https://github.com/pollockjj/ComfyUI-MultiGPU "%customNodesPath%/ComfyUI-MultiGPU" >> "%basePath%\logs\Missing_nodes.txt" 2>&1

echo   - WanStartEndFramesNative
git clone https://github.com/Flow-two/ComfyUI-WanStartEndFramesNative "%customNodesPath%/ComfyUI-WanStartEndFramesNative" >> "%basePath%\logs\Missing_nodes.txt" 2>&1

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

echo   - UltimateSDUpscale
git clone https://github.com/ssitu/ComfyUI_UltimateSDUpscale "%customNodesPath%/ComfyUI_UltimateSDUpscale" >> "%basePath%\logs\Missing_nodes.txt" 2>&1

echo   - Controlnet_AUX
git clone https://github.com/Fannovel16/comfyui_controlnet_aux "%customNodesPath%/comfyui_controlnet_aux" >> "%basePath%\logs\Missing_nodes.txt" 2>&1
"%pythonPath%\python.exe" -s -m pip install -r "%customNodesPath%/comfyui_controlnet_aux/requirements.txt" --no-warn-script-location >> "%basePath%\logs\Missing_nodes.txt" 2>&1

echo   - X-Flux
git clone https://github.com/XLabs-AI/x-flux-comfyui "%customNodesPath%/x-flux-comfyui" >> "%basePath%\logs\Missing_nodes.txt" 2>&1
"%pythonPath%\python.exe" -s -m pip install -r "%customNodesPath%/x-flux-comfyui/requirements.txt" --no-warn-script-location >> "%basePath%\logs\Missing_nodes.txt" 2>&1

echo   - RMBG
git clone https://github.com/1038lab/ComfyUI-RMBG "%customNodesPath%/ComfyUI-RMBG" >> "%basePath%\logs\Missing_nodes.txt" 2>&1
"%pythonPath%\python.exe" -s -m pip install -r "%customNodesPath%/ComfyUI-RMBG/requirements.txt" --no-warn-script-location >> "%basePath%\logs\Missing_nodes.txt" 2>&1

echo Installation complete
pause