from fastapi import FastAPI
from app.routes.predict import router as predict_router
from app.routes.chat import router as chat_router
from fastapi.middleware.cors import CORSMiddleware


app = FastAPI(
    title="NeuroFlora API"
)

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

app.include_router(predict_router)
app.include_router(chat_router)

@app.get("/")

async def root():

    return {
        "message": "NeuroFlora API Running"
    }