Update app.py

app.py

@@ -86,7 +86,8 @@ def generate_heatmap_for_site(site_name, df):
         "Red": [255, 0, 0]
     }
 
-    site_df["Fault_Level"] = site_df["Fault_Count__c"].apply(get_fault_level)
+    print(df.columns)  # Add this to see the available columns
+    site_df["Fault_Level"] = site_df["Faults_Count"].apply(get_fault_level)
     site_df["color"] = site_df["Fault_Level"].map(color_map)
 
     # Create a Pydeck map for the site
@@ -129,7 +130,7 @@ def generate_heatmap_for_site(site_name, df):
         layers=[layer],
         tooltip=tooltip
     )
-
+print(df.columns)
 # Divide into four columns (Hyderabad, Kurnool, Ballari, Gadwal)
 col1, col2 = st.columns(2)
 with col1: