import streamlit as st
from repo_directory import Tyre_Degradation
from repo_directory import button

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

colors_df = Tyre_Degradation.get_colors(YEAR_SELECTED)

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', 'SQ'))

laps, f1session, all_driver_compound, all_compounds_used, all_team_compound = Tyre_Degradation.get_laps(YEAR_SELECTED, RACE_SELECTED, SESSION)

DRIVERS_SELECTED = st.multiselect(
    'Select Driver and Tyre',
     all_driver_compound,
    )

TEAMS_SELECTED = st.multiselect(
    'Select Team and Tyre',
     all_team_compound,
    )


compounds_used = st.multiselect(
    'Select Tyres',
     all_compounds_used,
    all_compounds_used)

Tyre_Degradation.plot(DRIVERS_SELECTED,compounds_used, f1session, laps, colors_df, TEAMS_SELECTED)