|
|
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"}
|
|
|
|
