@echo off
title BiRefNet Matting - 服务器端 (Server)
chcp 65001 >nul
cls
echo.
echo ============================================================
echo.
echo        ╔═══════════════════════════════════════╗
echo        ║   BiRefNet Matting - 服务器端         ║
echo        ║   FastAPI Image Processing Server     ║
echo        ╚═══════════════════════════════════════╝
echo.
echo ============================================================
echo.
echo 正在启动 FastAPI 服务器...
echo.

cd /d "%~dp0"

REM 检查 Python 是否安装
python --version >nul 2>&1
if errorlevel 1 (
    echo 错误: 未检测到 Python，请先安装 Python 3.8 或更高版本
    echo.
    echo 按任意键退出...
    pause >nul
    exit /b 1
)

REM 检查虚拟环境是否存在
if exist ".venv" (
    echo 激活虚拟环境...
    call .venv\Scripts\activate.bat
    
    REM 检查 FastAPI 是否已安装
    echo 检查 FastAPI 依赖...
    python -c "import fastapi" >nul 2>&1
    if errorlevel 1 (
        echo 检测到 FastAPI 依赖未安装，正在安装...
        echo.
        pip install -r requirements_python.txt
        if errorlevel 1 (
            echo FastAPI 依赖安装失败，请检查网络连接
            echo.
            echo 按任意键退出...
            pause >nul
            exit /b 1
        ) else (
            echo FastAPI 依赖安装完成！
        )
        echo.
    )
) else (
    echo 创建虚拟环境...
    python -m venv .venv
    call .venv\Scripts\activate.bat
    echo.
    echo 正在安装依赖...
    pip install -r requirements_python.txt
    if errorlevel 1 (
        echo 依赖安装失败，请检查网络连接
        echo.
        echo 按任意键退出...
        pause >nul
        exit /b 1
    )
    echo 依赖安装完成！
    echo.
)

echo.
echo ============================================================
echo                   服务器启动信息
echo ============================================================
echo  服务器地址: http://localhost:8000
echo  API 文档:   http://localhost:8000/docs
echo  处理接口:   POST /api/process
echo ============================================================
echo.
echo  按 Ctrl+C 停止服务器
echo.

python server.py

echo.
echo 服务器已停止
echo 按任意键退出...
pause >nul