Upload app.py

src/app.py

@@ -234,12 +234,14 @@ async def get_vehicle_view(vehicle_id: str):
 @app.get("/api/stop/{stop_code}")
 async def get_stop_view(stop_code: str):
     # 1. Translate Pole Number to Database ID
-    stop_info = db.execute("SELECT stop_id, stop_name FROM stops WHERE CAST(stop_code AS VARCHAR) = ? LIMIT 1", [str(stop_code)]).fetchone()
+    stop_info = db.execute("SELECT stop_id, stop_name, stop_lat, stop_lon FROM stops WHERE CAST(stop_code AS VARCHAR) = ? LIMIT 1", [str(stop_code)]).fetchone()
     if not stop_info:
         return {"error": "Stop code not found"}
     
     target_id = str(stop_info[0])
     stop_name = stop_info[1]
+    stop_lat = stop_info[2]
+    stop_lon = stop_info[3]
 
     # 2. Get the Cache structure (dict with vehicles, predictions, alerts)
     cached_data = await ttc_cache.get_data()
@@ -295,7 +297,15 @@ async def get_stop_view(stop_code: str):
                     })
 
     arrivals.sort(key=lambda x: x['eta_mins'])
-    return {"stop_name": stop_name, "stop_code": stop_code, "arrivals": arrivals}
+    return {
+        "stop_name": stop_name,
+        "stop_code": stop_code,
+        "location": {
+            "lat": stop_lat,
+            "lon": stop_lon
+        },
+        "arrivals": arrivals
+    }
 
 @app.get("/api/alerts")
 async def get_all_alerts():