Update app.py

app.py

@@ -2,9 +2,9 @@ import base64
 import gradio as gr
 import os
 import json
+import requests
 from google import genai
 from google.genai import types
-from gradio_client import Client
 
 route="""	
 how to handle special case "zugverbindung".
@@ -50,17 +50,47 @@ def is_train_connection_query(text):
 
 def get_train_connections(start_loc, dest_loc):
     """
-    Calls the MCP server to get train connections.
+    Calls the MCP server via HTTP to get train connections.
     """
     try:
-        client = Client("mgokg/db-timetable-api")
-        result = client.predict(
-            query=f"Verbindung von {start_loc} nach {dest_loc}",
-            api_name="/db_timetable_api_ui_wrapper"
-        )
-        return result
+        mcp_url = "https://mgokg-db-timetable-api.hf.space/gradio_api/mcp/"
+        
+        # MCP protocol request structure
+        payload = {
+            "jsonrpc": "2.0",
+            "id": 1,
+            "method": "tools/call",
+            "params": {
+                "name": "db_timetable_api_ui_wrapper",
+                "arguments": {
+                    "query": f"Verbindung von {start_loc} nach {dest_loc}"
+                }
+            }
+        }
+        
+        headers = {
+            "Content-Type": "application/json"
+        }
+        
+        response = requests.post(mcp_url, json=payload, headers=headers, timeout=30)
+        response.raise_for_status()
+        
+        result = response.json()
+        
+        # Extract the result from MCP response
+        if "result" in result:
+            content = result["result"].get("content", [])
+            if content and len(content) > 0:
+                return content[0].get("text", "Keine Verbindungen gefunden")
+        
+        return f"Unerwartete Antwortstruktur: {result}"
+        
+    except requests.exceptions.Timeout:
+        return "Die Anfrage hat zu lange gedauert. Bitte versuche es erneut."
+    except requests.exceptions.RequestException as e:
+        return f"Fehler beim Abrufen der Zugverbindungen: {e}"
     except Exception as e:
-        return f"Error calling MCP server: {e}"
+        return f"Unerwarteter Fehler: {e}"
 
 def generate(input_text):
     # Check if this is a train connection query
@@ -110,7 +140,7 @@ def generate(input_text):
                 return "Konnte Start- oder Zielort nicht identifizieren. Bitte gib beide Orte an.", ""
                 
         except json.JSONDecodeError as e:
-            return f"Error parsing location data: {e}", ""
+            return f"Error parsing location data: {e}\nResponse was: {response.text}", ""
         except Exception as e:
             return f"Error processing train connection request: {e}", ""
     
@@ -166,4 +196,4 @@ if __name__ == '__main__':
         input_textbox = gr.Textbox(lines=3, label="", placeholder="Enter message here...")
         submit_button = gr.Button("send")
         submit_button.click(fn=generate, inputs=input_textbox, outputs=[output_textbox, input_textbox])
-    demo.launch(show_error=True)
+    demo.launch(show_error=True)