start.bat

@echo off
chcp 65001 >nul
setlocal enabledelayedexpansion
title AI Chat - Uvicorn 启动器

echo ============================================
echo   AI Chat 本地开发服务器启动脚本
echo ============================================
echo.

:: 检查 Python
where python >nul 2>&1
if %ERRORLEVEL% neq 0 (
    echo [错误] 未找到 Python，请先安装 Python 3.11+
    pause
    exit /b 1
)

:: 检查当前目录
if not exist "src\controllers\main.py" (
    echo [错误] 请在工作目录 c:\Users\root\Documents\Programs\aaa 下运行此脚本
    pause
    exit /b 1
)

:: 检查 config.yaml
if not exist "src\configs\config.yaml" (
    echo [提示] config.yaml 不存在，正在从示例文件创建...
    copy src\configs\config.yaml.example src\configs\config.yaml
    echo [完成] 请编辑 src\configs\config.yaml 填入你的 API Key 等配置
    echo.
)

:: 显示当前数据库 URL
echo 当前数据库配置：
findstr "database_url" src\configs\config.yaml
echo.

:: 询问是否修改数据库 URL
set /p CHANGE_DB="是否修改数据库连接地址？(y/n，默认 n): "
if /i "!CHANGE_DB!"=="y" (
    set /p NEW_DB_URL="请输入新的 database_url（例如 sqlite:///app.db）: "
    if defined NEW_DB_URL (
        echo 正在更新 database_url...
        powershell -Command "(Get-Content src\configs\config.yaml) -replace 'database_url:.*', 'database_url: !NEW_DB_URL!' | Set-Content src\configs\config.yaml"
        echo [完成] 数据库连接地址已更新
    ) else (
        echo [跳过] 未输入内容，保持原配置
    )
) else (
    echo [跳过] 使用当前配置
)

echo.

:: 运行数据库迁移
echo 正在运行数据库迁移...
cd src
alembic -c configs/alembic.ini upgrade head
if %ERRORLEVEL% neq 0 (
    echo [警告] 数据库迁移失败，请检查数据库连接
)
cd ..

echo.
echo ============================================
echo   启动 Uvicorn 开发服务器...
echo   API 地址: http://localhost:8000
echo   API 文档: http://localhost:8000/docs
echo   按 Ctrl+C 停止服务器
echo ============================================
echo.

:: 设置 Python 路径并启动
set PYTHONPATH=%PYTHONPATH%;%CD%\src
python -m uvicorn src.controllers.main:app --reload --host 0.0.0.0 --port 8000

pause