Update app.py

app.py

@@ -17,7 +17,6 @@ SITES = {
     "Ballari": {"coords": [12.9716, 77.5946], "zone": "Solar Power Plant"}
 }
 
-# ---- Fixed Placement for Each Site ----
 def generate_fixed_pole_locations(base_lat, base_lon, num_poles):
     area_width = 0.01
     area_height = 0.01
@@ -30,7 +29,6 @@ def generate_fixed_pole_locations(base_lat, base_lon, num_poles):
         for i in range(rows) for j in range(cols)
     ][:num_poles]
 
-# ---- Simulate Pole Data ----
 def simulate_pole(pole_id, site_name, lat, lon):
     solar_kwh = round(random.uniform(3.0, 7.5), 2)
     wind_kwh = round(random.uniform(0.5, 2.0), 2)
@@ -88,13 +86,11 @@ if selected_site in SITES:
         df = pd.DataFrame(all_data)
         site_df = df[df['Site'] == selected_site]
 
-    # Summary
     col1, col2, col3 = st.columns(3)
     col1.metric("Total Poles", site_df.shape[0])
     col2.metric("Red Alerts", site_df[site_df['Alert Level'] == 'Red'].shape[0])
     col3.metric("Power Insufficiencies", site_df[site_df['Power Status'] == 'Insufficient'].shape[0])
 
-    # Table
     st.subheader(f"📋 Pole Data Table for {selected_site}")
     with st.expander("Filter Options"):
         alert_filter = st.multiselect("Alert Level", options=site_df['Alert Level'].unique(), default=site_df['Alert Level'].unique())
@@ -106,16 +102,13 @@ if selected_site in SITES:
     ]
     st.dataframe(filtered_df, use_container_width=True)
 
-    # Energy Chart
     st.subheader("🔋 Energy Generation per Pole")
-
     energy_long_df = site_df[['Pole ID', 'Solar (kWh)', 'Wind (kWh)']].melt(
         id_vars='Pole ID',
         value_vars=['Solar (kWh)', 'Wind (kWh)'],
         var_name='Energy Source',
         value_name='kWh'
     )
-
     bar_chart = alt.Chart(energy_long_df).mark_bar().encode(
         x=alt.X('Pole ID:N', sort=None, title='Pole ID'),
         y=alt.Y('kWh:Q'),
@@ -125,24 +118,8 @@ if selected_site in SITES:
         width=800,
         height=400
     ).configure_axisX(labelAngle=45)
-
     st.altair_chart(bar_chart, use_container_width=True)
 
-    # Fault Type Filter
-    st.subheader("⚠️ Map Filter: Select Fault Type(s)")
-    fault_options = ['High Vibration (>3g)', 'Camera Offline', 'Power Insufficient']
-    selected_faults = st.multiselect("Show poles with these fault conditions:", options=fault_options, default=fault_options)
-
-    def fault_condition(row):
-        return (
-            ('High Vibration (>3g)' in selected_faults and row['Vibration (g)'] > 3) or
-            ('Camera Offline' in selected_faults and row['Camera Status'] == 'Offline') or
-            ('Power Insufficient' in selected_faults and row['Power Status'] == 'Insufficient')
-        )
-
-    fault_df = site_df[site_df.apply(fault_condition, axis=1)] if selected_faults else site_df
-
-    # Map Color Logic
     def get_color(alert):
         if alert == 'Green':
             return [0, 255, 0, 160]
@@ -152,10 +129,9 @@ if selected_site in SITES:
             return [255, 0, 0, 160]
         return [128, 128, 128, 160]
 
-    fault_df['color'] = fault_df['Alert Level'].apply(get_color)
+    site_df['color'] = site_df['Alert Level'].apply(get_color)
 
-    # Map
-    st.subheader("📍 Pole Locations with Selected Faults")
+    st.subheader("📍 Pole Locations")
     st.pydeck_chart(pdk.Deck(
         initial_view_state=pdk.ViewState(
             latitude=SITES[selected_site]["coords"][0],
@@ -166,7 +142,7 @@ if selected_site in SITES:
         layers=[
             pdk.Layer(
                 'ScatterplotLayer',
-                data=fault_df,
+                data=site_df,
                 get_position='[Longitude, Latitude]',
                 get_color='color',
                 get_radius=100,