Update backend/app/main.py

backend/app/main.py

@@ -4,23 +4,40 @@ from fastapi.middleware.cors import CORSMiddleware
 from backend.app.db import init_db
 from backend.app.api.health import router as health_router
 from backend.app.api.datasets import router as datasets_router
+from backend.app.api.experiments import router as experiments_router
+from backend.app.api.compare import router as compare_router
+from backend.app.api.exports import router as exports_router
+from backend.app.api.presets import router as presets_router
+
+app = FastAPI(title="ClusterBuster API")
 
-app = FastAPI()
 
 @app.on_event("startup")
-def startup():
+def on_startup() -> None:
     init_db()
 
+
+origins = [
+    "http://localhost:3000",
+    "https://cluster-buster.vercel.app",
+]
+
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],
+    allow_origins=origins,
+    allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
 )
 
 app.include_router(health_router, prefix="/api")
 app.include_router(datasets_router, prefix="/api")
+app.include_router(experiments_router, prefix="/api")
+app.include_router(compare_router, prefix="/api")
+app.include_router(exports_router, prefix="/api")
+app.include_router(presets_router, prefix="/api")
+
 
 @app.get("/")
 def root():
-    return {"ok": True}
+    return {"ok": True, "service": "clusterbuster-api"}