Update app.py

app.py

@@ -231,34 +231,40 @@ def main():
     tab1, tab2, tab3, tab4, tab5 = st.tabs(["Crash Statistics", "Crash Map", "Crash Trend", "Crash Injuries/Fatalities","Distribution by Category"])
     
     with tab1:
-        # Age group selection
-        age_groups = ['All Ages', '16-25', '26-35', '36-45', '46-55', '56-65', '65+']
-        selected_age = st.selectbox('Select Age Group:', age_groups)
-        
-        # Create and display the main violation distribution chart
-        fig, violations = create_violation_distribution_chart(df, selected_age)
-        
-        # Use plotly_events to capture click
-        selected_points = plotly_events(fig, click_event=True, hover_event=False, select_event=True)
-        
-        
-        # If user clicked on a bar, show severity distribution
-        if selected_points:
-            clicked_violation = violations.iloc[selected_points[0]['pointIndex']]['Violation']
-            severity_fig = create_severity_distribution_for_violation(df, clicked_violation)
-            st.plotly_chart(severity_fig, use_container_width=True)
-        
-        # Display total incidents info
-        if selected_age == 'All Ages':
-            total_incidents = len(df)
-        else:
-            total_incidents = len(df[
-                (df['Age_Group_Drv1'] == selected_age) | 
-                (df['Age_Group_Drv2'] == selected_age)
-            ])
-        
-        st.markdown(f"### Total Incidents for {selected_age}")
-        st.markdown(f"**{total_incidents:,}** incidents")
+    # Age group selection
+    age_groups = ['All Ages', '16-25', '26-35', '36-45', '46-55', '56-65', '65+']
+    selected_age = st.selectbox('Select Age Group:', age_groups)
+
+    # Create and display the main violation distribution chart
+    fig, violations = create_violation_distribution_chart(df, selected_age)
+
+    # Display the figure using plotly_events only (no extra st.plotly_chart)
+    selected_points = plotly_events(
+        fig,
+        click_event=True,
+        hover_event=False,
+        select_event=True,
+        key="violation_chart"
+    )
+
+    # If user clicked on a bar, show severity distribution
+    if selected_points:
+        clicked_violation = violations.iloc[selected_points[0]['pointIndex']]['Violation']
+        severity_fig = create_severity_distribution_for_violation(df, clicked_violation)
+        st.plotly_chart(severity_fig, use_container_width=True)
+
+    # Display total incidents info
+    if selected_age == 'All Ages':
+        total_incidents = len(df)
+    else:
+        total_incidents = len(df[
+            (df['Age_Group_Drv1'] == selected_age) | 
+            (df['Age_Group_Drv2'] == selected_age)
+        ])
+
+    st.markdown(f"### Total Incidents for {selected_age}")
+    st.markdown(f"**{total_incidents:,}** incidents")
+
 
     with tab2:
         years = sorted(df['Year'].unique())