@echo off
setlocal

set SRC=K:\AI\training\ldf\singled\2. autohandling_test\output\checkpoints
set DST=M:\AI\sd\novelai-webui-aki-v3-r\models\Lora\qwen

echo ============================================
echo  Deploy LoRA checkpoints to WebUI
echo ============================================
echo Source: %SRC%
echo Target: %DST%
echo.

if not exist "%SRC%" (
    echo [ABORT] Checkpoint dir not found. Run training first.
    pause
    exit /b 1
)

set COUNT=0
for %%f in ("%SRC%\myface_lora*.safetensors") do (
    copy /y "%%f" "%DST%\" >nul
    echo [COPIED] %%~nxf
    set /a COUNT+=1
)

if %COUNT%==0 (
    echo [ABORT] No myface_lora*.safetensors in source dir.
    pause
    exit /b 1
)

echo.
echo Done. %COUNT% checkpoint(s) deployed to WebUI.
echo.
echo === Verification steps in WebUI ===
echo 1. Select checkpoint: Qwen-Rapid-AIO-NSFW-v23
echo 2. Add LoRA tag to prompt: ^<lora:myface_lora-0000XX:0.8^>
echo    (XX = epoch number, e.g. 000040 = epoch 40)
echo 3. Prompt MUST include trigger word: lm_face_v1
echo 4. Compare auto sample images in %SRC%\sample
echo    to pick the best epoch first.
echo.
pause