Prevent premature disconnection to the mcp server

Signed-off-by: Aivin V. Solatorio <avsolatorio@gmail.com>

mcp_remote_client.py

@@ -72,7 +72,8 @@ class MCPClientWrapper:
         try:
             # If there's an existing session, close it
             if self.exit_stack:
-                await self.exit_stack.aclose()
+                return "Already connected to an MCP server. Please disconnect first."
+                # await self.exit_stack.aclose()
 
             self.exit_stack = AsyncExitStack()
 
@@ -460,10 +461,15 @@ def gradio_interface(
             fn=client.connect, inputs=server_path, outputs=status, show_progress="full"
         )
 
-        msg.submit(client.process_message, [msg, chatbot], [chatbot, msg])
+        msg.submit(
+            client.process_message,
+            [msg, chatbot],
+            [chatbot, msg],
+            concurrency_limit=10,
+        )
         # clear_btn.click(lambda: [], None, chatbot)
 
-        demo.unload(client.disconnect)
+        # demo.unload(client.disconnect)
 
     return demo
 
@@ -474,7 +480,10 @@ if __name__ == "__main__":
             "Warning: ANTHROPIC_API_KEY not found in environment. Please set it in your .env file."
         )
 
-    interface = gradio_interface()
+    # interface = gradio_interface(server_path_or_url="wdi_mcp_server.py")
+    interface = gradio_interface(
+        server_path_or_url="https://avsolatorio-test-data-mcp-server.hf.space/gradio_api/mcp/sse"
+    )
     interface.launch(
         server_name=os.getenv("SERVER_NAME", "127.0.0.1"),
         server_port=os.getenv("SERVER_PORT", 7860),