@echo off
chcp 65001 >nul
setlocal enabledelayedexpansion

echo ============================================================
echo 信用卡欺诈检测系统 - 一键安装脚本
echo ============================================================
echo.

REM 检查Python是否安装
python --version >nul 2>&1
if %errorlevel% neq 0 (
    echo [错误] 未检测到Python，请先安装Python 3.10或更高版本
    echo 下载地址: https://www.python.org/downloads/
    pause
    exit /b 1
)

echo [步骤 1/5] 检查Python版本...
for /f "tokens=2" %%i in ('python --version 2^>^&1') do set PYTHON_VERSION=%%i
echo Python版本: %PYTHON_VERSION%
echo.

REM 检查Python版本是否满足要求
for /f "tokens=1,2 delims=." %%a in ("%PYTHON_VERSION%") do (
    set MAJOR=%%a
    set MINOR=%%b
)

if %MAJOR% lss 3 (
    echo [错误] Python版本过低，需要3.10或更高版本
    pause
    exit /b 1
)

if %MAJOR% equ 3 (
    if %MINOR% lss 10 (
        echo [错误] Python版本过低，需要3.10或更高版本
        pause
        exit /b 1
    )
)

echo [步骤 2/5] 安装Python依赖...
echo 正在安装依赖包，这可能需要几分钟...
pip install -r requirements.txt
if %errorlevel% neq 0 (
    echo [错误] 依赖安装失败
    echo 请检查网络连接或尝试使用国内镜像源:
    echo pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
    pause
    exit /b 1
)
echo 依赖安装完成
echo.

echo [步骤 3/5] 检查数据文件...
if not exist "data\creditcard.csv" (
    echo [警告] 未找到数据文件 data\creditcard.csv
    echo.
    echo 请从以下地址下载数据集:
    echo https://www.kaggle.com/datasets/mlg-ulb/creditcardfraud
    echo.
    echo 下载后将 creditcard.csv 文件放入 data\ 目录
    echo.
    set /p DATA_READY="数据文件已准备好吗？(Y/N): "
    if /i not "!DATA_READY!"=="Y" (
        echo 安装已取消
        pause
        exit /b 1
    )
) else (
    echo 数据文件已存在
)
echo.

echo [步骤 4/5] 检查模型文件...
if not exist "models\random_forest_model.joblib" (
    echo 模型文件不存在，开始训练模型...
    echo 这可能需要几分钟，请耐心等待...
    python src\train.py
    if %errorlevel% neq 0 (
        echo [错误] 模型训练失败
        pause
        exit /b 1
    )
    echo 模型训练完成
) else (
    echo 模型文件已存在
)
echo.

echo [步骤 5/5] 运行环境检查...
python check_environment.py
if %errorlevel% neq 0 (
    echo [警告] 环境检查发现问题，但将继续启动应用
)
echo.

echo ============================================================
echo 安装完成！正在启动Web界面...
echo ============================================================
echo.
echo 提示:
echo - Web界面将在浏览器中自动打开
echo - 如果没有自动打开，请访问: http://localhost:8501
echo - 按 Ctrl+C 可以停止服务
echo.

python src\agent_app.py

pause