Update pages/Download_Raw_Data.py

pages/Download_Raw_Data.py

@@ -14,8 +14,20 @@ SESSION = st.selectbox(
     'Select Session',
     ('FP1', 'FP2', 'FP3', 'Q', 'SQ', 'R'))
 
-laps = dl.get_laps(YEAR_SELECTED, RACE_SELECTED, SESSION)
+laps, drivers_list = dl.get_laps(YEAR_SELECTED, RACE_SELECTED, SESSION)
 laps.to_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')  
+    st.download_button('Click to Download LapTimes Data', f, file_name=f'{YEAR_SELECTED}-{RACE_SELECTED}-{SESSION}.csv') 
+
+st.write('For Detailed Telemetry Data')
+DRIVERS_SELECTED = st.multiselect(
+    'Select Driver',
+    drivers_list)
+
+telemetry = dl.selected_telemetry()
+
+telemetry.to_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')