@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.

:: Set explicit venv python path (avoids conflict with global Python 3.13)
set "PYEXE=%~dp0venv\Scripts\python.exe"

:: [1/4] Check virtual environment
echo [1/4] Checking virtual environment...
if not exist "%PYEXE%" (
    echo [ERROR] venv not found. Run: python -m venv venv
    pause
    exit /b 1
)
echo [OK] venv found.

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

:: [3/4] Apply huggingface_hub compatibility patch
echo [3/4] Applying compatibility patch...
if exist "%~dp0scripts\patch_timesfm.py" (
    "%PYEXE%" "%~dp0scripts\patch_timesfm.py" >nul 2>&1
    echo [OK] Patch applied.
) else (
    echo [WARN] Patch script not found, skipping.
)

:: [4/4] Launch server
echo [4/4] Launching server...
echo.
echo [INFO] URL: http://127.0.0.1:8000
echo [INFO] Browser opens automatically in ~3 seconds.
echo [INFO] Press CTRL+C to stop.
echo.

"%PYEXE%" -m backend.launcher

if %ERRORLEVEL% neq 0 (
    echo.
    echo [ERROR] Server exited with error. See log above.
    pause
)