Add disconnect on reload

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

mcp_remote_client.py

@@ -134,6 +134,12 @@ class MCPClientWrapper:
             self.session = None
             return error_msg
 
+    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]]
     ):
@@ -425,6 +431,8 @@ def gradio_interface(
         msg.submit(client.process_message, [msg, chatbot], [chatbot, msg])
         clear_btn.click(lambda: [], None, chatbot)
 
+        demo.unload(client.disconnect)
+
     return demo