import streamlit as st
from repo_directory import LapTime_Difference
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))

SESSION = st.selectbox(
    'Select Session',
    ('R','FP1', 'FP2', 'FP3','SS', 'Q', 'SQ', ))


laps, drivers = LapTime_Difference.get_data(YEAR_SELECTED, RACE_SELECTED, SESSION)   


DRIVER_1 = st.selectbox(
    'Select Driver 1',
    drivers)

DRIVER_2 = st.selectbox(
    'Select Driver 2',
    drivers)



LapTime_Difference.plot(laps, DRIVER_1, DRIVER_2)