@echo off
echo ============================================================
echo  LOGISTICS RL — SETUP SCRIPT
echo  Installing all required dependencies locally
echo ============================================================
echo.

SET SCRIPT_DIR=%~dp0
SET PY=C:\Users\Jerry\AppData\Local\Programs\Python\Python311\python.exe
SET VENV=%SCRIPT_DIR%.venv

IF NOT EXIST "%PY%" (
    echo ERROR: Python 3.11 not found at %PY%
    echo Please install Python 3.11 from https://www.python.org/downloads/
    pause
    exit /b 1
)

echo [1/3] Creating virtual environment...
"%PY%" -m venv "%VENV%"

echo [2/3] Installing packages (this may take 2-3 minutes)...
"%VENV%\Scripts\pip.exe" install --quiet fastapi "uvicorn[standard]" pydantic requests openenv-core

echo [3/3] Verifying install...
"%VENV%\Scripts\python.exe" -c "from server.environment import LogisticsShipmentEnvironment; env = LogisticsShipmentEnvironment(); obs = env.reset(); print('OK: Shipments loaded:', len(obs.shipments))"

IF %ERRORLEVEL% EQU 0 (
    echo.
    echo ============================================================
    echo  SETUP COMPLETE! 
    echo  Run start.bat to launch the server.
    echo ============================================================
) ELSE (
    echo.
    echo ERROR: Setup verification failed. Check the output above.
)

pause