Update app.py

Add CORS, function to get vehicle location

app.py

@@ -1,4 +1,5 @@
 from flask import Flask, render_template, request, jsonify
+from flask_cors import CORS
 import easyocr
 import re
 import requests
@@ -7,7 +8,7 @@ import tempfile
 import os
 
 app = Flask(__name__)
-
+CORS(app)
 os.makedirs('/tmp/.EasyOCR', exist_ok=True)
 
 reader = easyocr.Reader(
@@ -18,6 +19,45 @@ reader = easyocr.Reader(
     user_network_directory='/tmp/.EasyOCR'   # ✅ Also tell it where user models are
 )
 
+def get_vehicle_locations_and_occupancy(vehicle_numbers):
+    url = "https://bustime.ttc.ca/gtfsrt/vehicles?debug"
+    headers = {
+        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/123.0.0.0 Safari/537.36',
+        'Referer': 'https://www.transsee.ca/',
+        'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8',
+        'Accept-Language': 'en-US,en;q=0.9',
+        'DNT': '1',  # Do Not Track
+        'Connection': 'keep-alive',
+        'Upgrade-Insecure-Requests': '1'
+    }
+    response = requests.get(url, headers=headers)
+    text = response.text
+
+    entities = text.split('entity {')[1:]  # skip the header
+
+    results = []
+    for entity in entities:
+        vehicle_id_matches = re.findall(r'vehicle\s*\{\s*id: "(\d+)"', entity)
+        if not vehicle_id_matches:
+            continue
+        vehicle_id = vehicle_id_matches[-1]
+        if vehicle_id not in vehicle_numbers:
+            continue
+
+        # Find latitude, longitude, occupancy_status
+        lat_match = re.search(r'latitude: (-?[\d\.]+)', entity)
+        lon_match = re.search(r'longitude: (-?[\d\.]+)', entity)
+        occ_match = re.search(r'occupancy_status: ([A-Z_]+)', entity)
+
+        results.append({
+            'vehicle_id': vehicle_id,
+            'latitude': float(lat_match.group(1)) if lat_match else None,
+            'longitude': float(lon_match.group(1)) if lon_match else None,
+            'occupancy_status': occ_match.group(1) if occ_match else None
+        })
+
+    return results
+
 @app.route('/', methods=['GET'])
 def health_check():
     return 'Backend is running!', 200
@@ -128,5 +168,22 @@ def seek():
         'vehicles': vehicles
     })
 
+@app.route('/vehicles', methods=['POST'])
+def vehicle_status():
+    if not request.is_json:
+        return jsonify({'error': 'Request must be JSON'}), 400
+    data = request.get_json()
+    vehicle_numbers = data.get('vehicle_numbers')
+    if not vehicle_numbers or not isinstance(vehicle_numbers, list):
+        return jsonify({'error': 'vehicle_numbers must be a list'}), 400
+
+    info = get_vehicle_locations_and_occupancy(vehicle_numbers)
+    return jsonify({'vehicles': info})
+
+#@app.route('/test_vehicle', methods=['GET'])
+#def test_vehicle():
+#    info = get_vehicle_locations_and_occupancy(['1217'])
+#    return jsonify(info)
+
 if __name__ == '__main__':
-    app.run(debug=True, port=5000)
+    app.run(debug=True, port=5000)