F1-analysis / pages /Drivers_in_Traffic.py
tracinginsights's picture
Update pages/Drivers_in_Traffic.py
2585fc7 verified
import streamlit as st
from repo_directory import Drivers_in_Traffic
from repo_directory import button, utils
st.write('Takes more than 5 minutes.')
YEAR_SELECTED = st.selectbox(
'Select year',
(2024, 2023, 2022, 2021, 2020, 2019, 2018))
RACE_SELECTED = st.selectbox(
'Select Race',
utils.get_events(YEAR_SELECTED))
SESSION = st.selectbox(
'Select Session',
utils.get_events(YEAR_SELECTED, RACE_SELECTED))
laps, f1session, drivers, driver_colors = Drivers_in_Traffic.get_data(YEAR_SELECTED, RACE_SELECTED, SESSION)
DRIVERS_SELECTED = st.multiselect(
'Select Drivers',
drivers,
)
tel = Drivers_in_Traffic.get_tel(laps, driver_colors)
Drivers_in_Traffic.plot(tel, DRIVERS_SELECTED)