Update app.py

app.py

@@ -6,6 +6,37 @@ from google import genai
 from google.genai import types
 from gradio_client import Client
 
+
+route="""	
+Achtung! Diese Anweisungen gelten nur wenn eine Zugverbindung angefragt wird. 
+how to handle special case "zugverbindung".
+
+Rolle: Du bist ein spezialisierter API-Handler für Reiseanfragen. 
+Aufgabe: Identifiziere Anfragen zu Zugverbindungen und extrahiere Start- und Zielort.
+
+Regeln:
+Wenn eine Zugverbindung von {start_loc} nach {dest_loc} angefragt wird, antworte ausschließlich mit dem unten stehenden JSON-Objekt.
+
+Fülle die Werte für start_loc und dest_loc basierend auf der Benutzereingabe aus.
+
+Wichtig: Gib absolut keinen Text vor oder nach dem JSON aus (keine Erklärungen, kein "Hier ist das Ergebnis").
+
+Achte auf die korrekte Bezeichnung der Bahnhöfe (z.B. "Hbf", "Bahnhof").
+
+JSON-Format:
+
+JSON
+
+{
+  "start_loc": "Eingabe Startort",
+  "dest_loc": "Eingabe Zielort"
+}
+
+Beispiel: Eingabe: "Ich möchte mit dem Zug von Bad Kissingen Bahnhof nach Würzburg Hbf fahren.Ausgabe: {"start_loc": "Bad Kissingen Bahnhof", "dest_loc": "Würzburg Hbf"}
+"""
+
+
+
 def clean_json_string(json_str):
     """
     Removes any comments or prefixes before the actual JSON content.
@@ -41,7 +72,7 @@ def generate(input_text):
         types.Content(
             role="user",
             parts=[
-                types.Part.from_text(text=input_text),
+                types.Part.from_text(text=f"input_text {route}",
             ],
         ),
     ]