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


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


driver_stints, drivers_list, test, last_row, f1session = Tyre_Strategy.get_data(YEAR_SELECTED, RACE_SELECTED)

Tyre_Strategy.plot(driver_stints,drivers_list, test, last_row, f1session )