Update app.py

app.py

@@ -267,6 +267,41 @@ def create_crash_trend_chart(df, weather=None):
     
     return fig
 
+def create_category_distribution_chart(df, selected_category, selected_year):
+    # Filter by selected year
+    if selected_year != 'All Years':
+        df = df[df['Year'] == int(selected_year)]
+
+    # Group by selected category and Injury Severity
+    grouped_data = df.groupby([selected_category, 'Injuryseverity']).size().reset_index(name='Count')
+
+    # Calculate percentages for each category value
+    total_counts = grouped_data.groupby(selected_category)['Count'].transform('sum')
+    grouped_data['Percentage'] = (grouped_data['Count'] / total_counts * 100).round(2)
+
+    # Create the stacked bar chart using Plotly
+    fig = px.bar(
+        grouped_data,
+        x=selected_category,
+        y='Count',
+        color='Injuryseverity',
+        text='Percentage',
+        title=f'Distribution of Incidents by {selected_category} ({selected_year})',
+        labels={'Count': 'Number of Incidents', selected_category: 'Category'},
+        height=600,
+    )
+
+    # Customize the chart appearance
+    fig.update_traces(texttemplate='%{text}%', textposition='inside')
+    fig.update_layout(
+        barmode='stack',
+        xaxis_tickangle=-45,
+        legend_title='Injury Severity',
+        margin=dict(t=50, b=100),
+    )
+
+    return fig
+
 def main():
     st.title('Traffic Crash Analysis')