Update modules/visuals.py

modules/visuals.py

@@ -19,20 +19,25 @@ def display_charts(df):
 def display_heatmap(df):
     # Map AlertLevel to numeric values for heatmap intensity
     alert_map = {"Green": 0, "Yellow": 1, "Red": 2}
+    df = df.copy()  # Avoid modifying the original DataFrame
     df["AlertValue"] = df["AlertLevel"].map(alert_map)
     
-    # Pivot table: Locations as rows, Poles as columns
+    # Pivot table for heatmap values (AlertValue)
     pivot_df = df.pivot_table(index="Location", columns="PoleID", values="AlertValue", fill_value=0)
     
-    # Create hover text with PoleID, AlertLevel, and Anomalies
-    hover_df = df.pivot_table(index="Location", columns="PoleID", values=["PoleID", "AlertLevel", "Anomalies"], aggfunc="first")
-    hover_text = pivot_df.copy()
+    # Create hover text DataFrame
+    hover_data = df[["Location", "PoleID", "AlertLevel", "Anomalies"]].copy()
+    hover_text = pivot_df.copy().astype(str)
     for loc in pivot_df.index:
         for pole in pivot_df.columns:
-            if pole in hover_df.loc[loc, "PoleID"].columns:
-                alert = hover_df.loc[loc, ("AlertLevel", pole)] if pd.notna(hover_df.loc[loc, ("AlertLevel", pole)]) else "None"
-                anomalies = hover_df.loc[loc, ("Anomalies", pole)] if pd.notna(hover_df.loc[loc, ("Anomalies", pole)]) else "None"
+            # Find matching row in hover_data
+            match = hover_data[(hover_data["Location"] == loc) & (hover_data["PoleID"] == pole)]
+            if not match.empty:
+                alert = match["AlertLevel"].iloc[0]
+                anomalies = match["Anomalies"].iloc[0]
                 hover_text.loc[loc, pole] = f"Pole: {pole}<br>Alert: {alert}<br>Anomalies: {anomalies}"
+            else:
+                hover_text.loc[loc, pole] = f"Pole: {pole}<br>Alert: None<br>Anomalies: None"
     
     # Create heatmap using Plotly
     fig = px.imshow(