Update modules/visuals.py

modules/visuals.py

@@ -1,35 +1,44 @@
 import streamlit as st
-import pandas as pd
 import plotly.express as px
 
-def display_summary(df: pd.DataFrame):
+def display_dashboard(df):
     st.subheader("📊 System Summary")
-    col1, col2, col3, col4 = st.columns(4)
+    col1, col2, col3 = st.columns(3)
     col1.metric("Total Poles", df.shape[0])
-    col2.metric("Red Alerts", df[df["AlertLevel"] == "Red"].shape[0])
-    col3.metric("Power Issues", df[df["PowerSufficient"] == "No"].shape[0])
-    col4.metric("Offline Cameras", df[df["CameraStatus"] == "Offline"].shape[0])
+    col2.metric("🚨 Red Alerts", df[df["Alert Level"] == "Red"].shape[0])
+    col3.metric("⚡ Power Issues", df[df["Power Sufficient"] == "No"].shape[0])
 
-def display_tel_map(df: pd.DataFrame):
-    st.subheader("🗺️ Telangana Pole Heatmap")
-    red_df = df[df["AlertLevel"] == "Red"]
+def display_map_with_alerts(df):
     fig = px.scatter_mapbox(
-        red_df,
-        lat="Location_Latitude",
-        lon="Location_Longitude",
-        color="AlertLevel",
-        hover_name="PoleID",
-        zoom=6.3,
-        height=500,
-        mapbox_style="carto-positron"
+        df,
+        lat="Latitude",
+        lon="Longitude",
+        color="Alert Level",
+        hover_name="Pole ID",
+        zoom=6.2,
+        mapbox_style="carto-positron",
+        height=500
     )
     st.plotly_chart(fig, use_container_width=True)
 
-def display_energy_trends(df: pd.DataFrame):
-    st.subheader("⚙️ Solar vs Wind")
-    st.plotly_chart(px.bar(df, x="PoleID", y=["SolarGen(kWh)", "WindGen(kWh)"], barmode="group"))
+    # Blinking red poles (HTML)
+    red_df = df[df["Alert Level"] == "Red"]
+    if not red_df.empty:
+        st.markdown("### 🔴 Blinking Red Alerts (Hub Notification)")
+        for _, row in red_df.iterrows():
+            st.markdown(
+                f"<div style='padding:8px;background:#ffe6e6;animation:blink 1s infinite;'>"
+                f"<b>{row['Pole ID']}</b> in <b>{row['Site']}</b>: {row['Anomalies']}</div>",
+                unsafe_allow_html=True
+            )
+        st.markdown("""
+            <style>
+            @keyframes blink {
+              50% { background-color: #ff4d4d; }
+            }
+            </style>
+        """, unsafe_allow_html=True)
 
-def display_scatter(df: pd.DataFrame):
-    st.subheader("📉 Tilt vs Vibration")
-    fig = px.scatter(df, x="Tilt(°)", y="Vibration(g)", color="AlertLevel", hover_data=["PoleID", "Site"])
-    st.plotly_chart(fig)
+def display_charts(df):
+    st.bar_chart(df.set_index("Pole ID")[["SolarGen(kWh)", "WindGen(kWh)"]])
+    st.scatter_chart(df.rename(columns={"Tilt(°)": "Tilt", "Vibration(g)": "Vibration"}).set_index("Pole ID")[["Tilt", "Vibration"]])