Update app.py

app.py

@@ -78,8 +78,34 @@ def create_severity_violation_chart(df, age_group=None):
         barmode='stack'
     )
     
+    # return fig
+    return fig, violations # this return is needed for part3
+
+# added interactivity plot for part3
+def create_severity_detail_chart(violations, selected_violation):
+    # Filter data based on selected violation
+    filtered_data = violations[violations['Violation'] == selected_violation]
+    
+    # Create a detailed plot for the selected violation
+    fig = px.bar(
+        filtered_data,
+        x='Severity',
+        y='count',
+        color='Severity',
+        title=f'Severity Level Distribution for Violation: {selected_violation}',
+        labels={'count': 'Number of Incidents', 'Severity': 'Severity Level'},
+        height=400
+    )
+    
+    fig.update_layout(
+        xaxis_tickangle=-45,
+        legend_title='Severity Level',
+        barmode='stack'
+    )
+    
     return fig
 
+
 def get_top_violations(df, age_group):
     if age_group == 'All Ages':
         violations = pd.concat([
@@ -322,6 +348,17 @@ def main():
         # Create and display chart
         fig = create_severity_violation_chart(df, selected_age)
         st.plotly_chart(fig, use_container_width=True)
+
+        # Interactive selection for linking with the second chart --> for part3
+        selected_violation = st.selectbox(
+            "Select a Violation Type to View Detailed Severity Levels:",
+            options=sorted(violations['Violation'].unique())
+        )
+        
+        # Create and display the second detailed chart for the selected violation type --> for part3
+        if selected_violation:
+            detailed_fig = create_severity_detail_chart(violations, selected_violation)
+            st.plotly_chart(detailed_fig, use_container_width=True)
         
         # Display statistics
         if selected_age == 'All Ages':