Update app.py

app.py

@@ -319,19 +319,19 @@ def main():
         st.plotly_chart(trend_fig, use_container_width=True)
 
     with tab4:
-    # 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.")
+        # 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__":