import streamlit as st
from repo_directory import Download_Raw_Data as dl
from repo_directory import button
import pandas as pd

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

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

laps, drivers_list, weather = dl.get_laps(YEAR_SELECTED, RACE_SELECTED, SESSION)
laps.to_csv(f'{YEAR_SELECTED}-{RACE_SELECTED}-{SESSION}.csv')
st.dataframe(pd.read_csv(f'{YEAR_SELECTED}-{RACE_SELECTED}-{SESSION}.csv'))


with open(f'{YEAR_SELECTED}-{RACE_SELECTED}-{SESSION}.csv') as f:
    st.download_button('Click to Download LapTimes Data', f, file_name=f'{YEAR_SELECTED}-{RACE_SELECTED}-{SESSION}.csv') 

weather.to_csv(f'{YEAR_SELECTED}-{RACE_SELECTED}-{SESSION}-weather.csv')
st.dataframe(pd.read_csv(f'{YEAR_SELECTED}-{RACE_SELECTED}-{SESSION}-weather.csv'))

with open(f'{YEAR_SELECTED}-{RACE_SELECTED}-{SESSION}-weather.csv') as f:
    st.download_button('Click to Download Weather Data', f, file_name=f'{YEAR_SELECTED}-{RACE_SELECTED}-{SESSION}-weather.csv') 

st.write('For Detailed Telemetry Data')
DRIVERS_SELECTED = st.multiselect(
    'Select Driver',
    drivers_list)

telemetry = dl.selected_telemetry(DRIVERS_SELECTED, laps)

telemetry.to_csv(f'{YEAR_SELECTED}-{RACE_SELECTED}-{SESSION}-Tel.csv')
st.dataframe(pd.read_csv(f'{YEAR_SELECTED}-{RACE_SELECTED}-{SESSION}-Tel.csv'))


with open(f'{YEAR_SELECTED}-{RACE_SELECTED}-{SESSION}-Tel.csv') as f:
   st.download_button('Click to Download Telemetry Data', f, file_name=f'{YEAR_SELECTED}-{RACE_SELECTED}-{SESSION}-Tel.csv')