@echo off
setlocal
chcp 65001 > nul
set target_dir=%~dp0
set custom_n=%target_dir%ComfyUI\custom_nodes
set py=%target_dir%python_embeded\
set gi=%target_dir%git\bin\
set PATH=%target_dir%git\bin;%target_dir%git\libexec;%py%Scripts;%py%Library\bin;%PATH%;

call :ColorText "=============================================================" "Yellow"
call :ColorText ".bat file for updating instaled ComfyUI and Nodes by Skiffbox" "Green"

cd %custom_n% && %gi%git pull
%py%python.exe -s -m pip install -r %target_dir%ComfyUI/requirements.txt
rem %py%python.exe -s -m pip install --upgrade torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu121 -r %target_dir%ComfyUI/requirements.txt pygit2
cd %target_dir%
endlocal
exit /b

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