Update modules/visuals.py

modules/visuals.py

@@ -1,32 +1,40 @@
 import streamlit as st
-import matplotlib.pyplot as plt
-import seaborn as sns
-import pandas as pd
 import pydeck as pdk
 
 def display_map(df):
-    st.subheader("🗺️ Smart Pole Locations Map")
+    st.subheader("📍 Smart Poles Location Map with Alerts")
 
-    # Basic map using st.map()
-    st.map(df[["Latitude", "Longitude"]])
+    alert_colors = {
+        "Green": [0, 255, 0],
+        "Yellow": [255, 255, 0],
+        "Red": [255, 0, 0]
+    }
+
+    df["color"] = df["Alert Level"].apply(lambda x: alert_colors.get(x, [0, 0, 0]))
 
-    # Optional: More detailed map using pydeck
     layer = pdk.Layer(
         "ScatterplotLayer",
         data=df,
         get_position='[Longitude, Latitude]',
-        get_color='[255, 0, 0, 160]',  # Red circles
-        get_radius=100,
+        get_color="color",
+        get_radius=4,
+        pickable=True,
     )
 
     view_state = pdk.ViewState(
         latitude=df["Latitude"].mean(),
         longitude=df["Longitude"].mean(),
-        zoom=6,
-        pitch=0,
+        zoom=8,
+        pitch=0
     )
 
-    st.pydeck_chart(pdk.Deck(layers=[layer], initial_view_state=view_state))
+    tooltip = {
+        "html": "<b>Pole:</b> {Pole ID} <br/><b>Site:</b> {Site} <br/><b>Alert:</b> {Alert Level} <br/><b>Anomalies:</b> {Anomalies}",
+        "style": {"backgroundColor": "steelblue", "color": "white"}
+    }
+
+    st.pydeck_chart(pdk.Deck(layers=[layer], initial_view_state=view_state, tooltip=tooltip))
+
 
 
 def display_dashboard(df):