@echo off
REM AI Glass System - Windows 快速安装脚本

echo ==========================================
echo   AI Glass System - 自动安装脚本
echo ==========================================
echo.

REM 检查 Python
echo 正在检查 Python...
python --version >nul 2>&1
if errorlevel 1 (
    echo [错误] 未找到 Python
    echo 请从 https://www.python.org/downloads/ 下载并安装 Python 3.9-3.11
    pause
    exit /b 1
)

python --version
echo [成功] Python 已安装

REM 检查 CUDA
echo.
echo 正在检查 CUDA...
nvidia-smi >nul 2>&1
if errorlevel 1 (
    echo [警告] 未检测到 NVIDIA GPU，将使用 CPU 模式（速度较慢）
    set HAS_GPU=0
) else (
    echo [成功] 检测到 NVIDIA GPU
    nvidia-smi --query-gpu=name,driver_version,memory.total --format=csv,noheader
    set HAS_GPU=1
)

REM 创建虚拟环境
echo.
echo 正在创建虚拟环境...
if exist venv (
    echo [警告] 虚拟环境已存在
    set /p RECREATE="是否删除并重新创建? (y/n): "
    if /i "%RECREATE%"=="y" (
        rmdir /s /q venv
        python -m venv venv
        echo [成功] 虚拟环境已重新创建
    )
) else (
    python -m venv venv
    echo [成功] 虚拟环境已创建
)

REM 激活虚拟环境
echo.
echo 正在激活虚拟环境...
call venv\Scripts\activate.bat

REM 升级 pip
echo.
echo 正在升级 pip...
python -m pip install --upgrade pip -q
echo [成功] pip 已升级

REM 安装 PyTorch
echo.
echo 正在安装 PyTorch...
if %HAS_GPU%==1 (
    echo 安装 GPU 版本 PyTorch ^(CUDA 11.8^)...
    pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 --index-url https://download.pytorch.org/whl/cu118 -q
) else (
    echo 安装 CPU 版本 PyTorch...
    pip install torch torchvision -q
)
echo [成功] PyTorch 已安装

REM 验证 PyTorch
echo.
echo 验证 PyTorch 安装...
python -c "import torch; print(f'PyTorch 版本: {torch.__version__}'); print(f'CUDA 可用: {torch.cuda.is_available()}')"

REM 安装 PyAudio
echo.
echo 正在安装 PyAudio...
echo [警告] PyAudio 在 Windows 上可能需要手动安装
echo 如果自动安装失败，请从以下地址下载 wheel 文件:
echo https://www.lfd.uci.edu/~gohlke/pythonlibs/#pyaudio
echo.
pip install pyaudio -q
if errorlevel 1 (
    echo [警告] PyAudio 自动安装失败，请手动安装
) else (
    echo [成功] PyAudio 已安装
)

REM 安装其他依赖
echo.
echo 正在安装 Python 依赖...
pip install -r requirements.txt -q
echo [成功] Python 依赖已安装

REM 创建 .env 文件
echo.
if not exist .env (
    echo 正在创建 .env 配置文件...
    copy .env.example .env >nul
    echo [成功] .env 文件已创建
    echo [提示] 请编辑 .env 文件，填入您的 DASHSCOPE_API_KEY
) else (
    echo [跳过] .env 文件已存在
)

REM 创建必要的目录
echo.
echo 正在创建目录结构...
if not exist recordings mkdir recordings
if not exist model mkdir model
if not exist music mkdir music
if not exist voice mkdir voice
echo [成功] 目录结构已创建

REM 检查模型文件
echo.
echo 正在检查模型文件...
set MISSING=0
if exist model\yolo-seg.pt (echo [成功] yolo-seg.pt) else (echo [缺失] yolo-seg.pt & set MISSING=1)
if exist model\yoloe-11l-seg.pt (echo [成功] yoloe-11l-seg.pt) else (echo [缺失] yoloe-11l-seg.pt & set MISSING=1)
if exist model\shoppingbest5.pt (echo [成功] shoppingbest5.pt) else (echo [缺失] shoppingbest5.pt & set MISSING=1)
if exist model\trafficlight.pt (echo [成功] trafficlight.pt) else (echo [缺失] trafficlight.pt & set MISSING=1)
if exist model\hand_landmarker.task (echo [成功] hand_landmarker.task) else (echo [缺失] hand_landmarker.task & set MISSING=1)

if %MISSING%==1 (
    echo.
    echo [警告] 部分模型文件缺失，请将模型文件放入 model\ 目录
)

REM 完成
echo.
echo ==========================================
echo [成功] 安装完成!
echo ==========================================
echo.
echo 下一步:
echo 1. 编辑 .env 文件，填入您的 API 密钥:
echo    notepad .env
echo.
echo 2. 确保所有模型文件已放入 model\ 目录
echo.
echo 3. 启动系统:
echo    venv\Scripts\activate
echo    python app_main.py
echo.
echo 4. 访问 http://localhost:8081
echo.
echo [提示] 每次使用前请激活虚拟环境:
echo   venv\Scripts\activate
echo.

pause