import streamlit as st
from repo_directory import GG_Plot
from repo_directory import button, utils

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

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

SESSION = st.selectbox(
    'Select Session',
    utils.get_sessions(YEAR_SELECTED, RACE_SELECTED))



laps, f1session, drivers = GG_Plot.get_data(YEAR_SELECTED, RACE_SELECTED, SESSION)   


DRIVERS_SELECTED = st.multiselect(
    'Select Drivers to compare',
    drivers)

GG_Plot.plot(DRIVERS_SELECTED, laps, f1session, SESSION)