Update pages/Download_Raw_Data.py

pages/Download_Raw_Data.py

@@ -15,7 +15,7 @@ SESSION = st.selectbox(
     'Select Session',
     ('FP1', 'FP2', 'FP3', 'Q', 'SQ', 'R'))
 
-laps, drivers_list = dl.get_laps(YEAR_SELECTED, RACE_SELECTED, SESSION)
+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'))
 
@@ -23,6 +23,12 @@ 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',