@echo off
TITLE Kronos AI Trading Terminal
COLOR 0B
chcp 65001 >nul 2>&1

echo.
echo ╔══════════════════════════════════════════════════════╗
echo ║     KRONOS AI TRADING TERMINAL — STARTUP             ║
echo ║     Powered by Google TimesFM 2.5 (200M params)      ║
echo ╚══════════════════════════════════════════════════════╝
echo.

:: ── [1/4] Check virtual environment ─────────────────────────────────────────
echo [1/4] Checking virtual environment...
if not exist "venv\Scripts\python.exe" (
    echo [ERROR] Virtual environment not found.
    echo         Run: python -m venv venv
    echo         Then: venv\Scripts\pip install -r requirements.txt
    pause
    exit /b 1
)

:: ── [2/4] Activate venv ──────────────────────────────────────────────────────
echo [2/4] Activating virtual environment...
call venv\Scripts\activate

:: ── [3/4] Ensure TimesFM is installed ───────────────────────────────────────
echo [3/4] Checking TimesFM dependency...
venv\Scripts\python.exe -c "import timesfm; timesfm.TimesFM_2p5_200M_torch" >nul 2>&1
if %ERRORLEVEL% neq 0 (
    echo [INFO] TimesFM not found. Installing from local clone...
    if exist "libs\timesfm\pyproject.toml" (
        echo [INFO] Installing from libs\timesfm ...
        venv\Scripts\python.exe -m pip install -e "libs\timesfm[torch]" --quiet
        if %ERRORLEVEL% neq 0 (
            echo [ERROR] TimesFM install failed from libs\timesfm.
            pause
            exit /b 1
        )
    ) else (
        echo [INFO] Cloning TimesFM from GitHub...
        git clone --depth=1 https://github.com/google-research/timesfm.git libs\timesfm
        if %ERRORLEVEL% neq 0 (
            echo [ERROR] git clone failed. Check your internet connection.
            pause
            exit /b 1
        )
        echo [INFO] Installing TimesFM...
        venv\Scripts\python.exe -m pip install -e "libs\timesfm[torch]" --quiet
        if %ERRORLEVEL% neq 0 (
            echo [ERROR] TimesFM install failed.
            pause
            exit /b 1
        )
    )
    echo [OK] TimesFM installed successfully.
) else (
    echo [OK] TimesFM is ready.
)

:: ── [4/4] Launch server ──────────────────────────────────────────────────────
echo [4/4] Launching Kronos AI Trading Terminal...
echo.
echo [INFO] Server will start at http://127.0.0.1:8000
echo [INFO] Browser will open automatically in ~3 seconds.
echo [INFO] TimesFM model downloads on FIRST forecast request (~800MB).
echo [INFO] Press CTRL+C to stop the server.
echo.

python -m backend.launcher

if %ERRORLEVEL% neq 0 (
    echo.
    echo [ERROR] Server exited with an error.
    echo         Check the log above for details.
    pause
)