chatui troubleshoot

app/main.py

@@ -505,6 +505,23 @@ if __name__ == "__main__":
     if is_hf_spaces:
         # On HF Spaces, just launch Gradio with API endpoints
         logger.info("🚀 Starting on Hugging Face Spaces")
+
+        # Register LangServe routes for ChatUI + main chatfed endpoint
+        @demo.app.on_event("startup")
+        async def startup_event():
+            add_routes(
+                demo.app,
+                chatui_adapter_runnable,
+                path="/chatfed-chatui",
+                input_type=ChatUIInput
+            )
+            add_routes(
+                demo.app,
+                process_chatfed_query_runnable,
+                path="/chatfed",
+                input_type=ChatFedInput,
+                output_type=ChatFedOutput
+            )
         
         # Add API endpoints to Gradio's FastAPI app
         @demo.app.get("/health")
@@ -515,13 +532,13 @@ if __name__ == "__main__":
                 "api_endpoints": "available_via_gradio"
             }
         
-        @demo.app.post("/chatfed")
-        async def chatfed_endpoint(input_data: dict):
-            return process_chatfed_query_langserve(input_data)
+        # @demo.app.post("/chatfed")
+        # async def chatfed_endpoint(input_data: dict):
+        #     return process_chatfed_query_langserve(input_data)
         
-        @demo.app.post("/chatfed-chatui")
-        async def chatui_endpoint(input_data: dict):
-            return chatui_adapter(input_data)
+        # @demo.app.post("/chatfed-chatui")
+        # async def chatui_endpoint(input_data: dict):
+        #     return chatui_adapter(input_data)
         
         @demo.app.post("/process_query")
         async def process_query_endpoint(