@echo off
SETLOCAL

echo.
echo ============================================
echo   SunoDoc Setup - Windows
echo ============================================
echo.

REM ---- Check Python ----
python --version >nul 2>&1
IF ERRORLEVEL 1 (
    echo ERROR: Python not found. Install Python 3.10+ from python.org
    pause
    exit /b 1
)

echo [1/6] Creating virtual environment...
python -m venv venv
call venv\Scripts\activate.bat

echo [2/6] Upgrading pip...
python -m pip install --upgrade pip setuptools wheel

echo [3/6] Installing base dependencies...
pip install gradio==4.44.0
pip install huggingface_hub
pip install TTS
pip install pywhispercpp
pip install Pillow
pip install psutil

echo.
echo ============================================
echo   CHOOSE YOUR llama-cpp-python BUILD
echo   (Check check_gpu.py output)
echo ============================================
echo.
echo   [1] NVIDIA GPU (CUDA 12.1)
echo   [2] NVIDIA GPU (CUDA 12.4)
echo   [3] CPU only
echo.
set /p GPU_CHOICE="Enter 1, 2, or 3: "

IF "%GPU_CHOICE%"=="1" (
    echo [4/6] Installing llama-cpp-python with CUDA 12.1 support...
    pip install llama-cpp-python --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu121
) ELSE IF "%GPU_CHOICE%"=="2" (
    echo [4/6] Installing llama-cpp-python with CUDA 12.4 support...
    pip install llama-cpp-python --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu124
) ELSE (
    echo [4/6] Installing llama-cpp-python for CPU...
    pip install llama-cpp-python
)

echo [5/6] Verifying installs...
python -c "import gradio; print('  gradio:', gradio.__version__)"
python -c "from llama_cpp import Llama; print('  llama-cpp-python: OK')"
python -c "from TTS.api import TTS; print('  Coqui TTS: OK')"
python -c "from pywhispercpp.model import Model; print('  pywhispercpp: OK')"

echo [6/6] Setup complete!
echo.
echo ============================================
echo   NEXT STEP: Download the model
echo   Run: python download_model.py
echo ============================================
echo.
pause