Update app.py

app.py

@@ -277,7 +277,7 @@ def main():
         df['Weather'] = 'Unknown'
     
     # Create tabs for different visualizations
-    tab1, tab2, tab3, tab4 = st.tabs(["Crash Statistics", "Crash Map", "Crash Trend", "Crash Injuries/Fatalities","Incidents Distribution"])
+    tab1, tab2, tab3, tab4, tab5 = st.tabs(["Crash Statistics", "Crash Map", "Crash Trend", "Crash Injuries/Fatalities","Distribution by Category"])
     
     with tab1:
         # Age group selection
@@ -351,7 +351,28 @@ def main():
         injuries_fatalities_chart = create_injuries_fatalities_chart(df, unit_type)
         st.altair_chart(injuries_fatalities_chart, use_container_width=True)
         st.markdown("#### TODO: add write-up for this 5th chart.")
-    
+        
+    with tab5:
+        # Dropdown for category selection
+        categories = [
+            'Collisionmanner',
+            'Lightcondition',
+            'Weather',
+            'SurfaceCondition',
+            'AlcoholUse_Drv1',
+            'Gender_Drv1',
+        ]
+        selected_category = st.selectbox("Select Category:", categories)
+
+        # Dropdown for year selection
+        years = ['All Years'] + sorted(df['Year'].dropna().unique().astype(int).tolist())
+        selected_year = st.selectbox("Select Year:", years)
+
+        # Generate and display the distribution chart
+        st.markdown(f"### Distribution of Incidents by {selected_category}")
+        distribution_chart = create_category_distribution_chart(df, selected_category, selected_year)
+        st.plotly_chart(distribution_chart, use_container_width=True)
+
 
 if __name__ == "__main__":
     main()