pages/Drivers_in_Traffic.py

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',
    utils.get_years())



RACE_SELECTED = st.selectbox(
    'Select Race',
    utils.get_events(YEAR_SELECTED))

SESSION = st.selectbox(
    'Select Session',
    utils.get_sessions(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)