@echo off
echo ==========================================
echo  Local Swarm - Windows Installer
echo ==========================================
echo.

REM Check Python
python --version >nul 2>&1
if errorlevel 1 (
    echo [ERROR] Python is not installed or not in PATH
    echo Please install Python 3.9+ from https://python.org
    exit /b 1
)

echo [1/4] Checking Python version...
for /f "tokens=2" %%a in ('python --version') do set PYTHON_VERSION=%%a
echo       Found Python %PYTHON_VERSION%

echo.
echo [2/4] Upgrading pip...
python -m pip install --upgrade pip

echo.
echo [3/4] Installing base dependencies...
pip install -r requirements.txt

REM Check for CUDA
nvidia-smi >nul 2>&1
if %errorlevel% == 0 (
    echo.
    echo [4/4] CUDA detected! Installing GPU-accelerated llama.cpp...
    pip uninstall -y llama-cpp-python
    pip install llama-cpp-python --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu121
    echo       GPU support enabled!
) else (
    echo.
    echo [4/4] No CUDA detected, using CPU backend...
    echo       CPU-only mode (slower but works on any hardware)
)

echo.
echo ==========================================
echo  Installation Complete!
echo ==========================================
echo.
echo To start Local Swarm:
echo   python main.py
echo.
echo To check hardware detection:
echo   python main.py --detect
echo.
echo For more options:
echo   python main.py --help
echo.
pause