@echo off
setlocal enabledelayedexpansion

set "basePath=%CD%"

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

:check_folder
if exist "%basePath%\ComfyUI" (
    echo ComfyUI folder detected
	set "modelsPath=%basePath%\ComfyUI\models"
) else if exist "%basePath%\ComfyUI_windows_portable" (
    echo ComfyUI folder detected
	set "modelsPath=%basePath%\ComfyUI_windows_portable\ComfyUI\models"
) else (
	echo ComfyUI folder not detected, give ComfyUI folder path:
    set /p "comfyPath=Path: "
    set "modelsPath=!comfyPath!\models"
    echo comfyPath = !comfyPath!
    echo modelsPath = !modelsPath!
    pause
)

:CHOOSE_OPTION
REM Ask user for installation type
echo Choose your preferred installation:
echo A) Fast-Lowvram install (GGUF)
echo B) Unoptimized normal model (bf16)
set /p "CHOICE=Enter your choice (A or B) and press Enter: "

if /i "%CHOICE%"=="A" (
    set "INSTALL_TYPE=fast-lowvram"
) else if /i "%CHOICE%"=="B" (
    set "INSTALL_TYPE=unoptimized"
	set "DOWNLOAD_GGUF=no"
) else (
echo Invalid choice. Please enter A or B.
    goto CHOOSE_OPTION
)

:CHOOSE_FLUX_GGUF
REM Ask user if they want to download FLUX GGUF Model
echo Do you want to download FLUX GGUF Models?
echo A) Q8_0 + T5_Q8 (24GB Vram)
echo B) Q5_K_S + T5_Q5_K_M (16GB Vram)
echo C) Q4_K_S + T5_Q3_K_L (less than 12GB Vram)
echo D) All
echo E) No
set /p "FLUX_GGUF_CHOICE=Enter your choice (A,B,C,D or E) and press Enter: "
	if /i "%FLUX_GGUF_CHOICE%"=="A" (
	set "DOWNLOAD_GGUF=yes"
) else if /i "%FLUX_GGUF_CHOICE%"=="B" (
	set "DOWNLOAD_GGUF=yes"
) else if /i "%FLUX_GGUF_CHOICE%"=="C" (
	set "DOWNLOAD_GGUF=yes"
) else if /i "%FLUX_GGUF_CHOICE%"=="D" (
	set "DOWNLOAD_GGUF=yes"
) else if /i "%FLUX_GGUF_CHOICE%"=="E" (
	set "DOWNLOAD_GGUF=no"
) else (
	echo Invalid choice. Please enter A or B.
	goto CHOOSE_FLUX_GGUF
)

REM Download VAE file
echo Downloading VAE file...
curl -L -o "%modelsPath%/vae/wan_2.1_vae.safetensors" https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/resolve/main/split_files/vae/wan_2.1_vae.safetensors?download=true

REM Download CLIP files
echo Downloading CLIP files...
curl -L -o "%modelsPath%/clip/umt5_xxl_fp8_e4m3fn_scaled.safetensors" https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/resolve/main/split_files/text_encoders/umt5_xxl_fp8_e4m3fn_scaled.safetensors?download=true

REM Download Diffusion models files
if "%INSTALL_TYPE%"=="unoptimized" (
	echo Downloading diffusion models file...
	echo T2V Quant Model :
	curl -L -o "%modelsPath%/diffusion_models/wan2.1_i2v_480p_14B_bf16.safetensors" https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/resolve/main/split_files/diffusion_models/wan2.1_i2v_480p_14B_bf16.safetensors?download=true
	echo I2V Quant Model :
	curl -L -o "%modelsPath%/diffusion_models/wan2.1_t2v_1.3B_bf16.safetensors" https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/resolve/main/split_files/diffusion_models/wan2.1_t2v_1.3B_bf16.safetensors?download=true
)
if "%DOWNLOAD_GGUF%"=="yes" (
    echo Downloading GGUF T2V Quant Model...
    if /i "%FLUX_GGUF_CHOICE%"=="A" (
	curl -L -o "%modelsPath%/diffusion_models/wan2.1-t2v-14b-Q8_0.gguf" https://huggingface.co/city96/Wan2.1-T2V-14B-gguf/resolve/main/wan2.1-t2v-14b-Q8_0.gguf?download=true
	) else if /i "%FLUX_GGUF_CHOICE%"=="B" (
	curl -L -o "%modelsPath%/diffusion_models/wan2.1-t2v-14b-Q5_K_M.gguf" https://huggingface.co/city96/Wan2.1-T2V-14B-gguf/resolve/main/wan2.1-t2v-14b-Q5_K_M.gguf?download=true
	) else if /i "%FLUX_GGUF_CHOICE%"=="C" (
	curl -L -o "%modelsPath%/diffusion_models/wan2.1-t2v-14b-Q3_K_S.gguf" https://huggingface.co/city96/Wan2.1-T2V-14B-gguf/resolve/main/wan2.1-t2v-14b-Q3_K_S.gguf?download=true
	) else if /i "%FLUX_GGUF_CHOICE%"=="D" (
	curl -L -o "%modelsPath%/diffusion_models/wan2.1-t2v-14b-Q8_0.gguf" https://huggingface.co/city96/Wan2.1-T2V-14B-gguf/resolve/main/wan2.1-t2v-14b-Q8_0.gguf?download=true
	curl -L -o "%modelsPath%/diffusion_models/wan2.1-t2v-14b-Q5_K_M.gguf" https://huggingface.co/city96/Wan2.1-T2V-14B-gguf/resolve/main/wan2.1-t2v-14b-Q5_K_M.gguf?download=true
	curl -L -o "%modelsPath%/diffusion_models/wan2.1-t2v-14b-Q3_K_S.gguf" https://huggingface.co/city96/Wan2.1-T2V-14B-gguf/resolve/main/wan2.1-t2v-14b-Q3_K_S.gguf?download=true
	)
)
if "%DOWNLOAD_GGUF%"=="yes" (
    echo Downloading GGUF I2V Quant Model...
    if /i "%FLUX_GGUF_CHOICE%"=="A" (
	curl -L -o "%modelsPath%/diffusion_models/wan2.1-i2v-14b-480p-Q8_0.gguf" https://huggingface.co/city96/Wan2.1-I2V-14B-480P-gguf/resolve/main/wan2.1-i2v-14b-480p-Q8_0.gguf?download=true
	) else if /i "%FLUX_GGUF_CHOICE%"=="B" (
	curl -L -o "%modelsPath%/diffusion_models/wan2.1-i2v-14b-480p-Q5_K_M.gguf" https://huggingface.co/city96/Wan2.1-I2V-14B-480P-gguf/resolve/main/wan2.1-i2v-14b-480p-Q5_K_M.gguf?download=true
	) else if /i "%FLUX_GGUF_CHOICE%"=="C" (
	curl -L -o "%modelsPath%/diffusion_models/wan2.1-i2v-14b-480p-Q3_K_S.gguf" https://huggingface.co/city96/Wan2.1-I2V-14B-480P-gguf/resolve/main/wan2.1-i2v-14b-480p-Q3_K_S.gguf?download=true
	) else if /i "%FLUX_GGUF_CHOICE%"=="D" (
	curl -L -o "%modelsPath%/diffusion_models/wan2.1-i2v-14b-480p-Q8_0.gguf" https://huggingface.co/city96/Wan2.1-I2V-14B-480P-gguf/resolve/main/wan2.1-i2v-14b-480p-Q8_0.gguf?download=true
	curl -L -o "%modelsPath%/diffusion_models/wan2.1-i2v-14b-480p-Q5_K_M.gguf" https://huggingface.co/city96/Wan2.1-I2V-14B-480P-gguf/resolve/main/wan2.1-i2v-14b-480p-Q5_K_M.gguf?download=true
	curl -L -o "%modelsPath%/diffusion_models/wan2.1-i2v-14b-480p-Q3_K_S.gguf" https://huggingface.co/city96/Wan2.1-I2V-14B-480P-gguf/resolve/main/wan2.1-i2v-14b-480p-Q3_K_S.gguf?download=true
	)
)

REM Download clip vision
echo Downloading clip vision file...
curl -L -o "%modelsPath%/clip_vision/clip_vision_h.safetensors" https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/resolve/main/split_files/clip_vision/clip_vision_h.safetensors?download=true

REM Download upscale model
echo Downloading upscale models...
curl -L -o "%modelsPath%/upscale_models/RealESRGAN_x4plus.pth" https://huggingface.co/spaces/Marne/Real-ESRGAN/resolve/main/RealESRGAN_x4plus.pth?download=true
curl -L -o "%modelsPath%/upscale_models/RealESRGAN_x4plus_anime_6B.pth" https://huggingface.co/spaces/Marne/Real-ESRGAN/resolve/main/RealESRGAN_x4plus_anime_6B.pth?download=true

echo Models downloaded.
pause