from fastapi import FastAPI
from .api import router as api_router

from .database import database, create_tables

app = FastAPI(
    title="人脸识别服务 V2 - PostgreSQL版",
    description="一个使用PostgreSQL和Milvus的高性能人脸识别服务。",
    version="2.1.0"
)

@app.on_event("startup")
async def on_startup():
    """应用启动时，连接数据库并创建表"""
    await database.connect()
    create_tables() # SQLAlchemy的create_all是同步的，所以这样调用没问题

@app.on_event("shutdown")
async def on_shutdown():
    """应用关闭时，断开数据库连接"""
    await database.disconnect()


# 挂载API路由
app.include_router(api_router, prefix="/api", tags=["Face Recognition"])

@app.get("/", summary="服务健康检查")
def read_root():
    return {"status": "ok", "message": "欢迎使用人脸识别服务 V2"}