Update modules/visuals.py

modules/visuals.py

@@ -20,9 +20,12 @@ def display_map_heatmap(df, location):
         st.warning("No data available for this location.")
         return
     
-    # Map AlertLevel to colors
+    # Map AlertLevel to sizes and styles
     df = df.copy()
     df["AlertColor"] = df["AlertLevel"].map({"Green": "green", "Yellow": "yellow", "Red": "red"})
+    df["MarkerSize"] = df["AlertLevel"].map({"Green": 10, "Yellow": 15, "Red": 20})
+    df["MarkerSymbol"] = df["AlertLevel"].map({"Green": "circle", "Yellow": "circle", "Red": "star"})
+    df["CustomClass"] = df["AlertLevel"].apply(lambda x: "blinking-red" if x == "Red" else "")
     
     # Create scatter map
     fig = px.scatter_mapbox(
@@ -31,8 +34,9 @@ def display_map_heatmap(df, location):
         lon="Longitude",
         color="AlertLevel",
         color_discrete_map={"Green": "green", "Yellow": "yellow", "Red": "red"},
-        size_max=15,
-        zoom=10,
+        size="MarkerSize",
+        size_max=25,
+        zoom=11,
         hover_data={
             "PoleID": True,
             "RFID": True,
@@ -45,6 +49,13 @@ def display_map_heatmap(df, location):
         title=f"Pole Alert Map - {location}",
         height=600
     )
+    fig.update_traces(
+        marker=dict(
+            symbol=df["MarkerSymbol"],
+            opacity=0.8,
+            line=dict(width=2, color="white") if df["AlertLevel"].eq("Red").any() else None
+        )
+    )
     fig.update_layout(
         mapbox_style="open-street-map",
         margin={"r":0,"t":50,"l":0,"b":0},