pages/Tyre_Strategy.py

import streamlit as st
from repo_directory import Tyre_Strategy as ts
from repo_directory import button


YEAR_SELECTED = st.selectbox(
    'Select Year',
    (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))
SESSION = "RACE"

driver_stints, drivers_list = ts.get_data(YEAR_SELECTED, RACE_SELECTED)

ts.plot(driver_stints,drivers_list )