Update app.py

app.py

@@ -3,154 +3,10 @@ import requests
 from datetime import datetime, timedelta
 import xml.etree.ElementTree as ET
 import json
-import os
 
 api_key=os.environ.get("api_key")
 client_id=os.environ.get("client_id")
 
-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'}
-    try:
-        response = requests.get(url, headers=headers, timeout=10)
-        response.raise_for_status()
-        root = ET.fromstring(response.content)
-        station = root.find('station')
-        if station is not None:
-            return {'name': station.get('name'), 'eva': station.get('eva')}
-    except Exception:
-        return None
-    return None
-
-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'}
-    try:
-        response = requests.get(url, headers=headers, timeout=10)
-        if response.status_code == 200:
-            return ET.fromstring(response.content)
-    except Exception:
-        pass
-    return None
-
-def get_connections(departure, destination, client_id, api_key):
-    """Sucht die nächsten 3 Verbindungen ab jetzt."""
-
-    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."
-
-    now = datetime.now()
-    found_connections = []
-    
-    # Suche über 3 Stunden-Slots hinweg (aktuelle + nächste zwei)
-    for hour_offset in range(3):
-        search_time = now + timedelta(hours=hour_offset)
-        datum = search_time.strftime("%y%m%d")
-        stunde = search_time.strftime("%H")
-        
-        root = fetch_timetable_hour(dep_info['eva'], datum, stunde, client_id, api_key)
-        if root is None: continue
-            
-        for train in root.findall('.//s'):
-            dp = train.find('dp')
-            if dp is None: continue
-            
-            path = dp.get('ppth', '')
-            if dest_info['name'].lower() in path.lower() or destination.lower() in path.lower():
-                pt = dp.get('pt', '')
-                if not pt: continue
-                
-                dep_dt = datetime.strptime(pt, "%y%m%d%H%M")
-                
-                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', '-'),
-                        "zug": f"{tl.get('c', '')} {tl.get('n', '')}",
-                        "_dt": dep_dt
-                    })
-
-    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
-
-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
-
-# --- Gradio UI ---
-with gr.Blocks(title="DB Mobile Fahrplan", theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 📱 DB Quick-Check")
-    
-    with gr.Accordion("⚙️ API-Konfiguration (Hier klicken)", open=False):
-        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")
-    
-    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]
-    )
-
-if __name__ == "__main__":
-    demo.launch()
-
-
-"""
-import gradio as gr
-import requests
-from datetime import datetime, timedelta
-import xml.etree.ElementTree as ET
-import json
-
 def 
     url = f"https://apis.deutschebahn.com/db-api-marketplace/apis/timetables/v1/station/{pattern}"
     headers = {
@@ -186,10 +42,6 @@ def fetch_timetable_hour(eva, date_str, hour_str, client_id, api_key):
     return None
 
 def search_next_3_connections(departure, destination, client_id, api_key):
-   
-    
-    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)
@@ -401,4 +253,3 @@ 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__":
-"""