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

app = FastAPI()

# Add CORS middleware to allow SSE connections from nginx UI
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],  # In production, specify exact origins
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

app.include_router(cortex_router)