Add disconnect to client

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

mcp_client.py

@@ -126,6 +126,12 @@ class MCPClientWrapper:
         tool_names = [tool["name"] for tool in self.tools]
         return f"Connected to MCP server. Available tools: {', '.join(tool_names)}"
 
+    async def disconnect(self):
+        if self.exit_stack:
+            await self.exit_stack.aclose()
+            self.exit_stack = None
+            self.session = None
+
     async def process_message(
         self, message: str, history: List[Union[Dict[str, Any], ChatMessage]]
     ):
@@ -353,7 +359,7 @@ def gradio_interface():
 
         chatbot = gr.Chatbot(
             value=[],
-            height=600,
+            height="70vh",
             type="messages",
             show_copy_button=True,
             avatar_images=("img/small-user.png", "img/small-robot.png"),
@@ -368,14 +374,18 @@ def gradio_interface():
             )
             clear_btn = gr.Button("Clear Chat", scale=1)
 
-        connect_btn.click(client.connect, inputs=server_path, outputs=status)
+        # connect_btn.click(client.connect, inputs=server_path, outputs=status)
 
         # Automatically call client.connect(...) as soon as the interface loads
-        demo.load(fn=client.connect, inputs=server_path, outputs=status)
+        demo.load(
+            fn=client.connect, inputs=server_path, outputs=status, show_progress="full"
+        )
 
         msg.submit(client.process_message, [msg, chatbot], [chatbot, msg])
         clear_btn.click(lambda: [], None, chatbot)
 
+        demo.unload(client.disconnect)
+
     return demo