Update modules/visuals.py

modules/visuals.py

@@ -1,24 +1,35 @@
-import pydeck as pdk
-import plotly.express as px
+import pandas as pd
+
+def display_dashboard(df: pd.DataFrame):
+    st.subheader("📊 System Summary")
+    col1, col2, col3, col4 = st.columns(4)
+    
+    col1.metric("Total Poles", df.shape[0])
+    col2.metric("🚨 Red Alerts", df[df["Alert_Level__c"] == "Red"].shape[0])
+    col3.metric("⚡ Power Issues", df[df["Power_Sufficient__c"] == "No"].shape[0])
+    col4.metric("📷 Offline Cameras", df[df["Camera_Status__c"] == "Offline"].shape[0])
+
 import streamlit as st
+import plotly.express as px
 
-# Function to generate the dashboard summary
-def display_dashboard(filtered_df):
-    st.title("VIEP Smart Poles Dashboard")
 
-    # Display system summary
-    col1, col2, col3 = st.columns(3)
-    col1.metric("Total Poles", filtered_df.shape[0])
-    col2.metric("Red Alerts", filtered_df[filtered_df['Alert_Level__c'] == 'Red'].shape[0])
-    col3.metric("Power Insufficiencies", filtered_df[filtered_df['Power_Status__c'] == 'Insufficient'].shape[0])
+def display_charts(df: pd.DataFrame):
+    fig_energy = px.bar(
+        df,
+        x="Name",
+        y=["Solar_Generation__c", "Wind_Generation__c"],
+    )
+    
 
-# Function to generate the alert level breakdown chart
-def display_charts(filtered_df):
-    st.subheader("Alert Level Breakdown")
-    alert_level_counts = filtered_df['Alert_Level__c'].value_counts()
-    fig = px.pie(names=alert_level_counts.index, values=alert_level_counts.values, title="Alert Levels")
-    st.plotly_chart(fig)
+    st.subheader("🚨 Alert Level Breakdown")
+    fig_alerts = px.histogram(
+        df,
+        x="Alert_Level__c",
+        title="Number of Poles by Alert Level"
+    )
+    st.plotly_chart(fig_alerts)
 
+import pydeck as pdk
 # Function to generate heatmap for a given site
 def generate_heatmap_for_site(site_name, df):
     site_df = df[df['Site__c'] == site_name]
@@ -75,4 +86,4 @@ def generate_heatmap_for_site(site_name, df):
         initial_view_state=view_state,
         layers=[layer],
         tooltip=tooltip
-    )
+    )