Update app.py

app.py

@@ -106,7 +106,33 @@ col3.metric("Energy", int(site_df['Energy'].sum()))
 
 
     # Table View
+   # Check if the selected site exists (ensure indentation here is consistent)
+if selected_site:
+    # Display the subheader with the selected site name
     st.subheader(f"📋 Pole Data Table for {selected_site}")
+
+    # Add the rest of your logic here if needed for the table
+    # For example, showing a data table for the selected site
+    st.dataframe(site_df)
+
+    # Other code related to displaying the metrics can go here too
+    col1, col2, col3 = st.columns(3)
+
+    col1.metric("Total Poles", len(site_df))
+
+    # Safely access 'AlertLevel' and count red alerts
+    if 'AlertLevel' in site_df.columns:
+        red_alerts = site_df[site_df['AlertLevel'] == 'Red'].shape[0]
+    else:
+        red_alerts = 0
+
+    col2.metric("Red Alerts", red_alerts)
+
+    # If you have other metrics like energy:
+    col3.metric("Energy", int(site_df['Energy'].sum()))
+else:
+    st.warning("Please select a valid site.")
+
     with st.expander("Filter Options"):
         alert_filter = st.multiselect("Alert Level", options=site_df['Alert Level'].unique(), default=site_df['Alert Level'].unique())
         camera_filter = st.multiselect("Camera Status", options=site_df['Camera Status'].unique(), default=site_df['Camera Status'].unique())