# app/main.py
from fastapi import FastAPI
from contextlib import asynccontextmanager
from .core.model_loader import load_models, unload_models
from app.api.card_inference import router as card_inference_router

from .core.config import settings


@asynccontextmanager
async def lifespan(app: FastAPI):
    print("--- 应用启动 ---")
    load_models()
    yield

    print("--- 应用关闭 ---")
    unload_models()


app = FastAPI(title="卡片框和缺陷检测服务", lifespan=lifespan)

app.include_router(card_inference_router, prefix=settings.API_prefix)