Update app.py

app.py

@@ -7,7 +7,11 @@ import json
 def get_station_info(pattern, client_id, api_key):
     """Ermittelt EVA-Nummer und offiziellen Namen."""
     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'}
+    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()
@@ -22,7 +26,11 @@ def get_station_info(pattern, client_id, api_key):
 def fetch_timetable_hour(eva, date_str, hour_str, client_id, api_key):
     """Ruft den Fahrplan für eine bestimmte Stunde ab."""
     url = f"https://apis.deutschebahn.com/db-api-marketplace/apis/timetables/v1/plan/{eva}/{date_str}/{hour_str}"
-    headers = {'DB-Client-Id': client_id, 'DB-Api-Key': api_key, 'accept': 'application/xml'}
+    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:
@@ -31,22 +39,24 @@ def fetch_timetable_hour(eva, date_str, hour_str, client_id, api_key):
         pass
     return None
 
-def get_connections(departure, destination, client_id, api_key):
-    """Sucht die nächsten 3 Verbindungen ab jetzt."""
+def search_next_3_connections(departure, destination, client_id, api_key):
+    """Hauptfunktion: Findet die nächsten 3 Verbindungen ab 'jetzt'."""
+    
     if not all([departure, destination, client_id, api_key]):
-        return None, "❌ Bitte alle Felder ausfüllen."
+        return {"error": "Bitte alle Felder ausfüllen."}
 
+    # 1. IBNRs ermitteln
     dep_info = get_station_info(departure, client_id, api_key)
     dest_info = get_station_info(destination, client_id, api_key)
     
-    if not dep_info or not dest_info:
-        return None, "❌ Bahnhof nicht gefunden. Bitte Namen prüfen."
+    if not dep_info: return {"error": f"Startort '{departure}' nicht gefunden."}
+    if not dest_info: return {"error": f"Zielort '{destination}' nicht gefunden."}
 
     now = datetime.now()
     found_connections = []
     
-    # Suche über 3 Stunden-Slots hinweg (aktuelle + nächste zwei)
-    for hour_offset in range(3):
+    # 2. Suche in der aktuellen und der nächsten Stunde (um 3 Verbindungen zu garantieren)
+    for hour_offset in range(3): # Wir schauen bis zu 2 Stunden voraus
         search_time = now + timedelta(hours=hour_offset)
         datum = search_time.strftime("%y%m%d")
         stunde = search_time.strftime("%H")
@@ -58,85 +68,61 @@ def get_connections(departure, destination, client_id, api_key):
             dp = train.find('dp')
             if dp is None: continue
             
+            # Prüfen ob Ziel im geplanten Pfad (ppth)
             path = dp.get('ppth', '')
             if dest_info['name'].lower() in path.lower() or destination.lower() in path.lower():
-                pt = dp.get('pt', '')
+                
+                pt = dp.get('pt', '') # Geplante Zeit: YYMMDDHHMM
                 if not pt: continue
                 
                 dep_dt = datetime.strptime(pt, "%y%m%d%H%M")
                 
+                # Nur Verbindungen in der Zukunft (ab Zeitstempel jetzt)
                 if dep_dt >= now:
                     tl = train.find('tl')
                     found_connections.append({
                         "startort": dep_info['name'],
                         "zielort": dest_info['name'],
-                        "abfahrtszeit": dep_dt.strftime("%H:%M"),
-                        "datum": dep_dt.strftime("%d.%m.%Y"),
-                        "gleis": dp.get('pp', '-'),
+                        "abfahrtszeit": dep_dt.strftime("%d.%m.%Y %H:%M"),
+                        "gleis": dp.get('pp', 'n/a'),
                         "zug": f"{tl.get('c', '')} {tl.get('n', '')}",
-                        "_dt": dep_dt
+                        "_dt": dep_dt # Hilfsfeld zum Sortieren
                     })
 
+    # 3. Sortieren und Top 3 auswählen
     found_connections.sort(key=lambda x: x['_dt'])
-    return found_connections[:3], None
-
-def format_markdown(connections):
-    """Erstellt eine Smartphone-optimierte Markdown-Ansicht."""
-    if not connections:
-        return "### ℹ️ Keine Verbindungen gefunden\nIn den nächsten 3 Stunden wurden keine Direktverbindungen gefunden."
-    
-    md = f"## 🚆 {connections[0]['startort']} → {connections[0]['zielort']}\n"
-    md += f"*{datetime.now().strftime('%H:%M')} Uhr Stand*\n\n"
     
-    for i, c in enumerate(connections, 1):
-        md += f"### {i}. {c['abfahrtszeit']} Uhr\n"
-        md += f"**{c['zug']}**  \n"
-        md += f"Gleis: **{c['gleis']}** | Datum: {c['datum']}\n"
-        md += "---\n" # Trennlinie für bessere Lesbarkeit auf dem Handy
-    
-    return md
+    # Bereinigen der Ergebnisse für die JSON-Ausgabe
+    final_results = []
+    for c in found_connections[:3]:
+        # Entferne das interne Datetime-Objekt vor der JSON-Konvertierung
+        res = {k: v for k, v in c.items() if k != "_dt"}
+        final_results.append(res)
 
-def main_interface(dep, dest, cid, akey):
-    conns, error = get_connections(dep, dest, cid, akey)
-    
-    if error:
-        return error, json.dumps({"error": error}, indent=2)
-    
-    # JSON-Bereinigung (internes Sortierfeld entfernen)
-    clean_json = []
-    for c in conns:
-        clean_json.append({k: v for k, v in c.items() if k != "_dt"})
-    
-    md_output = format_markdown(clean_json)
-    json_output = json.dumps(clean_json, indent=4, ensure_ascii=False)
-    
-    return md_output, json_output
+    return final_results
 
 # --- Gradio UI ---
-with gr.Blocks(title="DB Mobile Fahrplan", theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 📱 DB Quick-Check")
+
+def ui_wrapper(dep, dest, cid, akey):
+    results = search_next_3_connections(dep, dest, cid, akey)
+    return json.dumps(results, indent=4, ensure_ascii=False)
+
+with gr.Blocks(title="DB JSON Fahrplan", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🚆 DB Verbindungs-Suche (JSON)")
+    gr.Markdown("Ermittelt die nächsten 3 Verbindungen ab dem aktuellen Zeitstempel.")
     
-    with gr.Accordion("⚙️ API-Konfiguration (Hier klicken)", open=False):
+    with gr.Row():
         cid_input = gr.Textbox(label="DB Client ID", type="password")
         akey_input = gr.Textbox(label="DB API Key", type="password")
         
     with gr.Row():
-        dep_input = gr.Textbox(label="Von", placeholder="z.B. Berlin", scale=2)
-        dest_input = gr.Textbox(label="Nach", placeholder="z.B. Hamburg", scale=2)
-    
-    btn = gr.Button("🔍 Verbindungen suchen", variant="primary")
+        dep_input = gr.Textbox(label="Abfahrtsort", placeholder="z.B. Berlin")
+        dest_input = gr.Textbox(label="Zielort", placeholder="z.B. Hamburg")
+        
+    btn = gr.Button("Suchen", variant="primary")
+    output = gr.JSON(label="JSON Ergebnis") # Nutzt Gradio JSON Komponente für schöneres Format
     
-    with gr.Tabs():
-        with gr.TabItem("📱 Ansicht"):
-            md_display = gr.Markdown("Geben Sie Start und Ziel ein und drücken Sie Suchen.")
-        with gr.TabItem("💻 JSON"):
-            json_display = gr.Code(label="API Response", language="json")
-            
-    btn.click(
-        fn=main_interface, 
-        inputs=[dep_input, dest_input, cid_input, akey_input], 
-        outputs=[md_display, json_display]
-    )
+    btn.click(fn=ui_wrapper, inputs=[dep_input, dest_input, cid_input, akey_input], outputs=output)
 
 if __name__ == "__main__":
     demo.launch()