Update pages/Corner_Analysis.py

pages/Corner_Analysis.py

@@ -14,20 +14,24 @@ SESSION = st.selectbox(
     'Select Session',
     utils.get_sessions(YEAR_SELECTED, RACE_SELECTED))
 
-
-laps, f1session, drivers = Corner_Analysis.get_data(YEAR_SELECTED, RACE_SELECTED, SESSION)   
-
-
-DRIVER_1 = st.selectbox(
-    'Select Driver 1',
-    drivers)
-
-DRIVER_2 = st.selectbox(
-    'Select Driver 2',
-    drivers)
-
-distance_range = st.slider(
-    'Select a range for distance ',
-    0, 7100, (1025, 2000))
-
-Corner_Analysis.plot(DRIVER_1, DRIVER_2, distance_range, laps, f1session)
+# Only proceed if all selections are made
+if YEAR_SELECTED and RACE_SELECTED and SESSION:
+    laps, f1session, drivers = Corner_Analysis.get_data(YEAR_SELECTED, RACE_SELECTED, SESSION)
+    
+    DRIVER_1 = st.selectbox(
+        'Select Driver 1',
+        drivers)
+    
+    DRIVER_2 = st.selectbox(
+        'Select Driver 2',
+        drivers)
+    
+    distance_range = st.slider(
+        'Select a range for distance',
+        0, 7100, (1025, 2000))
+    
+    # Only plot if both drivers are selected
+    if DRIVER_1 and DRIVER_2:
+        Corner_Analysis.plot(DRIVER_1, DRIVER_2, distance_range, laps, f1session)
+else:
+    st.info("Please select year, race, and session to continue.")