import streamlit as st
from repo_directory import Circuit_plots
from repo_directory import button

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

def total_rounds(YEAR):
    if YEAR == 2023:
        return (1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22)
    if YEAR == 2022:
        return (1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22)
    if YEAR == 2021:
        return (1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22)
    if YEAR == 2020:
        return (1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17)
    if YEAR == 2019:
        return (1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21)
    if YEAR == 2018:
        return (1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21)

RACE = st.selectbox(
    'Select Race',    
    total_rounds(YEAR))


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

Circuit_plots.plot_speed(YEAR,RACE,SESSION)

Circuit_plots.plot_speed2(YEAR,RACE,SESSION)

Circuit_plots.plot_brake(YEAR,RACE,SESSION)

Circuit_plots.plot_throttle(YEAR,RACE,SESSION)

Circuit_plots.plot_elevation(YEAR,RACE,SESSION)

Circuit_plots.plot_lon_acc(YEAR,RACE,SESSION)


Circuit_plots.plot_lat_acc(YEAR,RACE,SESSION)

Circuit_plots.plot_vertical_acc(YEAR,RACE,SESSION)