Update app.py

app.py

@@ -47,9 +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)
+    # 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],
@@ -64,16 +64,21 @@ def generate_heatmap_for_site(site_name, df):
     layer = pdk.Layer(
         "ScatterplotLayer",
         data=site_df,
-        get_position='[Longitude__c, Latitude__c]',
+        get_position='[Longitude__c, Latitude__c]',  # Ensure the column names are correct
         get_color="color",
-        get_radius=80, # You can adjust the radius if needed
+        get_radius=80,  # You can adjust the radius if needed
         pickable=True,
         auto_highlight=True
     )
 
+    # Handle potential missing coordinates for map center
+    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 for the map
     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
     )
@@ -92,7 +97,8 @@ def generate_heatmap_for_site(site_name, df):
             "color": "white"
         }
     }
-   # Return the heatmap
+
+    # Return the heatmap
     return pdk.Deck(
         map_style="mapbox://styles/mapbox/dark-v10",
         initial_view_state=view_state,