backend only code

webui/app.py

@@ -16,6 +16,19 @@ from fastapi.templating import Jinja2Templates
 from webui.runner import ModelRunner
 from webui.weights import get_weights_dir
 
+from fastapi.middleware.cors import CORSMiddleware
+
+app = FastAPI()
+
+# CORS for local frontend
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["http://localhost:8000", "http://127.0.0.1:8000", "null"],
+    allow_credentials=False,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
 PROJECT_ROOT = Path(__file__).resolve().parents[1]  # repo root
 WEBUI_DIR = Path(__file__).resolve().parent
 UPLOAD_DIR = WEBUI_DIR / "uploads"
@@ -23,12 +36,12 @@ RESULT_DIR = WEBUI_DIR / "results"
 UPLOAD_DIR.mkdir(parents=True, exist_ok=True)
 RESULT_DIR.mkdir(parents=True, exist_ok=True)
 
-app = FastAPI()
-templates = Jinja2Templates(directory=str(WEBUI_DIR / "templates"))
-
-app.mount("/static", StaticFiles(directory=str(WEBUI_DIR / "static")), name="static")
 app.mount("/results", StaticFiles(directory=str(RESULT_DIR)), name="results")
 
+@app.get("/health")
+def health():
+    return {"ok": True}
+
 # ---- weights repo ----
 WEIGHTS_REPO = os.getenv("TASKCLIP_WEIGHTS_REPO", "BiasLab2025/taskclip-weights")
 WEIGHTS_DIR = get_weights_dir(WEIGHTS_REPO)