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

from api.routes.chat import router as chat_router
from api.routes.image import router as image_router
from api.routes.rag import router as rag_router


app = FastAPI(title="Kora HF")

app.add_middleware(
    CORSMiddleware,
    allow_origins=["https://kora.ai"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

app.include_router(chat_router, prefix="/chat", tags=["Chat"])
app.include_router(image_router, prefix="/image", tags=["Image"])
app.include_router(rag_router, prefix="/rag", tags=["RAG"])



@app.get("/")
async def health_check():
    return {
        "status": "online",
        "service": "Kora HF"
    }