Update app.py

app.py

@@ -28,10 +28,10 @@ def display_pole_data(pole_df):
 # Display Charts
 def display_charts(pole_df):
     st.subheader("⚙️ Energy Generation Trends")
-    st.bar_chart(pole_df[['Solar_Generation__c', 'Wind_Generation__c']].set_index('Pole_ID__c'))
+    st.bar_chart(pole_df[['Solar_Generation__c', 'Wind_Generation__c']].set_index('Pole__c'))
 
     st.subheader("📉 Tilt vs Vibration")
-    st.scatter_chart(pole_df[['Tilt__c', 'Vibration__c']].set_index('Pole_ID__c'))
+    st.scatter_chart(pole_df[['Tilt__c', 'Vibration__c']].set_index('Pole__c'))
 
 # Main function to run the dashboard
 def main():