Update app.py

app.py

@@ -114,18 +114,24 @@ if static_path.exists() and static_path.is_dir():
             return FileResponse(str(css_path), media_type="text/css")
         return {"error": "CSS file not found"}, 404
     
+
     # Catch-all route for Angular routing (must be last!)
     @app.get("/{full_path:path}")
     async def serve_angular(full_path: str):
         # Don't catch API routes
-        if full_path.startswith("api/") or full_path in ["start_session", "chat", "health"]:
-            return {"error": "Not found"}, 404
+        if full_path.startswith("api/") or full_path.startswith("health"):
+            from fastapi import HTTPException
+            raise HTTPException(status_code=404, detail="Not found")
         
-        # Return Angular index.html for all other routes
+        # For Angular routes, always return index.html
         index_path = static_path / "index.html"
         if index_path.exists():
+            log(f"🔀 Serving index.html for Angular route: /{full_path}")
             return FileResponse(str(index_path), media_type="text/html")
-        return {"error": "UI not found"}, 404
+        
+        # If no index.html, return 404
+        from fastapi import HTTPException
+        raise HTTPException(status_code=404, detail="UI not found")
 else:
     log("⚠️ Static directory not found")
     # No UI built, just health endpoint