Upload tools/vélib_disponibilites.py with huggingface_hub

tools/vélib_disponibilites.py

@@ -0,0 +1,104 @@
+import gradio as gr
+import json
+from PIL import Image
+import numpy as np
+
+# --- User Defined Logic ---
+import requests
+import difflib
+from typing import Optional
+
+def velib_disponibilites(query: str, token: str, commune: Optional[str] = None, max_results: int = 10) -> str:
+    """Search Vélib stations by approximate name and return real‑time availability.
+
+    Args:
+        query: Partial station name to search for.
+        token: API token for the PRIM marketplace.
+        commune: Optional city/commune name to filter stations.
+        max_results: Maximum number of stations to return.
+
+    Returns:
+        Human‑readable string describing each matching station and its current
+        bike/dock availability. Returns an error message if the data cannot be
+        retrieved or no stations match.
+    """
+    base_url = "https://prim.iledefrance-mobilites.fr/marketplace/velib"
+    headers = {"Authorization": f"Bearer {token}"} if token else {}
+
+    try:
+        info_resp = requests.get(f"{base_url}/station_information.json", headers=headers, timeout=10)
+        info_resp.raise_for_status()
+        stations_info = info_resp.json()
+
+        status_resp = requests.get(f"{base_url}/station_status.json", headers=headers, timeout=10)
+        status_resp.raise_for_status()
+        stations_status = status_resp.json()
+    except requests.RequestException as e:
+        return f"Erreur lors de la récupération des données Vélib : {e}"
+
+    stations_data = stations_info.get('data', {}).get('stations', [])
+    status_data = stations_status.get('data', {}).get('stations', [])
+
+    stations_dict = {s['station_id']: s for s in stations_data}
+
+    for status in status_data:
+        sid = status['station_id']
+        if sid in stations_dict:
+            stations_dict[sid].update({
+                'num_bikes_available': status.get('num_bikes_available', 0),
+                'num_bikes_available_mechanical': status.get('num_bikes_available_types', {}).get('mechanical', 0),
+                'num_bikes_available_ebike': status.get('num_bikes_available_types', {}).get('ebike', 0),
+                'num_docks_available': status.get('num_docks_available', 0),
+                'last_reported': status.get('last_reported', '')
+            })
+
+    stations = list(stations_dict.values())
+
+    if commune:
+        commune_lower = commune.lower()
+        stations = [s for s in stations if commune_lower in s.get('address', '').lower()]
+        if not stations:
+            return f"Aucune station trouvée à '{commune}'."
+
+    station_names = [s.get('name', '') for s in stations if s.get('name')]
+    matches = difflib.get_close_matches(query, station_names, n=max_results, cutoff=0.3)
+
+    if not matches:
+        matches = [name for name in station_names if query.lower() in name.lower()][:max_results]
+
+    if not matches:
+        return f"Aucune station ne correspond à la recherche '{query}'."
+
+    lines = []
+    for name in matches:
+        station = next(s for s in stations if s.get('name') == name)
+        total = station.get('num_bikes_available', 0)
+        mech = station.get('num_bikes_available_mechanical', 0)
+        elec = station.get('num_bikes_available_ebike', 0)
+        docks = station.get('num_docks_available', 0)
+        address = station.get('address', '')
+        address_parts = [p.strip() for p in address.split(',')]
+        commune_name = address_parts[-1] if address_parts else 'Non spécifiée'
+
+        line = (
+            f"🚲 Station: {station.get('name', 'Inconnu')}\n"
+            f"📍 Adresse: {address} ({commune_name})\n"
+            f"🗺️  Coordonnées: {station.get('lat', 'N/A')}, {station.get('lon', 'N/A')}\n"
+            f"💺 Capacité totale: {station.get('capacity', 'N/A')}\n"
+            f"🚴 Vélos disponibles: {total} (dont {mech} mécaniques, {elec} électriques)\n"
+            f"🔌 Bornettes libres: {docks}\n"
+            f"⏰ Dernière mise à jour: {station.get('last_reported', 'Inconnue')}\n"
+        )
+        lines.append(line)
+
+    return "\n".join(lines)
+
+# --- Interface Factory ---
+def create_interface():
+    return gr.Interface(
+        fn=velib_disponibilites,
+        inputs=[gr.Textbox(label=k) for k in ['query', 'token']],
+        outputs=gr.Textbox(label="Formatted list of matching Vélib stations with real‑time availability"),
+        title="vélib_disponibilites",
+        description="Auto-generated tool: vélib_disponibilites"
+    )