Update app.py

app.py

@@ -1,3 +1,116 @@
+import gradio as gr
+import os
+import asyncio
+from typing import List
+from google import genai
+from google.genai import types
+from mcp import ClientSession, StdioServerParameters
+from mcp.client.stdio import stdio_client
+
+# Konfiguration des MCP-Servers für die DB-Fahrplanauskunft [1, 3]
+server_params = StdioServerParameters(
+    command="npx",
+    args=[
+        "mcp-remote",
+        "https://mgokg-db-timetable-api.hf.space/gradio_api/mcp/",
+        "--transport",
+        "streamable-http"
+    ]
+)
+
+async def generate(input_text):
+    try:
+        client = genai.Client(api_key=os.environ.get("GEMINI_API_KEY"))
+    except Exception as e:
+        return f"Error initializing client: {e}", ""
+
+    model = "gemini-2.0-flash" # Empfohlen für Tool-Calling [6]
+    
+    # Aufbau der Verbindung zum MCP-Server [9, 10]
+    async with stdio_client(server_params) as (read, write):
+        async with ClientSession(read, write) as session:
+            await session.initialize()
+            
+            # MCP-Tools abrufen und für Gemini konvertieren [9, 11]
+            mcp_tools_list = await session.list_tools()
+            mcp_declarations = [
+                {
+                    "name": tool.name,
+                    "description": tool.description or "Get train connections between stations",
+                    "parameters": tool.inputSchema,
+                }
+                for tool in mcp_tools_list.tools
+            ]
+
+            # Kombinierte Tools: Google Search + MCP Tools [2, 7]
+            tools = [
+                types.Tool(google_search=types.GoogleSearch()),
+                types.Tool(function_declarations=mcp_declarations)
+            ]
+
+            contents = [types.Content(role="user", parts=[types.Part.from_text(text=input_text)])]
+            
+            # Initialer Aufruf
+            response = await client.aio.models.generate_content(
+                model=model,
+                contents=contents,
+                config=types.GenerateContentConfig(
+                    tools=tools,
+                    temperature=0.4
+                )
+            )
+            
+            contents.append(response.candidates.content)
+            
+            # Tool Calling Loop (für Fahrplandaten oder Google Search) [7, 8]
+            turn_count = 0
+            while response.function_calls and turn_count < 5:
+                turn_count += 1
+                tool_responses = []
+                
+                for fc in response.function_calls:
+                    # Ausführung der MCP-Tools (z.B. db_timetable_api_ui_wrapper) [8, 12]
+                    try:
+                        tool_result = await session.call_tool(fc.name, fc.args)
+                        result_text = tool_result.content.text if not tool_result.isError else tool_result.content.text
+                        tool_responses.append(types.Part.from_function_response(
+                            name=fc.name, response={"result": result_text}
+                        ))
+                    except Exception as e:
+                        tool_responses.append(types.Part.from_function_response(
+                            name=fc.name, response={"error": str(e)}
+                        ))
+
+                contents.append(types.Content(role="user", parts=tool_responses))
+                response = await client.aio.models.generate_content(
+                    model=model, contents=contents, config=types.GenerateContentConfig(tools=tools)
+                )
+                contents.append(response.candidates.content)
+
+            return response.text, ""
+
+# Gradio UI (Original Funktionalität und Design beibehalten) [13]
+def ui_wrapper(input_text):
+    return asyncio.run(generate(input_text))
+
+if __name__ == '__main__':
+    with gr.Blocks() as demo:
+        gr.Markdown("# Gemini 2.0 Flash + Websearch + DB Timetable (MCP)")
+        output_textbox = gr.Markdown()
+        input_textbox = gr.Textbox(lines=3, label="", placeholder="Frage nach einer Zugverbindung (z.B. Berlin nach München)...")
+        submit_button = gr.Button("Senden")
+        
+        submit_button.click(
+            fn=ui_wrapper,
+            inputs=input_textbox,
+            outputs=[output_textbox, input_textbox]
+        )
+    demo.launch(show_error=True)
+
+
+
+
+"""
 import base64
 import gradio as gr
 import os
@@ -62,3 +175,4 @@ if __name__ == '__main__':
         submit_button = gr.Button("send")
         submit_button.click(fn=generate,inputs=input_textbox,outputs=[output_textbox, input_textbox])
     demo.launch(show_error=True)
+"""