Update app.py

app.py

@@ -28,17 +28,16 @@ def simulate_pole(pole_id, site_name):
     total_power = solar_kwh + wind_kwh
     power_status = 'Sufficient' if total_power >= power_required else 'Insufficient'
 
-    tilt_angle = round(random.uniform(0, 45), 2)
     vibration = round(random.uniform(0, 5), 2)
     camera_status = random.choice(['Online', 'Offline'])
 
     alert_level = 'Green'
-    if tilt_angle > 30 or vibration > 3:
+    if vibration > 3:
         alert_level = 'Yellow'
-    if tilt_angle > 40 or vibration > 4.5:
+    if vibration > 4.5:
         alert_level = 'Red'
 
-    health_score = max(0, 100 - (tilt_angle + vibration * 10))
+    health_score = max(0, 100 - ( vibration * 10))
     timestamp = datetime.now() - timedelta(hours=random.randint(0, 6))
 
     return {
@@ -51,7 +50,6 @@ def simulate_pole(pole_id, site_name):
         'Power Required (kWh)': power_required,
         'Total Power (kWh)': total_power,
         'Power Status': power_status,
-        'Tilt Angle (°)': tilt_angle,
         'Vibration (g)': vibration,
         'Camera Status': camera_status,
         'Health Score': round(health_score, 2),
@@ -86,12 +84,33 @@ if selected_site in SITES:
     filtered_df = site_df[(site_df['Alert Level'].isin(alert_filter)) & (site_df['Camera Status'].isin(camera_filter))]
     st.dataframe(filtered_df, use_container_width=True)
 
-    # Charts
-    st.subheader("📊 Energy Generation Comparison")
-    st.bar_chart(site_df[['Solar (kWh)', 'Wind (kWh)']].mean())
-
-    st.subheader("📈 Tilt vs. Vibration")
-    st.scatter_chart(site_df[['Tilt Angle (°)', 'Vibration (g)']])
+st.subheader("🔋 Energy Generation per Pole")
+
+# Reshape the data for comparison: one row per Pole ID and energy source
+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'
+)
+
+# Plot a grouped bar chart using Altair
+import altair as alt
+
+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'),
+    color='Energy Source:N',
+    tooltip=['Pole ID', 'Energy Source', 'kWh']
+).properties(
+    width=800,
+    height=400
+).configure_axisX(labelAngle=45)
+
+st.altair_chart(bar_chart, use_container_width=True)
+
+    st.subheader("📈 Vibration")
+    st.scatter_chart(site_df[[ 'Vibration (g)']])
 
 # Map showing all poles by alert level color
 st.subheader("📍 All Pole Locations with Alert Levels")