Update app.py

app.py

@@ -9,34 +9,27 @@ from mcp import ClientSession
 from mcp.client.streamable_http import streamablehttp_client
 
 async def generate(input_text):
-    # MCP Server URL von Hugging Face
     mcp_url = "https://mgokg-db-timetable-api.hf.space"
     
     try:
-        # Gemini Client initialisieren
         client = genai.Client(api_key=os.environ.get("GEMINI_API_KEY"))
         
-        # Verbindung über den Streamable HTTP Transport
-        # Wir nutzen eine explizite Fehlerbehandlung für den Netzwerk-Stream
         async with streamablehttp_client(mcp_url) as (read_stream, write_stream):
             async with ClientSession(read_stream, write_stream) as mcp_session:
-                # WICHTIG: Handshake initialisieren
                 await mcp_session.initialize()
                 
-                # Gemini Modellkonfiguration
-                # Gemini kann direkt auf mcp_session zugreifen, um Tools zu extrahieren
                 model_id = "gemini-2.0-flash" 
                 
                 generate_content_config = types.GenerateContentConfig(
                     temperature=0.4,
                     tools=[
                         types.Tool(google_search=types.GoogleSearch()),
-                        mcp_session # Übergabe der aktiven MCP-Sitzung
+                        # Überprüfen, ob mcp_session hier korrekt verwendet wird
+                        types.Tool(mcp_session=mcp_session)  # Beispiel für die korrekte Verwendung
                     ],
                 )
 
                 response_text = ""
-                # Streaming der Antwort
                 async for chunk in client.aio.models.generate_content_stream(
                     model=model_id,
                     contents=input_text,
@@ -48,11 +41,9 @@ async def generate(input_text):
                 return response_text, ""
 
     except Exception as e:
-        # Detaillierte Fehlerausgabe für TaskGroup-Fehler
         return f"Verbindungsfehler zum MCP-Server oder Gemini: {str(e)}", ""
 
 def gradio_wrapper(input_text):
-    # Führt die asynchrone Logik im synchronen Gradio-Kontext aus
     return asyncio.run(generate(input_text))
 
 if __name__ == '__main__':
@@ -65,4 +56,4 @@ if __name__ == '__main__':
         
         btn.click(fn=gradio_wrapper, inputs=input_tx, outputs=[output_md, input_tx])
         
-    demo.launch()
+    demo.launch()