@echo off
echo Recompiling LightRAG WebUI...
echo.

:: Check if Python is available
python --version >nul 2>&1
if errorlevel 1 (
    echo ERROR: Python is not available in PATH
    echo Please ensure Python is installed and added to system PATH
    pause
    exit /b 1
)

:: Install/update dependencies
echo Step 1: Installing/updating dependencies...
pip install -r requirements.txt

:: Install spaCy model if not present
echo.
echo Step 2: Checking spaCy model...
python -c "import spacy; spacy.load('en_core_web_sm')" >nul 2>&1
if errorlevel 1 (
    echo Installing spaCy English model...
    python -m spacy download en_core_web_sm
) else (
    echo spaCy model already installed
)

:: Verify GPU support
echo.
echo Step 3: Verifying GPU support...
python verify_gpu_support.py

:: Test database connections
echo.
echo Step 4: Testing database connections...
python test_database_connections.py

:: Run quick system test
echo.
echo Step 5: Running system validation...
python simple_production_test.py

echo.
echo ========================================
echo WebUI compilation and setup complete!
echo.
echo To start the server, run: zrun.bat
echo WebUI will be available at: http://localhost:3015
echo ========================================
echo.

pause