Update server.py

server.py

@@ -1,53 +1,53 @@
-# app.py
+import logging
 from fastapi import FastAPI, UploadFile, File
 from fastapi.responses import HTMLResponse, JSONResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.middleware.cors import CORSMiddleware
 import base64
+import mmcv
+
 from run_inference import infer_images
 
+# ─── Configure logging ─────────────────────────────────────────────────────────
+logging.basicConfig(level=logging.INFO)
+
 app = FastAPI()
 
-# 1) CORS: allow requests from your Space’s frontend origin
+# ─── CORS: allow only your Space’s hf.space domain ─────────────────────────────
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["https://huggingface.co/spaces/yaghi27/regnet-detr3d"],
+    allow_origins=["https://yaghi27-regnet-detr3d.hf.space"],
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
 )
 
-# 2) Mount a `static/` folder so that any JS/CSS referenced in index.html will load
+# ─── (Optional) serve any JS/CSS under ./static/ ────────────────────────────────
 app.mount("/static", StaticFiles(directory="static"), name="static")
 
 
 @app.get("/", response_class=HTMLResponse)
 async def root():
-    # serve the HTML page
     with open("index.html", "r", encoding="utf-8") as f:
         return HTMLResponse(f.read())
 
+
 @app.post("/infer")
 async def run_inference(images: list[UploadFile] = File(...)):
     img_paths = []
     for upload in images:
         data = await upload.read()
-        # force 3-channel BGR
+        # Drop any alpha channel, force 3-channel BGR
         bgr = mmcv.imfrombytes(data, flag="color")
         tmp = f"/tmp/{upload.filename}"
         mmcv.imwrite(bgr, tmp)
         img_paths.append(tmp)
 
-    # wrap the inference in try/except
     try:
         bev_paths = infer_images(img_paths)
     except Exception as e:
         logging.exception("❌ inference failed")
-        # return the error message to the client for debugging
-        return JSONResponse(
-            status_code=500,
-            content={"error": str(e)}
-        )
+        return JSONResponse(status_code=500, content={"error": str(e)})
 
     output = []
     for p in bev_paths: