Update pages/Drivers_in_Traffic.py

pages/Drivers_in_Traffic.py

@@ -1,22 +1,22 @@
 import streamlit as st
 from repo_directory import Drivers_in_Traffic
-from repo_directory import button
+from repo_directory import button, utils
 
 st.write('Takes more than 5 minutes.')
 
 YEAR_SELECTED = st.selectbox(
     'Select year',
-    (2023, 2022, 2021, 2020, 2019, 2018))
+    (2024, 2023, 2022, 2021, 2020, 2019, 2018))
 
 
 
 RACE_SELECTED = st.selectbox(
     'Select Race',
-    (1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23))
+    utils.get_events(YEAR_SELECTED))
 
 SESSION = st.selectbox(
     'Select Session',
-    ('R', 'SQ', 'FP1', 'FP2', 'FP3', 'Q', 'SS'))
+    utils.get_events(YEAR_SELECTED, RACE_SELECTED))
 
 laps, f1session, drivers, driver_colors = Drivers_in_Traffic.get_data(YEAR_SELECTED, RACE_SELECTED, SESSION)