from fastapi import FastAPI
from settings import settings
from api import router, service

app = FastAPI(
    title="Remove Background",
    description="图片去背景",
    version="1.0.0"
)

# 注册路由
app.include_router(router)


@app.on_event("startup")
async def startup_event():
    """应用启动时初始化队列批处理机制"""
    if settings.enable_queue_batch:
        await service._start_queue_processor()


@app.on_event("shutdown")
async def shutdown_event():
    """应用关闭时清理资源"""
    await service.cleanup()

if __name__ == "__main__":
    import uvicorn
    uvicorn.run(
        "app:app",
        host=settings.host,
        port=settings.port,
        reload=settings.debug
    )