Update app.py

app.py

@@ -129,7 +129,7 @@ def create_map(df, selected_year):
     
     return m
 
-def create_injuries_fatalities_chart(crash_data):
+def create_injuries_fatalities_chart(crash_data, unit_type):
 
     # 5th visualization title
     st.header("5. Total Injuries and Fatalities by Month")
@@ -149,14 +149,14 @@ def create_injuries_fatalities_chart(crash_data):
     # Dropdown for Unit Type selection
     # st.sidebar.selectbox("Select Unit Type", options=['Total'] + crash_data['Unittype_One'].dropna().unique().tolist())  # previous location of dropdown in sidebar
     # unit_type = st.selectbox("Select Unit Type", options=['Total'] + crash_data['Unittype_One'].dropna().unique().tolist())
-    unit_type_pairs = set()
-    for _, row in crash_data[['Unittype_One', 'Unittype_Two']].dropna().iterrows():
-        if row['Unittype_One'] != 'Driverless' or row['Unittype_Two'] != 'Driverless':
-            pair = ' vs '.join(sorted([row['Unittype_One'], row['Unittype_Two']]))
-            unit_type_pairs.add(pair)
-    # unit_type_pairs = list(unit_type_pairs) # modified as below to sort the dropdown options in alphabetical order
-    unit_type_pairs = sorted(list(unit_type_pairs))
-    unit_type = st.selectbox("Select Unit Type Pair", options=['Total'] + unit_type_pairs)
+    # unit_type_pairs = set()
+    # for _, row in crash_data[['Unittype_One', 'Unittype_Two']].dropna().iterrows():
+    #     if row['Unittype_One'] != 'Driverless' or row['Unittype_Two'] != 'Driverless':
+    #         pair = ' vs '.join(sorted([row['Unittype_One'], row['Unittype_Two']]))
+    #         unit_type_pairs.add(pair)
+    # # unit_type_pairs = list(unit_type_pairs) # modified as below to sort the dropdown options in alphabetical order
+    # unit_type_pairs = sorted(list(unit_type_pairs))
+    # unit_type = st.selectbox("Select Unit Type Pair", options=['Total'] + unit_type_pairs)
 
     # Filter data based on the selected unit type
     if unit_type == 'Total':
@@ -243,30 +243,6 @@ def create_injuries_fatalities_chart(crash_data):
     
     return line_chart
 
-def create_crash_trend_chart(df, weather=None):
-    if weather and weather != 'All Conditions':
-        df = df[df['Weather'] == weather]
-    
-    # Group data by year and count unique Incident IDs
-    trend_data = df.groupby('Year')['Incidentid'].nunique().reset_index()
-    trend_data.columns = ['Year', 'Crash Count']
-    
-    # Create line graph
-    fig = px.line(
-        trend_data,
-        x='Year',
-        y='Crash Count',
-        title=f'Crash Trend Over Time ({weather})',
-        labels={'Year': 'Year', 'Crash Count': 'Number of Unique Crashes'},
-        markers=True,
-        height=600
-    )
-    
-    fig.update_traces(line=dict(width=2), marker=dict(size=8))
-    fig.update_layout(legend_title_text='Trend')
-    
-    return fig
-
 def main():
     st.title('Traffic Crash Analysis')
     
@@ -277,7 +253,7 @@ def main():
         df['Weather'] = 'Unknown'
     
     # Create tabs for different visualizations
-    tab1, tab2, tab3 = st.tabs(["Crash Statistics", "Crash Map", "Crash Trend"])
+    tab1, tab2, tab3, tab4 = st.tabs(["Crash Statistics", "Crash Map", "Crash Trend", "Crash Injuries/Fatalities"])
     
     with tab1:
         # Age group selection
@@ -342,5 +318,21 @@ def main():
         trend_fig = create_crash_trend_chart(df, selected_weather)
         st.plotly_chart(trend_fig, use_container_width=True)
 
+    with tab3:
+    # Dropdown for Unit Type selection
+    unit_type_pairs = set()
+    for _, row in df[['Unittype_One', 'Unittype_Two']].dropna().iterrows():
+        if row['Unittype_One'] != 'Driverless' or row['Unittype_Two'] != 'Driverless':
+            pair = ' vs '.join(sorted([row['Unittype_One'], row['Unittype_Two']]))
+            unit_type_pairs.add(pair)
+    unit_type_pairs = sorted(list(unit_type_pairs))
+    unit_type = st.selectbox("Select Unit Type Pair", options=['Total'] + unit_type_pairs)
+
+    # Create 5th Visualization: Injuries and fatalities chart
+    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.")
+    
+
 if __name__ == "__main__":
     main()