update main.py STATIC FILES (SPA SUPPORT)

main.py

@@ -1188,24 +1188,23 @@ async def health_check():
 
 # ================ STATIC FILES (SPA SUPPORT) ================
 
-# Resolve static dir from this file so it works regardless of cwd
-_STATIC_DIR = Path(__file__).resolve().parent / "static"
-_ASSETS_DIR = _STATIC_DIR / "assets"
+FRONTEND_DIR = "dist" if os.path.exists("dist/index.html") else "static"
 
-# 1. Mount the assets folder (JS/CSS) first so /assets/* is never caught by catch-all
-if _ASSETS_DIR.is_dir():
-    app.mount("/assets", StaticFiles(directory=str(_ASSETS_DIR)), name="assets")
+assets_path = os.path.join(FRONTEND_DIR, "assets")
+if os.path.exists(assets_path):
+    app.mount("/assets", StaticFiles(directory=assets_path), name="assets")
 
-# 2. Catch-all for SPA: serve index.html for app routes, never for /assets (would break JS MIME type)
 @app.get("/{full_path:path}")
 async def serve_react_app(full_path: str, request: Request):
     if full_path.startswith("api") or full_path.startswith("ws"):
         raise HTTPException(status_code=404, detail="Not Found")
-    # Don't serve HTML for asset paths; let them 404 so we don't break module script loading
-    if full_path.startswith("assets") or full_path.startswith("assets/"):
-        raise HTTPException(status_code=404, detail="Not Found")
-
-    index_path = _STATIC_DIR / "index.html"
-    if index_path.is_file():
-        return FileResponse(str(index_path))
-    return {"message": "React app not found. Please run 'npm run build' and copy dist to static."}
+    
+    file_path = os.path.join(FRONTEND_DIR, full_path)
+    if os.path.isfile(file_path):
+        return FileResponse(file_path)
+    
+    index_path = os.path.join(FRONTEND_DIR, "index.html")
+    if os.path.exists(index_path):
+        return FileResponse(index_path)
+    else:
+        return {"message": "React app not found. Please run npm run build."}