Update app.py

app.py

@@ -47,6 +47,9 @@ display_charts(filtered_df)
 def generate_heatmap_for_site(site_name, df):
     site_df = df[df['Site__c'] == site_name]
 
+     # Ensure that Alert_Level__c is treated as a string (for color mapping)
+    site_df['Alert_Level__c'] = site_df['Alert_Level__c'].astype(str)
+    
     # Define color mapping for alert levels
     color_map = {
         "Green": [0, 255, 0],
@@ -54,7 +57,7 @@ def generate_heatmap_for_site(site_name, df):
         "Red": [255, 0, 0]
     }
 
-    # Assign colors based on Alert Level
+    # Create a color column based on Alert_Level__c
     site_df["color"] = site_df["Alert_Level__c"].map(color_map)
 
     # Create a Pydeck map for the site
@@ -63,7 +66,7 @@ def generate_heatmap_for_site(site_name, df):
         data=site_df,
         get_position='[Longitude__c, Latitude__c]',
         get_color="color",
-        get_radius=80,
+        get_radius=80, # You can adjust the radius if needed
         pickable=True,
         auto_highlight=True
     )
@@ -89,7 +92,7 @@ def generate_heatmap_for_site(site_name, df):
             "color": "white"
         }
     }
-
+   # Return the heatmap
     return pdk.Deck(
         map_style="mapbox://styles/mapbox/dark-v10",
         initial_view_state=view_state,