Update app.py

app.py

@@ -5,10 +5,43 @@ import xml.etree.ElementTree as ET
 import json
 import os
 
-# Load credentials from environment variables
+# Lade Anmeldeinformationen aus Umgebungsvariablen
 CLIENT_ID = os.environ.get('client_id')
 API_KEY = os.environ.get('api_key')
 
+def json_to_markdown_table(json_data):
+    """
+    Konvertiert JSON-Daten in eine professionell formatierte Markdown-Tabelle
+    """
+    if isinstance(json_data, str):
+        data = json.loads(json_data)
+    else:
+        data = json_data
+    
+    if not data:
+        return "Keine Daten vorhanden"
+    
+    header_mapping = {
+        "startort": "🚉 Startort",
+        "zielort": "🎯 Zielort",
+        "abfahrtszeit": "⏰ Abfahrt",
+        "ankunftszeitszeit": "⏱️ Ankunft",
+        "gleis": "🛤️ Gleis",
+        "duration": "⏳ Dauer"
+    }
+    
+    columns = list(data[0].keys())
+    headers = [header_mapping.get(col, col) for col in columns]
+    
+    markdown = "| " + " | ".join(headers) + " |\n"
+    markdown += "|" + "|".join(["---" for _ in headers]) + "|\n"
+    
+    for row in data:
+        values = [str(row.get(col, "")) for col in columns]
+        markdown += "| " + " | ".join(values) + " |\n"
+    
+    return markdown
+
 def get_station_info(pattern, client_id, api_key):
     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'}
@@ -59,7 +92,6 @@ def search_next_3_connections(departure, destination, client_id, api_key):
             tl = train.find('tl')
             if dp is None or tl is None: continue
             
-            # Check if destination is in the path
             path = dp.get('ppth', '')
             if dest_info['name'].lower() in path.lower() or destination.lower() in path.lower():
                 pt = dp.get('pt', '')
@@ -68,17 +100,12 @@ def search_next_3_connections(departure, destination, client_id, api_key):
                 dep_dt = datetime.strptime(pt, "%y%m%d%H%M")
                 
                 if dep_dt >= now:
-                    # To get arrival time, we must check the destination station for the same train ID
-                    # For performance, we assume arrival is within +2 hours of departure
                     arr_time_str = "N/A"
                     duration_str = "N/A"
                     
-                    # Search destination timetable for the arrival
-                    # We check the hour of departure and the next hour for arrival
                     arrival_root = fetch_timetable_hour(dest_info['eva'], dep_dt.strftime("%y%m%d"), dep_dt.strftime("%H"), client_id, api_key)
                     
                     if arrival_root is not None:
-                        # Find the train by number (n) and category (c)
                         for arr_train in arrival_root.findall('.//s'):
                             arr_tl = arr_train.find('tl')
                             arr_ar = arr_train.find('ar')
@@ -88,7 +115,6 @@ def search_next_3_connections(departure, destination, client_id, api_key):
                                     arr_dt = datetime.strptime(arr_pt, "%y%m%d%H%M")
                                     arr_time_str = arr_dt.strftime("%H:%M")
                                     
-                                    # Calculate duration
                                     diff = arr_dt - dep_dt
                                     duration_str = f"{int(diff.total_seconds() // 60)} min"
                                     break
@@ -105,20 +131,19 @@ def search_next_3_connections(departure, destination, client_id, api_key):
 
     found_connections.sort(key=lambda x: x['_dt'])
     
-    # Return list of first 3 matches formatted without internal sort key
     return [{k: v for k, v in c.items() if k != "_dt"} for c in found_connections[:3]]
 
 # --- Gradio UI ---
 
 def ui_wrapper(dep, dest):
     if not CLIENT_ID or not API_KEY:
-        return {"error": "API credentials not configured."}
+        return {"error": "API-Anmeldeinformationen nicht konfiguriert."}
     
     results = search_next_3_connections(dep, dest, CLIENT_ID, API_KEY)
-    return results # Return list directly for gr.JSON
+    return results
 
 with gr.Blocks(title="DB JSON Fahrplan", theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# │ DB │ Deutsche Bahn train connections")
+    gr.Markdown("# │ DB │ Deutsche Bahn Zugverbindungen")
     
     with gr.Row():
         dep_input = gr.Textbox(label="Abfahrtsort", placeholder="z.B. Schweinfurt Hbf")
@@ -126,8 +151,9 @@ with gr.Blocks(title="DB JSON Fahrplan", theme=gr.themes.Soft()) as demo:
         
     btn = gr.Button("Suchen", variant="primary")
     output = gr.JSON(label="JSON Ergebnis")
+    md_output = gr.Markdown()
     
     btn.click(fn=ui_wrapper, inputs=[dep_input, dest_input], outputs=output)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()