Update modules/visuals.py

modules/visuals.py

@@ -1,26 +1,31 @@
 import pandas as pd
+import streamlit as st
+import plotly.express as px
+import pydeck as pdk
 
+# Function to display system summary metrics
 def display_dashboard(df: pd.DataFrame):
     st.subheader("📊 System Summary")
     col1, col2, col3, col4 = st.columns(4)
     
+    # Metrics for Total Poles, Red Alerts, Power Issues, Offline Cameras
     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 display charts related to energy generation and alert levels
 def display_charts(df: pd.DataFrame):
+    # Energy Generation Bar Chart
     fig_energy = px.bar(
         df,
         x="Name",
         y=["Solar_Generation__c", "Wind_Generation__c"],
+        title="Energy Generation (Solar vs Wind)"
     )
-    
+    st.plotly_chart(fig_energy)
 
+    # Alert Level Breakdown Histogram
     st.subheader("🚨 Alert Level Breakdown")
     fig_alerts = px.histogram(
         df,
@@ -29,7 +34,6 @@ def display_charts(df: pd.DataFrame):
     )
     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]
@@ -47,6 +51,9 @@ def generate_heatmap_for_site(site_name, df):
     # Create a color column based on Alert_Level__c
     site_df["color"] = site_df["Alert_Level__c"].map(color_map)
 
+    # Drop rows with missing Longitude or Latitude to avoid errors in the map
+    site_df = site_df.dropna(subset=["Longitude__c", "Latitude__c"])
+
     # Create a Pydeck map for the site
     layer = pdk.Layer(
         "ScatterplotLayer",
@@ -58,13 +65,18 @@ def generate_heatmap_for_site(site_name, df):
         auto_highlight=True
     )
 
+    # Set the view state to center the map on the site
+    latitude = site_df["Latitude__c"].mean() if not site_df["Latitude__c"].isnull().all() else 0
+    longitude = site_df["Longitude__c"].mean() if not site_df["Longitude__c"].isnull().all() else 0
+
     view_state = pdk.ViewState(
-        latitude=site_df["Location_Latitude__c"].mean(),
-        longitude=site_df["Location_Longitude__c"].mean(),
+        latitude=latitude,
+        longitude=longitude,
         zoom=10,
         pitch=40
     )
 
+    # Tooltip for when you hover over a pole
     tooltip = {
         "html": """
             <b>Pole Name:</b> {Name}<br>
@@ -80,10 +92,10 @@ def generate_heatmap_for_site(site_name, df):
         }
     }
 
-    # Return the heatmap
+    # Return the heatmap as a Pydeck map
     return pdk.Deck(
         map_style="mapbox://styles/mapbox/dark-v10",
         initial_view_state=view_state,
         layers=[layer],
         tooltip=tooltip
-    )
+    )