Update app.py

forgot to grab vehicle locations

app.py

@@ -26,7 +26,6 @@ def getRouteList():
         'Connection': 'keep-alive',
         'Upgrade-Insecure-Requests': '1'
     }
-
     try:
         response = requests.get(url, headers=headers, timeout=10)
         response.raise_for_status()
@@ -65,7 +64,6 @@ def getRouteList():
             
             # Replace dash between number and name with space (e.g., "26-Dupont" -> "26 Dupont")
             text = re.sub(r'(\d+)-([A-Za-z])', r'\1 \2', text)
-            if not 'Night Bus' in text: continue
 
             routes[rid] = text
 
@@ -84,7 +82,6 @@ def getRouteList():
             text = re.sub(r'(\d+)-([A-Za-z])', r'\1 \2', text)
 
             routes[rid] = text
-
     return jsonify(routes)
 
 @app.route('/listvehiclesbyroute', methods=['POST'])
@@ -115,6 +112,9 @@ def listVehiclesByRoute():
         return jsonify({"error": "No vehicles found"}), 404
 
     vehicles = []
+    all_popups = soup.find_all("div", class_="leaflet-popup-content")
+
+
     for p in core_div.find_all("p", id=True):
         vehicle_id = p.find("a").get_text(strip=True) if p.find("a") else None
         spans = p.find_all("span")
@@ -153,12 +153,36 @@ def listVehiclesByRoute():
             # If it's mm:ss format, keep as is
         
         delay = ("-" if late else "+") + delay_value
+
+        # Extract coordinates from JavaScript AddMarker calls
+        lat, lon = None, None
+        
+        # Find all script tags and look for AddMarker calls
+        scripts = soup.find_all("script")
+        for script in scripts:
+            if script.string and "AddMarker" in script.string:
+                # Split the script content by AddMarker calls and find the one with our vehicle ID
+                add_marker_calls = script.string.split("AddMarker(")
+                for call in add_marker_calls[1:]:  # Skip the first split which is before AddMarker
+                    if f'"{vehicle_id}"' in call:
+                        # Extract coordinates from the first part [lat, lon]
+                        coords_match = re.search(r'\[([0-9\.\-]+),\s*([0-9\.\-]+)\]', call)
+                        if coords_match:
+                            lat, lon = float(coords_match.group(1)), float(coords_match.group(2))
+                            print(f"Found coordinates for vehicle {vehicle_id}: lat={lat}, lon={lon}")
+                            break
+                if lat is not None:
+                    break
+        sneakygolem = direction.split(' ')
+        branch = sneakygolem[-1] if len(sneakygolem) > 1 else ""
         vehicles.append({
             "vehicle_id": vehicle_id,
             "delay": delay,
             "time": time,
-            "branch": direction.split(' ')[-1],
-            "destination": destination
+            "branch": branch,
+            "destination": destination,
+            "latitude": lat,
+            "longitude": lon
         })
 
     return jsonify({