Update app.py

app.py

@@ -1,5 +1,194 @@
 import gradio as gr
 import requests
+from datetime import datetime
+import xml.etree.ElementTree as ET
+
+def get_station_info(pattern, client_id, api_key):
+    """
+    Sucht über die Timetables API nach einem Bahnhof und gibt EVA-Nummer 
+    und den korrekten Namen zurück.
+    """
+    url = f"https://apis.deutschebahn.com/db-api-marketplace/apis/timetables/v1/station/{pattern}"
+    
+    headers = {
+        'DB-Client-Id': client_id,
+        'DB-Api-Key': api_key,
+        'accept': 'application/xml'
+    }
+
+    try:
+        response = requests.get(url, headers=headers, timeout=10)
+        response.raise_for_status()
+        
+        root = ET.fromstring(response.content)
+        stations = []
+        for station in root.findall('station'):
+            stations.append({
+                'name': station.get('name'),
+                'eva': station.get('eva')
+            })
+        
+        # Falls Stationen gefunden wurden, nehmen wir die erste (beste Übereinstimmung)
+        return stations[0] if stations else None
+
+    except Exception as e:
+        print(f"Fehler bei Stationssuche: {e}")
+        return None
+
+def search_connections(departure, destination, client_id, api_key):
+    """Sucht Verbindungen zwischen zwei Bahnhöfen"""
+    
+    if not departure or not destination:
+        return "❌ Bitte Abfahrts- und Zielort eingeben"
+    
+    if not client_id or not api_key:
+        return "❌ Bitte Client ID und API Key eingeben"
+    
+    # 1. IBNR (EVA) für Abfahrtsort ermitteln
+    dep_info = get_station_info(departure, client_id, api_key)
+    if not dep_info:
+        return f"❌ Abfahrtsort '{departure}' wurde nicht gefunden."
+    
+    # 2. IBNR (EVA) für Zielort ermitteln (für genaueres Matching im Pfad)
+    dest_info = get_station_info(destination, client_id, api_key)
+    if not dest_info:
+        return f"❌ Zielort '{destination}' wurde nicht gefunden."
+
+    departure_eva = dep_info['eva']
+    departure_name = dep_info['name']
+    destination_name = dest_info['name']
+    
+    # Aktuelles Datum und Stunde
+    now = datetime.now()
+    datum = now.strftime("%y%m%d")  # YYMMDD
+    stunde = now.strftime("%H")      # HH
+    
+    # API-Aufruf für den Fahrplan
+    url = f"https://apis.deutschebahn.com/db-api-marketplace/apis/timetables/v1/plan/{departure_eva}/{datum}/{stunde}"
+    
+    headers = {
+        "DB-Client-Id": client_id,
+        "DB-Api-Key": api_key,
+        "accept": "application/xml"
+    }
+    
+    try:
+        response = requests.get(url, headers=headers, timeout=10)
+        
+        if response.status_code != 200:
+            return f"❌ API Fehler beim Abrufen des Fahrplans: {response.status_code}"
+        
+        root = ET.fromstring(response.content)
+        connections = []
+        
+        for train in root.findall('.//s'):
+            dp = train.find('dp')
+            if dp is None:
+                continue
+            
+            # Route des Zugs (ppth = planned path)
+            path = dp.get('ppth', '')
+            
+            # Prüfen ob Zielort im Pfad enthalten ist
+            # Wir prüfen sowohl den eingegebenen Namen als auch den gefundenen offiziellen Namen
+            if destination_name.lower() in path.lower() or destination.lower() in path.lower():
+                # Zugdetails extrahieren
+                tl = train.find('tl') # Train Label
+                train_class = tl.get('c', '') # z.B. ICE
+                train_number = tl.get('n', '') # z.B. 123
+                full_train_name = f"{train_class} {train_number}"
+                
+                planned_time = dp.get('pt', '')
+                platform = dp.get('pp', 'n/a')
+                
+                if len(planned_time) == 10: # Format YYMMDDHHMM
+                    planned_time = f"{planned_time[8:10]}:{planned_time[10:12]}"
+                elif len(planned_time) == 4:
+                    planned_time = f"{planned_time[:2]}:{planned_time[2:]}"
+                
+                route = ' → '.join(path.split('|'))
+                
+                connections.append({
+                    'train': full_train_name,
+                    'time': planned_time,
+                    'platform': platform,
+                    'route': route
+                })
+        
+        if not connections:
+            return f"ℹ️ Keine direkten Verbindungen von {departure_name} nach {destination_name} gefunden.\n(Suche im Zeitraum {now.strftime('%H:00')} - {int(now.strftime('%H'))+1}:00 Uhr)"
+        
+        result = f"🚆 Verbindungen von: {departure_name}\n"
+        result += f"🏁 Ziel: {destination_name}\n"
+        result += f"📅 Datum: {now.strftime('%d.%m.%Y')}\n\n"
+        
+        for i, conn in enumerate(connections[:15], 1):
+            result += f"─────────────────────────────────\n"
+            result += f"{i}. {conn['train']}\n"
+            result += f"   ⏰ Abfahrt: {conn['time']} Uhr\n"
+            result += f"   ���� Gleis: {conn['platform']}\n"
+            result += f"   🗺️  Route: {conn['route'][:100]}...\n\n"
+        
+        return result
+        
+    except Exception as e:
+        return f"❌ Fehler: {str(e)}"
+
+# Gradio Interface
+with gr.Blocks(title="DB Timetable API - Live Suche", theme=gr.themes.Soft()) as demo:
+    gr.Markdown(
+        """
+        # 🚆 DB Live-Fahrplanauskunft
+        Ermittelt automatisch die IBNR (EVA-Nummer) und sucht nach Verbindungen.
+        """
+    )
+    
+    with gr.Row():
+        with gr.Column():
+            client_id_input = gr.Textbox(label="DB Client ID", type="password")
+            api_key_input = gr.Textbox(label="DB API Key", type="password")
+    
+    with gr.Row():
+        with gr.Column():
+            departure_input = gr.Textbox(
+                label="Abfahrtsort", 
+                placeholder="z.B. Berlin Hbf oder Frankfurt"
+            )
+        with gr.Column():
+            destination_input = gr.Textbox(
+                label="Zielort", 
+                placeholder="z.B. München Hbf"
+            )
+    
+    search_button = gr.Button("🔍 Verbindungen suchen", variant="primary")
+    
+    output = gr.Textbox(label="Ergebnisse", lines=20)
+    
+    gr.Markdown(
+        """
+        ### ℹ️ Funktionsweise
+        1. Die App sendet Ihren Suchbegriff an den `/station` Endpoint.
+        2. Die erste gefundene **EVA-Nummer (IBNR)** wird für die Abfrage genutzt.
+        3. Der Fahrplan für die aktuelle Stunde wird abgerufen.
+        4. Es wird gefiltert, ob der Zielbahnhof im geplanten Fahrtverlauf vorkommt.
+        """
+    )
+    
+    search_button.click(
+        fn=search_connections,
+        inputs=[departure_input, destination_input, client_id_input, api_key_input],
+        outputs=output
+    )
+
+if __name__ == "__main__":
+    demo.launch()
+
+
+
+
+"""
+import gradio as gr
+import requests
 from datetime import datetime, timedelta
 import xml.etree.ElementTree as ET
 import json
@@ -125,4 +314,4 @@ with gr.Blocks(title="DB JSON Fahrplan", theme=gr.themes.Soft()) as demo:
     btn.click(fn=ui_wrapper, inputs=[dep_input, dest_input, cid_input, akey_input], outputs=output)
 
 if __name__ == "__main__":
-    demo.launch()
+"""