Claude Code: The error at line 758 suggests

app.py

@@ -753,34 +753,36 @@ def create_agent_office_with_ws():
     """
     app = create_agent_office()
 
-    # Add a custom route for polling agent thoughts
-    # This is a simple polling endpoint; for true WebSocket, use websockets library
-    @app.get("/api/thoughts")
+    # Add custom routes for polling agent thoughts
+    # Use Gradio's add_api_route() method for custom FastAPI routes
     def api_get_thoughts(limit: int = 50):
         """Get recent agent thoughts as JSON."""
         return ws_manager.get_recent_thoughts(limit)
 
-    @app.get("/api/thoughts/stats")
     def api_get_stats():
         """Get WebSocket manager statistics."""
         return ws_manager.get_stats()
 
-    @app.post("/api/thoughts/clear")
     def api_clear_thoughts():
         """Clear the thought history."""
         ws_manager.clear_history()
         return {"status": "cleared"}
 
-    @app.get("/api/agent/status")
     def api_agent_status():
         """Get current agent status as JSON."""
         return load_cain_status()
 
-    @app.get("/api/agent/registry")
     def api_agent_registry():
         """Get agent registry as JSON."""
         return load_agent_registry()
 
+    # Add routes using Gradio's add_api_route method
+    app.add_api_route("/api/thoughts", api_get_thoughts, methods=["GET"])
+    app.add_api_route("/api/thoughts/stats", api_get_stats, methods=["GET"])
+    app.add_api_route("/api/thoughts/clear", api_clear_thoughts, methods=["POST"])
+    app.add_api_route("/api/agent/status", api_agent_status, methods=["GET"])
+    app.add_api_route("/api/agent/registry", api_agent_registry, methods=["GET"])
+
     return app