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

YEAR_SELECTED = st.selectbox(
    'Select Year',
    utils.get_years())

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

df = PitStops.get_pitstops(RACE_SELECTED, YEAR_SELECTED) #dhl pitstops

try:
    PitStops.plot_event_pitstops(df, RACE_SELECTED)
except:
    pass

try:    
    PitStops.plot_full_season_median(YEAR_SELECTED)
except:
    pass
    
try:
    race_names_df,  pit_stops_df, drivers_df =  PitStops.load_data() 
    
    ergast_pitstops, grandprix = PitStops.get_ergast_pitstops(pit_stops_df, drivers_df, race_names_df, YEAR_SELECTED, RACE_SELECTED)
    
    df_agg = PitStops.combine_dfs(ergast_pitstops, df)
except:
    pass

try:    
    PitStops.plot_event_ratings(df_agg, grandprix)
except:
    pass