@echo off
chcp 65001 >nul
cd /d "%~dp0"

echo ========================================
echo   道路损伤检测系统 - 环境安装脚本
echo ========================================
echo.

echo 正在检查Python环境...
python --version >nul 2>&1
if errorlevel 1 (
    echo 错误: 未找到Python环境
    echo 请先安装Python 3.8或更高版本
    echo 下载地址: https://www.python.org/downloads/
    pause
    exit /b 1
)

echo ✓ Python环境检查通过
python --version
echo.

echo 正在检查pip...
pip --version >nul 2>&1
if errorlevel 1 (
    echo 错误: pip未找到
    echo 请重新安装Python并确保包含pip
    pause
    exit /b 1
)

echo ✓ pip检查通过
echo.

echo 正在升级pip...
python -m pip install --upgrade pip
echo.

echo 正在安装项目依赖...
echo 这可能需要几分钟时间，请耐心等待...
echo.

pip install -r requirements.txt

if errorlevel 1 (
    echo.
    echo 安装过程中出现错误，尝试使用国内镜像源...
    echo.
    pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple/
)

echo.
echo 正在创建必要的文件夹...
if not exist "input_videos" mkdir "input_videos"
if not exist "output_frames" mkdir "output_frames"
if not exist "logs" mkdir "logs"

echo ✓ 文件夹创建完成
echo.

echo 正在测试安装...
python -c "import cv2, numpy, ultralytics; print('✓ 所有依赖库安装成功')"

if errorlevel 1 (
    echo ✗ 依赖库测试失败
    echo 请检查安装过程中的错误信息
    pause
    exit /b 1
)

echo.
echo ========================================
echo           安装完成!
echo ========================================
echo.
echo 使用说明:
echo 1. 将视频文件放入 input_videos 文件夹
echo 2. 双击 start_detection.bat 开始检测
echo 3. 或者运行: python run_detection.py
echo.
echo 详细说明请查看 README.md 文件
echo.
pause