Update app.py

app.py

@@ -93,29 +93,42 @@ if selected_site in SITES:
     st.subheader("📈 Tilt vs. Vibration")
     st.scatter_chart(site_df[['Tilt Angle (°)', 'Vibration (g)']])
 
-    # Map with Red Alerts
-    st.subheader("📍 Red Alert Pole Locations")
-    red_df = site_df[site_df['Alert Level'] == 'Red']
-    if not red_df.empty:
-        st.pydeck_chart(pdk.Deck(
-            initial_view_state=pdk.ViewState(
-                latitude=SITES[selected_site][0],
-                longitude=SITES[selected_site][1],
-                zoom=12,
-                pitch=50
-            ),
-            layers=[
-                pdk.Layer(
-                    'ScatterplotLayer',
-                    data=red_df,
-                    get_position='[Longitude, Latitude]',
-                    get_color='[255, 0, 0, 160]',
-                    get_radius=100,
-                )
-            ]
-        ))
+# Map showing all poles by alert level color
+st.subheader("📍 All Pole Locations with Alert Levels")
+
+# Map color logic based on Alert Level
+def get_color(alert):
+    if alert == 'Green':
+        return [0, 255, 0, 160]
+    elif alert == 'Yellow':
+        return [255, 255, 0, 160]
+    elif alert == 'Red':
+        return [255, 0, 0, 160]
     else:
-        st.info("No red alerts at this time.")
+        return [128, 128, 128, 160]  # default gray
+
+site_df['color'] = site_df['Alert Level'].apply(get_color)
+
+st.pydeck_chart(pdk.Deck(
+    initial_view_state=pdk.ViewState(
+        latitude=SITES[selected_site][0],
+        longitude=SITES[selected_site][1],
+        zoom=12,
+        pitch=50
+    ),
+    layers=[
+        pdk.Layer(
+            'ScatterplotLayer',
+            data=site_df,
+            get_position='[Longitude, Latitude]',
+            get_color='color',
+            get_radius=100,
+            pickable=True,
+        )
+    ],
+    tooltip={"text": "Pole ID: {Pole ID}\nAlert: {Alert Level}\nHealth: {Health Score}"}
+))
+
 
 else:
     st.warning("Invalid site. Please enter one of: Hyderabad, Gadwal, Kurnool, Ballari")