@echo off
chcp 65001 >nul
REM ============================================
REM  Windows 环境安装脚本
REM  安装所有依赖和模型
REM ============================================

echo ==========================================
echo   加密货币预测系统 - Windows 安装脚本
echo ==========================================
echo.

REM 检查 conda
where conda >nul 2>&1
if %errorlevel% neq 0 (
    echo [错误] 未找到 conda
    echo 请先安装 Miniconda: https://docs.conda.io/en/latest/miniconda.html
    pause
    exit /b 1
)

echo [1/6] 创建 conda 环境...
call conda create -n crypto_pred_312 python=3.12 -y
if %errorlevel% neq 0 (
    echo [警告] 环境可能已存在，继续...
)

echo.
echo [2/6] 激活环境...
call conda activate crypto_pred_312

echo.
echo [3/6] 安装 PyTorch (CPU 版本)...
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

echo.
echo [4/6] 安装项目依赖...
pip install -r requirements.txt

echo.
echo [5/6] 安装 IBM TTM-R2 模型库...
echo 注意: tsfm-public 包名使用连字符
pip install tsfm-public

echo.
echo [6/6] 验证安装...
python -c "import torch; print(f'PyTorch: {torch.__version__}')"
python -c "import tsfm_public; print('tsfm_public: 已安装')"
python -c "from training.config import TrainingConfig; c=TrainingConfig(); print(f'设备: {c.get_device()}')"

echo.
echo ==========================================
echo   安装完成!
echo ==========================================
echo.
echo 下一步:
echo   1. 复制 .env.example 为 .env 并配置数据库
echo   2. 双击 start_system.bat 启动系统
echo   3. 访问 http://localhost:8000
echo.

pause