Claude Code: Fix the in (around line 780). The issue is try

app.py

@@ -23,6 +23,8 @@ from dataclasses import dataclass, asdict
 from enum import Enum
 
 import gradio as gr
+from fastapi import FastAPI
+from fastapi.responses import JSONResponse
 
 # ========== Configuration ==========
 
@@ -749,41 +751,46 @@ def create_agent_office():
 def create_agent_office_with_ws():
     """
     Create the Agent Office with WebSocket API endpoints.
-    Returns the Gradio app with additional routes for WebSocket functionality.
+    Returns a FastAPI app with the Gradio app mounted at root.
     """
-    app = create_agent_office()
+    # Create FastAPI app first
+    fastapi_app = FastAPI(title="HuggingClaw Agent Office API")
 
-    # 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):
+    # Define API route handlers
+    @fastapi_app.get("/api/thoughts")
+    async def api_get_thoughts(limit: int = 50):
         """Get recent agent thoughts as JSON."""
         return ws_manager.get_recent_thoughts(limit)
 
-    def api_get_stats():
+    @fastapi_app.get("/api/thoughts/stats")
+    async def api_get_stats():
         """Get WebSocket manager statistics."""
         return ws_manager.get_stats()
 
-    def api_clear_thoughts():
+    @fastapi_app.post("/api/thoughts/clear")
+    async def api_clear_thoughts():
         """Clear the thought history."""
         ws_manager.clear_history()
         return {"status": "cleared"}
 
-    def api_agent_status():
+    @fastapi_app.get("/api/agent/status")
+    async def api_agent_status():
         """Get current agent status as JSON."""
         return load_cain_status()
 
-    def api_agent_registry():
+    @fastapi_app.get("/api/agent/registry")
+    async 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"])
+    # Create Gradio app
+    gradio_app = create_agent_office()
 
-    return app
+    # Mount Gradio app to FastAPI app at root path
+    # This makes the Gradio UI available at "/" and API routes at "/api/*"
+    gradio_app = gr.mount_gradio_app(fastapi_app, gradio_app, path="/")
+
+    return gradio_app
 
 
 # ========== Main Entry Point ==========