import streamlit as st
from repo_directory import Top_Speed_at_Speed_Trap
from repo_directory import button

YEAR = st.selectbox(
    'Select Year',
    (2023, 2022, 2021, 2020, 2019, 2018))

RACE = 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',
    ('FP1', 'FP2', 'FP3', 'Q', 'SQ', 'R'))

laps = Top_Speed_at_Speed_Trap.get_laps(YEAR, RACE, SESSION)

fastest_st = Top_Speed_at_Speed_Trap.get_fastest_speedtrap(laps)

max_speeds = Top_Speed_at_Speed_Trap.get_max_speeds(fastest_st, laps,YEAR)

Top_Speed_at_Speed_Trap.plot(max_speeds, fastest_st, YEAR, RACE, SESSION)