Update app.py

app.py

@@ -1,19 +1,44 @@
 import asyncio
+from mcp import ClientSession
+from mcp.client.sse import sse_client
 
-async def connect_to_server(server):
-    try:
-        # Simulierte Verbindung zum Server
-        await asyncio.sleep(1)  # Simuliert eine Verzögerung
-        raise ConnectionError("Verbindungsfehler")  # Simuliert einen Verbindungsfehler
-    except Exception as e:
-        print(f"Fehler beim Verbinden mit {server}: {e}")
+async def run_mcp_client():
+    # Die Basis-URL deines Hugging Face Spaces (mit /sse Endpunkt für das Python SDK)
+    server_url = "https://mgokg-db-timetable-api.hf.space"
+
+    print(f"Verbinde zu MCP Server: {server_url}...")
+
+    # 1. Transport-Layer aufbauen (SSE)
+    async with sse_client(url=server_url) as (read_stream, write_stream):
+        
+        # 2. MCP-Session initialisieren
+        async with ClientSession(read_stream, write_stream) as session:
+            await session.initialize()
+            print("Verbindung erfolgreich hergestellt!\n")
 
-async def main():
-    servers = ["MCP-Server", "Gemini"]
-    tasks = [connect_to_server(server) for server in servers]
-    
-    # Verwendung von asyncio.gather, um alle Aufgaben zu verwalten
-    await asyncio.gather(*tasks, return_exceptions=True)
+            # --- Beispiel 1: Verfügbare Tools auflisten ---
+            print("Verfügbare Tools:")
+            tools_result = await session.list_tools()
+            for tool in tools_result.tools:
+                print(f"- {tool.name}: {tool.description}")
 
-# Ausführung des Hauptprogramms
-asyncio.run(main())
+            # --- Beispiel 2: Ein Tool aufrufen ---
+            # Ersetze 'get_timetable' und die Argumente durch die echten Tool-Namen deiner API
+            if tools_result.tools:
+                tool_to_call = tools_result.tools[0].name
+                print(f"\nRufe Tool '{tool_to_call}' auf...")
+                
+                # Beispielhafter Aufruf (Passe die Argumente an deine API an!)
+                result = await session.call_tool(
+                    name=tool_to_call,
+                    arguments={"station": "Berlin Hbf"} 
+                )
+                print(f"Ergebnis: {result.content}")
+
+if __name__ == "__main__":
+    try:
+        asyncio.run(run_mcp_client())
+    except KeyboardInterrupt:
+        pass
+    except Exception as e:
+        print(f"Fehler: {e}")