Update app.py

app.py

@@ -6,6 +6,12 @@ from bs4 import BeautifulSoup
 import tempfile
 import os
 
+from selenium import webdriver
+from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.common.by import By
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
+
 app = Flask(__name__)
 CORS(app)
 
@@ -193,6 +199,41 @@ def listVehiclesByRoute():
         "vehicles": vehicles
     })
 
+@app.route("/alerts", methods=["GET"])
+def serviceAlerts():
+    url = "https://www.transsee.ca/showmessages?a=ttc"
+
+    # Headless Chrome
+    options = Options()
+    options.add_argument("--headless=new")
+    options.add_argument("--disable-gpu")
+    options.add_argument("--no-sandbox")
+
+    driver = webdriver.Chrome(options=options)
+    driver.get(url)
+
+    # Wait until at least one HighAlert is loaded
+    WebDriverWait(driver, 15).until(
+        EC.presence_of_element_located((By.CSS_SELECTOR, "section.HighAlert"))
+    )
+
+    html = driver.page_source
+    driver.quit()
+
+    soup = BeautifulSoup(html, "html.parser")
+
+    alerts = []
+    for section in soup.select("section.HighAlert"):
+        title = section.find("b").get_text(strip=True) if section.find("b") else ""
+        route = section.find("div").get_text(strip=True) if section.find("div") else ""
+
+        alerts.append({
+            "title": title,
+            "route": route
+        })
+
+    return jsonify(alerts)
+
 @app.route('/', methods=['GET'])
 def health_check():
     return 'Backend is running!', 200