Update backend/app/main.py

backend/app/main.py

@@ -1,22 +1,22 @@
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 
-from backend.app.api.clusters import router as clusters_router
-from backend.app.api.compare import router as compare_router
+from backend.app.api.health import router as health_router
 from backend.app.api.datasets import router as datasets_router
-from backend.app.api.exports import router as exports_router
 from backend.app.api.experiments import router as experiments_router
-from backend.app.api.health import router as health_router
-from backend.app.api.presets import router as presets_router
-from backend.app.db import Base, engine
+from backend.app.api.compare import router as compare_router
+from backend.app.api.exports import router as exports_router
 
-Base.metadata.create_all(bind=engine)
+app = FastAPI(title="ClusterBuster API")
 
-app = FastAPI(title="ClusterForge API", version="1.0.0")
+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=["*"],
@@ -25,11 +25,10 @@ app.add_middleware(
 app.include_router(health_router, prefix="/api")
 app.include_router(datasets_router, prefix="/api")
 app.include_router(experiments_router, prefix="/api")
-app.include_router(clusters_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, "service": "clusterforge-api", "docs": "/docs"}
+    return {"ok": True, "service": "clusterbuster-api"}