import streamlit as st
from repo_directory import PitStops
from repo_directory import button
import datetime

YEAR_SELECTED = st.selectbox(
    'Select Year',
    (2023, 2022, 2021, 2020, 2019, 2018))



season_events, events_list = PitStops.get_season_events(YEAR_SELECTED)

RACE_SELECTED =  st.selectbox(
    'Select Race',
    events_list)

event_id = PitStops.get_event_id(season_events, RACE_SELECTED)
df = PitStops.get_pitstops(event_id)

PitStops.plot_event_pitstops(df, RACE_SELECTED)

PitStops.plot_full_season_median(YEAR_SELECTED)