Update app.py

app.py

@@ -3,6 +3,11 @@ import requests
 from datetime import datetime, timedelta
 import xml.etree.ElementTree as ET
 import json
+import os
+
+# Load credentials from environment variables
+CLIENT_ID = os.environ.get('client_id')
+API_KEY = os.environ.get('api_key')
 
 def get_station_info(pattern, client_id, api_key):
     """Ermittelt EVA-Nummer und offiziellen Namen."""
@@ -103,18 +108,17 @@ def search_next_3_connections(departure, destination, client_id, api_key):
 
 # --- Gradio UI ---
 
-def ui_wrapper(dep, dest, cid, akey):
-    results = search_next_3_connections(dep, dest, cid, akey)
+def ui_wrapper(dep, dest):
+    if not CLIENT_ID or not API_KEY:
+        return {"error": "API credentials not configured. Please set client_id and api_key as environment variables."}
+    
+    results = search_next_3_connections(dep, dest, CLIENT_ID, API_KEY)
     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")
         dest_input = gr.Textbox(label="Zielort", placeholder="z.B. Hamburg")
@@ -122,7 +126,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, cid_input, akey_input], outputs=output)
+    btn.click(fn=ui_wrapper, inputs=[dep_input, dest_input], outputs=output)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(mcp_server=True)