@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 Do you want to download LTXV base models?
echo A) 13B (30Gb)
echo B) 2B (7Gb)
echo C) All
echo D) No
set /p "CHOICE=Enter your choice (A,B,C or D) and press Enter: "
	if /i "%CHOICE%"=="A" (
	set "DOWNLOAD=yes"
) else if /i "%CHOICE%"=="B" (
	set "DOWNLOAD=yes"
) else if /i "%CHOICE%"=="C" (
	set "DOWNLOAD=yes"
) else if /i "%CHOICE%"=="D" (
	set "DOWNLOAD=no"
) else (
	echo Invalid choice. Please enter A,B,C,D or E.
	goto CHOOSE_OPTION
)

:CHOOSE_LTXV_GGUF
REM Ask user if they want to download WAN GGUF Model
echo Do you want to download LTXV GGUF models?
echo A) Q8_0 (24GB Vram)
echo B) Q5_K_M (16GB Vram)
echo C) Q3_K_S (less than 12GB Vram)
echo D) All
echo E) No
set /p "LTXV_GGUF_CHOICE=Enter your choice (A,B,C,D or E) and press Enter: "
	if /i "%LTXV_GGUF_CHOICE%"=="A" (
	set "DOWNLOAD_GGUF=yes"
) else if /i "%LTXV_GGUF_CHOICE%"=="B" (
	set "DOWNLOAD_GGUF=yes"
) else if /i "%LTXV_GGUF_CHOICE%"=="C" (
	set "DOWNLOAD_GGUF=yes"
) else if /i "%LTXV_GGUF_CHOICE%"=="D" (
	set "DOWNLOAD_GGUF=yes"
) else if /i "%LTXV_GGUF_CHOICE%"=="E" (
	set "DOWNLOAD_GGUF=no"
) else (
	echo Invalid choice. Please enter A,B,C,D or E.
	goto CHOOSE_LTXV_GGUF
)

if "%DOWNLOAD%"=="yes" (
    echo Downloading checkpoint models file...
	powershell -NoProfile -Command "$p = Join-Path $env:modelsPath 'checkpoints\LTXV'; if (-Not (Test-Path $p)) { New-Item -ItemType Directory -Path $p | Out-Null }"
    if /i "%CHOICE%"=="A" (
	echo 13B Model :
	curl -L -o "%modelsPath%/checkpoints/LTXV/ltxv-13b-0.9.7-dev.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/checkpoints/LTXV/ltxv-13b-0.9.7-dev.safetensors?download=true
	
	) else if /i "%CHOICE%"=="B" (
	echo 2B Model :
	curl -L -o "%modelsPath%/checkpoints/LTXV/ltxv-2b-0.9.6-dev-04-25.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/checkpoints/LTXV/ltxv-2b-0.9.6-dev-04-25.safetensors?download=true
	
	) else if /i "%CHOICE%"=="C" (
	echo 13B Model :
	curl -L -o "%modelsPath%/checkpoints/LTXV/ltxv-13b-0.9.7-dev.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/checkpoints/LTXV/ltxv-13b-0.9.7-dev.safetensors?download=true
	echo 2B Model :
	curl -L -o "%modelsPath%/checkpoints/LTXV/ltxv-2b-0.9.6-dev-04-25.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/checkpoints/LTXV/ltxv-2b-0.9.6-dev-04-25.safetensors?download=true
	) 
)

REM Download VAE file
echo Downloading VAE file...
curl -L -o "%modelsPath%/vae/ltxv-13b-0.9.7-vae-BF16.safetensors" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/vae/ltxv-13b-0.9.7-vae-BF16.safetensors?download=true

if "%DOWNLOAD_GGUF%"=="yes" (
    echo Downloading GGUF Quant Model...
	powershell -NoProfile -Command "$p = Join-Path $env:modelsPath 'unet\LTXV'; if (-Not (Test-Path $p)) { New-Item -ItemType Directory -Path $p | Out-Null }"
    if /i "%LTXV_GGUF_CHOICE%"=="A" (
	curl -L -o "%modelsPath%/unet/LTXV/ltxv-13b-0.9.7-dev-Q8_0.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/LTXV/ltxv-13b-0.9.7-dev-Q8_0.gguf?download=true
	) else if /i "%LTXV_GGUF_CHOICE%"=="B" (
	curl -L -o "%modelsPath%/unet/LTXV/ltxv-13b-0.9.7-dev-Q5_K_M.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/LTXV/ltxv-13b-0.9.7-dev-Q5_K_M.gguf?download=true
	) else if /i "%LTXV_GGUF_CHOICE%"=="C" (
	curl -L -o "%modelsPath%/unet/LTXV/ltxv-13b-0.9.7-dev-Q3_K_S.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/LTXV/ltxv-13b-0.9.7-dev-Q3_K_S.gguf?download=true
	) else if /i "%LTXV_GGUF_CHOICE%"=="D" (
	curl -L -o "%modelsPath%/unet/LTXV/ltxv-13b-0.9.7-dev-Q8_0.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/LTXV/ltxv-13b-0.9.7-dev-Q8_0.gguf?download=true
	curl -L -o "%modelsPath%/unet/LTXV/ltxv-13b-0.9.7-dev-Q5_K_M.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/LTXV/ltxv-13b-0.9.7-dev-Q5_K_M.gguf?download=true
	curl -L -o "%modelsPath%/unet/LTXV/ltxv-13b-0.9.7-dev-Q3_K_S.gguf" https://huggingface.co/UmeAiRT/ComfyUI-Auto_installer/resolve/main/models/unet/LTXV/ltxv-13b-0.9.7-dev-Q3_K_S.gguf?download=true
	)
)

echo Models downloaded.
pause