from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware

from routers.Bot import botRouter
from routers.Chat import chatRouter
from routers.Datasource import reportDataRouter
from routers.F8 import f8Router
from routers.Knowledge import knowledgeRouter
from routers.Public import publicRouter
from routers.RabbitMQ import rqRouter
from routers.Report import reportRouter
from routers.Service import serviceRouter
from routers.Vision import visionRouter
from service.Analyze import mq_pull_analysis
from utils import MyUtils

app = FastAPI(title="BBIT_AI")

origins = [
    "http://localhost:8090",  # Vite dev 默认端口
    "http://127.0.0.1:5173",
    "http://s1.ronsunny.cn:8089",
    "*",  # ⚠️ 生产环境不要用
]

app.add_middleware(
    CORSMiddleware,
    allow_origins=origins,
    allow_credentials=True,
    allow_methods=["*"],  # 必须包含 OPTIONS、GET 等
    allow_headers=["*"],
)
routers = [
    chatRouter,
    reportRouter,
    knowledgeRouter,
    reportDataRouter,
    serviceRouter,
    botRouter,
    visionRouter,
    rqRouter,
]

for r in routers:
    app.include_router(r, prefix="/llm", tags=["llm"])

app.include_router(f8Router, prefix="/f8", tags=["f8"])

app.include_router(publicRouter, prefix="/api/public", tags=["api"])

MyUtils.async_new_task(mq_pull_analysis)