Update app.py

app.py

@@ -7,8 +7,8 @@ import aiohttp
 from typing import Dict, Any, Optional, List
 from google import genai
 from google.genai import types
-from contextlib import AsyncExitStack
-from mcp import ClientSession
+from mcp import ClientSession, StdioServerParameters
+from mcp.client.stdio import stdio_client
 from mcp.client.streamable_http import streamablehttp_client
 import nest_asyncio
 
@@ -21,7 +21,6 @@ class MCPTrainClient:
     def __init__(self):
         self.session = None
         self.exit_stack = None
-        self.tools = []
         self.mcp_server_url = "https://mgokg-db-timetable-api.hf.space/gradio_api/mcp/"
         
     async def connect(self) -> str:
@@ -32,23 +31,18 @@ class MCPTrainClient:
             
             self.exit_stack = AsyncExitStack()
             
-            # Connect using Streamable HTTP transport
-            transport = await self.exit_stack.enter_async_context(
-                streamablehttp_client(self.mcp_server_url)
-            )
-            
-            self.read_stream, self.write_stream = transport
+            # Connect using Streamable HTTP transport - FIXED VERSION
             self.session = await self.exit_stack.enter_async_context(
-                ClientSession(self.read_stream, self.write_stream)
+                streamablehttp_client(self.mcp_server_url)
             )
             
+            # Initialize the session
             await self.session.initialize()
             
             # List available tools
             response = await self.session.list_tools()
-            self.tools = response.tools
+            tool_names = [tool.name for tool in response.tools]
             
-            tool_names = [tool.name for tool in self.tools]
             return f"✅ Connected to MCP server. Available tools: {', '.join(tool_names)}"
             
         except Exception as e:
@@ -80,7 +74,6 @@ class MCPTrainClient:
             await self.exit_stack.aclose()
             self.exit_stack = None
             self.session = None
-            self.tools = []
 
 
 class GeminiMCPIntegration: