Update pages/Race_Launch_Performance_Ratings.py

pages/Race_Launch_Performance_Ratings.py

@@ -6,17 +6,25 @@ YEAR_SELECTED = st.selectbox(
     'Select Year',
     (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))
 
-
 SESSION = st.selectbox(
     'Select Session',
     ('R', 'SQ'))
 
+reaction_df, drivers = rl.get_laps_and_drivers(YEAR_SELECTED, RACE_SELECTED, SESSION)
+
+DRIVERS_REMOVED = st.multiselect(
+    'Remove drivers starting from pitlane',
+    drivers.tolist()
+    )
+# # remove Driver = 'ZHO' from laps due to pit lane start
+# laps = laps[~laps["Driver"].isin(DRIVERS_REMOVED)]
+
+REMAINING_DRIVERS = drivers.tolist() - DRIVERS_REMOVED
 
+ratings_df = rl.get_ratings(laps, drivers, reaction_df, REMAINING_DRIVERS)
 
-rl.plot_ratings()
+rl.plot_ratings(ratings_df)