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

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

colors_df = Tyre_Degradation.get_colors(YEAR_SELECTED)

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