@echo off
setlocal
cd /d "%~dp0"

echo ============================================================
echo Taotern 200M Branch-Only Chat - Windows Test Launcher
echo ============================================================
echo.
echo This will create/use .venv, install CUDA PyTorch, install the
echo packaged TaoTrain/Taotern_SSM code, then launch fixed chat.
echo.

echo [1/2] Trying setup with PyTorch CUDA 12.8...
powershell -NoProfile -ExecutionPolicy Bypass -File "%~dp0setup_windows.ps1" -TorchFlavor cu128
if errorlevel 1 (
  echo.
  echo CUDA 12.8 setup failed. Trying CUDA 12.6...
  powershell -NoProfile -ExecutionPolicy Bypass -File "%~dp0setup_windows.ps1" -TorchFlavor cu126
  if errorlevel 1 (
    echo.
    echo CUDA setup failed. You can try CPU-only setup manually:
    echo   powershell -ExecutionPolicy Bypass -File .\setup_windows.ps1 -TorchFlavor cpu
    echo.
    pause
    exit /b 1
  )
)

echo.
echo [2/2] Launching fixed chat...
powershell -NoProfile -ExecutionPolicy Bypass -File "%~dp0run_chat_fixed.ps1"
echo.
pause