from __future__ import annotations

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

from storage.media_routers import router as media_router
from storage.db_routers import router as db_router
from storage.embeddings_routers import router as embeddings_router
from storage.pending_videos_routers import router as pending_videos_router
from main_process.main_router import router as main_router
from main_process.salamandra_router import router as salamandra_router
from main_process.moe_router import router as moe_router
from main_process.refinement_router import router as refinement_router
from storage.data_routers import router as data_router
from preprocessing_router import router as preprocessing_router


app = FastAPI(title="Veureu Engine API", version="0.2.0")

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

app.include_router(data_router)
app.include_router(main_router)
app.include_router(salamandra_router)
app.include_router(moe_router)
app.include_router(refinement_router)
app.include_router(media_router)
app.include_router(db_router)
app.include_router(embeddings_router)
app.include_router(pending_videos_router)
app.include_router(preprocessing_router, prefix="/preprocessing")


@app.get("/")
def root():
    return {"ok": True, "service": "veureu-engine"}