3d7bd70355
- 新增 Alembic 初始迁移、领域明细模型及归档持久化与重试链路\n- 拆分视频会话注册表、分类处理、推理时间窗聚合与流处理\n- 消耗日志：TSV/Markdown 含 top2/top3；item_id 优先产品编码；待确认记「待确认」行，语音确认后落正式行并更新汇总\n- 待确认时内存/DB 明细为占位行，确认后替换；拒绝时移除占位\n- 分类 probs 先 detach/cpu 再转 NumPy，修复 MPS/CUDA 上推理被静默跳过\n- 补充集成测试、归档与设备张量等单测 Made-with: Cursor
98 lines
2.8 KiB
Python
98 lines
2.8 KiB
Python
import sys
|
||
from contextlib import asynccontextmanager
|
||
|
||
import uvicorn
|
||
from fastapi import FastAPI
|
||
from fastapi.middleware.cors import CORSMiddleware
|
||
from loguru import logger
|
||
|
||
from app.api import router as api_router
|
||
from app.config import settings
|
||
from app.database import check_database, engine, init_db_schema
|
||
from app.dependencies import build_container
|
||
|
||
|
||
def configure_logging() -> None:
|
||
"""集中配置 loguru sink；由 create_app 显式调用，避免 import-time 副作用。"""
|
||
logger.remove()
|
||
logger.add(
|
||
sys.stderr,
|
||
format=(
|
||
"<green>{time:YYYY-MM-DD HH:mm:ss}</green> | "
|
||
"<level>{level: <8}</level> | "
|
||
"<cyan>{name}</cyan>:<cyan>{function}</cyan> - <level>{message}</level>"
|
||
),
|
||
)
|
||
|
||
|
||
@asynccontextmanager
|
||
async def lifespan(app: FastAPI):
|
||
await check_database()
|
||
if settings.auto_create_schema:
|
||
await init_db_schema()
|
||
logger.info("Database connection verified; schema auto-created (dev mode)")
|
||
else:
|
||
logger.info(
|
||
"Database connection verified; auto_create_schema=false, "
|
||
"expecting 'alembic upgrade head' to have run"
|
||
)
|
||
container = build_container(settings)
|
||
app.state.container = container
|
||
await container.start()
|
||
try:
|
||
yield
|
||
finally:
|
||
await container.shutdown()
|
||
await engine.dispose()
|
||
logger.info("Database engine disposed")
|
||
|
||
|
||
def create_app() -> FastAPI:
|
||
configure_logging()
|
||
application = FastAPI(
|
||
title="Operation Room Monitor",
|
||
lifespan=lifespan,
|
||
)
|
||
if settings.demo_cors_enabled:
|
||
origins = settings.parsed_demo_cors_origins()
|
||
if origins:
|
||
application.add_middleware(
|
||
CORSMiddleware,
|
||
allow_origins=origins,
|
||
allow_credentials=origins != ["*"],
|
||
allow_methods=["*"],
|
||
allow_headers=["*"],
|
||
)
|
||
logger.info("CORS enabled for demo client; origins={}", origins)
|
||
application.include_router(api_router)
|
||
if settings.demo_orchestrator_enabled:
|
||
from app.routers import demo_orch
|
||
|
||
application.include_router(demo_orch.router)
|
||
logger.info(
|
||
"Demo orchestrator enabled: POST /internal/demo/orchestrate-and-start",
|
||
)
|
||
else:
|
||
logger.info(
|
||
"Demo orchestrator disabled (DEMO_ORCHESTRATOR_ENABLED=false): "
|
||
"GET /internal/demo/orchestrator-status for status; "
|
||
"POST /internal/demo/orchestrate-and-start is not registered",
|
||
)
|
||
return application
|
||
|
||
|
||
app = create_app()
|
||
|
||
|
||
def main() -> None:
|
||
uvicorn.run(
|
||
"main:app",
|
||
host=settings.server_host,
|
||
port=settings.server_port,
|
||
reload=settings.server_reload,
|
||
)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|