# Filter out Safety car/VSC laps
import streamlit as st
from repo_directory.v2 import Average_Race_Pace
from repo_directory import button
from repo_directory.v2 import utils

import pandas as pd


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

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

SESSION_SELECTED = st.selectbox('Select Session', ('R','SR'))

_, df = Average_Race_Pace.plot(YEAR_SELECTED, RACE_SELECTED, SESSION_SELECTED,1.07)

st.dataframe(pd.DataFrame(df)[['Driver','LapTime','Diff','Team']])