@echo off
setlocal

echo ============================================================
echo  LLM Paper Introduction Builder — Windows Setup
echo ============================================================
echo.

REM Check Python
python --version >nul 2>&1
if errorlevel 1 (
    echo [ERROR] Python was not found. Please install Python 3.10 or newer from:
    echo         https://www.python.org/downloads/
    echo         Make sure to check "Add Python to PATH" during installation.
    pause
    exit /b 1
)

echo [OK] Python found:
python --version
echo.

REM Create virtual environment
if not exist ".venv" (
    echo [STEP] Creating virtual environment...
    python -m venv .venv
    if errorlevel 1 (
        echo [ERROR] Failed to create virtual environment.
        pause
        exit /b 1
    )
    echo [OK] Virtual environment created.
) else (
    echo [OK] Virtual environment already exists, skipping creation.
)
echo.

REM Activate and upgrade pip
echo [STEP] Upgrading pip...
.venv\Scripts\python.exe -m pip install --upgrade pip --quiet
echo.

REM Install dependencies
echo [STEP] Installing dependencies (this may take a few minutes)...
.venv\Scripts\pip.exe install ^
    streamlit>=1.36 ^
    openai>=1.0 ^
    anthropic>=0.100 ^
    google-genai>=1.0 ^
    requests>=2.31
if errorlevel 1 (
    echo [ERROR] Failed to install one or more packages.
    pause
    exit /b 1
)
echo.
echo [OK] Core packages installed.
echo.

REM Optional: local GPU models
set /p GPU="Do you want to install local GPU model support (transformers, torch)? [y/N]: "
if /i "%GPU%"=="y" (
    echo [STEP] Installing local model packages (large download)...
    .venv\Scripts\pip.exe install transformers>=4.43 accelerate>=0.33 safetensors>=0.4 sentencepiece>=0.2
    .venv\Scripts\pip.exe install torch --index-url https://download.pytorch.org/whl/cu121
    echo [OK] Local model packages installed.
    echo.
)

echo ============================================================
echo  Setup complete!
echo  Run the app with:  run_app.bat
echo ============================================================
echo.
pause