Update app.py

app.py

@@ -7,6 +7,17 @@ from streamlit_plotly_events import plotly_events
 from folium.plugins import HeatMap, MarkerCluster
 from streamlit_folium import st_folium
 
+# To fix the color scheme in crash stats plot
+severity_colors = {
+    "No Injury": "#1f77b4",
+    "Possible Injury": "#aec7e8",
+    "Non Incapacitating Injury": "#ff7f0e",
+    "Incapacitating Injury": "#ffbb78",
+    "Suspected Minor Injury": "#2ca02c",
+    "Suspected Serious Injury": "#98df8a",
+    "Fatal": "#d62728",
+}
+
 @st.cache_data
 def load_and_preprocess_data(file_path):
     # Read the data
@@ -71,7 +82,7 @@ def create_severity_violation_chart(df, age_group=None):
         title=f'Crash Severity Distribution by Violation Type - {age_group}',
         labels={'count': 'Number of Incidents', 'Violation': 'Violation Type'},
         height=600,
-        color_discrete_sequence=px.colors.qualitative.Set1 # --> for part 3
+        color_discrete_map=severity_colors # --> for part 3
     )
     
     fig.update_layout(
@@ -94,7 +105,8 @@ def create_interactive_pie_chart(violations, selected_violation):
         names='Severity',
         values='count',
         title=f'Severity Level Distribution for Violation: {selected_violation}',
-        height=400
+        height=400,
+        color_discrete_map=severity_colors
     )
     
     return fig