Update app.py

app.py

@@ -63,64 +63,85 @@ st.plotly_chart(px.bar(
 # --- Alert Level Breakdown Chart ---
 display_charts(filtered_df)
 
-# --- Heatmap Section ---
-st.subheader("🌡 Pole Fault Heatmap")
-
-# Map Color based on Faults
-def get_fault_level(fault_count):
-    if fault_count <= 1:
-        return "Green"
-    elif 2 <= fault_count <= 3:
-        return "Yellow"
-    else:
-        return "Red"
-
-filtered_df["Fault_Level"] = filtered_df["Fault_Count__c"].apply(get_fault_level)
-
-color_map = {
-    "Green": [0, 255, 0],
-    "Yellow": [255, 255, 0],
-    "Red": [255, 0, 0]
-}
-filtered_df["color"] = filtered_df["Fault_Level"].map(color_map)
-
-# Pydeck Map
-layer = pdk.Layer(
-    "ScatterplotLayer",
-    data=filtered_df,
-    get_position='[Longitude__c, Latitude__c]',
-    get_color="color",
-    get_radius=80,
-    pickable=True,
-    auto_highlight=True
-)
 
-view_state = pdk.ViewState(
-    latitude=filtered_df["Location_Latitude__c"].mean(),
-    longitude=filtered_df["Location_Longitude__c"].mean(),
-    zoom=10,
-    pitch=40
-)
+# Fetch the raw data from Salesforce
+df = fetch_poles()
 
-tooltip = {
-    "html": """
-    <b>Pole Name:</b> {Name}<br/>
-    <b>Site:</b> {Site__c}<br/>
-    <b>Fault Count:</b> {Fault_Count__c}<br/>
-    <b>Alert Level:</b> {Alert_Level__c}<br/>
-    <b>RFID Tag:</b> {RFID_Tag__c}<br/>
-    <b>Tilt:</b> {Tilt__c}<br/>
-    <b>Vibration:</b> {Vibration__c}
-    """,
-    "style": {
-        "backgroundColor": "steelblue",
-        "color": "white"
+# Define function to generate heatmap based on site
+def generate_heatmap_for_site(site_name, df):
+    site_df = df[df['Site__c'] == site_name]
+
+    # Define color mapping for fault levels
+    def get_fault_level(fault_count):
+        if fault_count <= 1:
+            return "Green"
+        elif 2 <= fault_count <= 3:
+            return "Yellow"
+        else:
+            return "Red"
+
+    color_map = {
+        "Green": [0, 255, 0],
+        "Yellow": [255, 255, 0],
+        "Red": [255, 0, 0]
     }
-}
 
-st.pydeck_chart(pdk.Deck(
-    map_style="mapbox://styles/mapbox/dark-v10",
-    initial_view_state=view_state,
-    layers=[layer],
-    tooltip=tooltip
-))
+    site_df["Fault_Level"] = site_df["Fault_Count__c"].apply(get_fault_level)
+    site_df["color"] = site_df["Fault_Level"].map(color_map)
+
+    # Create a Pydeck map for the site
+    layer = pdk.Layer(
+        "ScatterplotLayer",
+        data=site_df,
+        get_position='[Longitude__c, Latitude__c]',
+        get_color="color",
+        get_radius=80,
+        pickable=True,
+        auto_highlight=True
+    )
+
+    view_state = pdk.ViewState(
+        latitude=site_df["Location_Latitude__c"].mean(),
+        longitude=site_df["Location_Longitude__c"].mean(),
+        zoom=10,
+        pitch=40
+    )
+
+    tooltip = {
+        "html": """
+            <b>Pole Name:</b> {Name}<br>
+            <b>Site:</b> {Site__c}<br>
+            <b>Fault Count:</b> {Fault_Count__c}<br>
+            <b>Alert Level:</b> {Alert_Level__c}<br>
+            <b>RFID Tag:</b> {RFID_Tag__c}<br>
+            <b>Tilt:</b> {Tilt__c}<br>
+            <b>Vibration:</b> {Vibration__c}
+        """,
+        "style": {
+            "backgroundColor": "steelblue",
+            "color": "white"
+        }
+    }
+
+    return pdk.Deck(
+        map_style="mapbox://styles/mapbox/dark-v10",
+        initial_view_state=view_state,
+        layers=[layer],
+        tooltip=tooltip
+    )
+
+# Divide into four columns (Hyderabad, Kurnool, Ballari, Gadwal)
+col1, col2 = st.columns(2)
+with col1:
+    st.subheader("Hyderabad")
+    st.pydeck_chart(generate_heatmap_for_site("Hyderabad", df))
+
+    st.subheader("Kurnool")
+    st.pydeck_chart(generate_heatmap_for_site("Kurnool", df))
+
+with col2:
+    st.subheader("Ballari")
+    st.pydeck_chart(generate_heatmap_for_site("Ballari", df))
+
+    st.subheader("Gadwal")
+    st.pydeck_chart(generate_heatmap_for_site("Gadwal", df))