Update app.py

app.py

@@ -1,59 +1,19 @@
-import os
 import asyncio
-import gradio as gr
-from google import genai
-from google.genai import types
 
-# Korrekte Import-Struktur für 2026
-from mcp import ClientSession
-from mcp.client.streamable_http import streamablehttp_client
-
-async def generate(input_text):
-    mcp_url = "https://mgokg-db-timetable-api.hf.space"
-    
+async def connect_to_server(server):
     try:
-        client = genai.Client(api_key=os.environ.get("GEMINI_API_KEY"))
-        
-        async with streamablehttp_client(mcp_url) as (read_stream, write_stream):
-            async with ClientSession(read_stream, write_stream) as mcp_session:
-                await mcp_session.initialize()
-                
-                model_id = "gemini-2.0-flash" 
-                
-                generate_content_config = types.GenerateContentConfig(
-                    temperature=0.4,
-                    tools=[
-                        types.Tool(google_search=types.GoogleSearch()),
-                        # Überprüfen, ob mcp_session hier korrekt verwendet wird
-                        types.Tool(mcp_session=mcp_session)  # Beispiel für die korrekte Verwendung
-                    ],
-                )
-
-                response_text = ""
-                async for chunk in client.aio.models.generate_content_stream(
-                    model=model_id,
-                    contents=input_text,
-                    config=generate_content_config,
-                ):
-                    if chunk.text:
-                        response_text += chunk.text
-                
-                return response_text, ""
-
+        # Simulierte Verbindung zum Server
+        await asyncio.sleep(1)  # Simuliert eine Verzögerung
+        raise ConnectionError("Verbindungsfehler")  # Simuliert einen Verbindungsfehler
     except Exception as e:
-        return f"Verbindungsfehler zum MCP-Server oder Gemini: {str(e)}", ""
+        print(f"Fehler beim Verbinden mit {server}: {e}")
 
-def gradio_wrapper(input_text):
-    return asyncio.run(generate(input_text))
+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)
 
-if __name__ == '__main__':
-    with gr.Blocks() as demo:
-        gr.Markdown("# Gemini Flash + DB Timetable (Hugging Face MCP)")
-        with gr.Row():
-            input_tx = gr.Textbox(label="Anfrage", placeholder="Nächster Zug von Berlin nach Hamburg?")
-            btn = gr.Button("Senden")
-        output_md = gr.Markdown()
-        
-        btn.click(fn=gradio_wrapper, inputs=input_tx, outputs=[output_md, input_tx])
-        
-    demo.launch()
+# Ausführung des Hauptprogramms
+asyncio.run(main())