Update app.py

app.py

@@ -3,11 +3,6 @@ import requests
 from datetime import datetime, timedelta
 import xml.etree.ElementTree as ET
 import json
-import os
-
-client_id = os.getenv('client_id')
-api_key = os.getenv('api_key')
-
 
 def get_station_info(pattern, client_id, api_key):
     """Ermittelt EVA-Nummer und offiziellen Namen."""
@@ -46,6 +41,9 @@ def fetch_timetable_hour(eva, date_str, hour_str, client_id, api_key):
 
 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 {"error": "Bitte alle Felder ausfüllen."}
 
     # 1. IBNRs ermitteln
     dep_info = get_station_info(departure, client_id, api_key)
@@ -77,7 +75,7 @@ def search_next_3_connections(departure, destination, client_id, api_key):
                 pt = dp.get('pt', '') # Geplante Zeit: YYMMDDHHMM
                 if not pt: continue
                 
-                dep_dt = datetime.strptime(pt, "%H%M")
+                dep_dt = datetime.strptime(pt, "%y%m%d%H%M")
                 
                 # Nur Verbindungen in der Zukunft (ab Zeitstempel jetzt)
                 if dep_dt >= now:
@@ -85,7 +83,7 @@ def search_next_3_connections(departure, destination, client_id, api_key):
                     found_connections.append({
                         "startort": dep_info['name'],
                         "zielort": dest_info['name'],
-                        "abfahrtszeit": dep_dt.strftime("%H:%M"),
+                        "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 # Hilfsfeld zum Sortieren
@@ -105,13 +103,17 @@ def search_next_3_connections(departure, destination, client_id, api_key):
 
 # --- Gradio UI ---
 
-def ui_wrapper(dep, dest, client_id, api_key):
-    results = search_next_3_connections(dep, dest, client_id, api_key)
+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.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="Abfahrtsort", placeholder="z.B. Berlin")
@@ -120,7 +122,7 @@ 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") # Nutzt Gradio JSON Komponente für schöneres Format
     
-    btn.click(fn=ui_wrapper, inputs=[dep_input, dest_input], outputs=output)
+    btn.click(fn=ui_wrapper, inputs=[dep_input, dest_input, cid_input, akey_input], outputs=output)
 
 if __name__ == "__main__":
     demo.launch()