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

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

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 = Corner_Analysis.get_data(YEAR_SELECTED, RACE_SELECTED, SESSION)   


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

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

distance_range = st.slider(
    'Select a range for distance ',
    0, 7100, (1025, 2000))

Corner_Analysis.plot(DRIVER_1, DRIVER_2, distance_range, laps, f1session)