from api import recommend_router, report_router, eval_report_router
from fastapi import FastAPI, Request, status
from fastapi.exceptions import RequestValidationError
from fastapi.responses import JSONResponse

import uvicorn

app = FastAPI()

# 添加全局异常处理器
@app.exception_handler(RequestValidationError)
async def validation_exception_handler(request: Request, exc: RequestValidationError):
    return JSONResponse(
        status_code=status.HTTP_400_BAD_REQUEST,
        content={
            "code": 400,
            "msg": "请求参数错误",
            "data": {
                "detail": exc.errors(),
                "body": exc.body
            }
        },
    )

url_prefix = '/brandcultivation/api/v1'
  
# 注册路由
app.include_router(recommend_router, prefix=url_prefix)
app.include_router(report_router, prefix=url_prefix)
app.include_router(eval_report_router, prefix=url_prefix)
    
if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0", port=7960)
    # report_dir = "./data/reports/00000000000000000000000011445301/440298"
    # upload_file(report_dir)