add all drivers and error message

app.py

@@ -16,6 +16,26 @@ ff1.Cache.enable_cache(cache_path)
 # Offline mode to prevent F1 API crashes on Hugging Face
 ff1.Cache.offline_mode(enabled=True)
 
+# Define drivers
+drivers_2023 = {'Fastest driver': 'Fastest driver', 'VER': 'Max Verstappen', 
+                'NOR': 'Lando Norris', 'GAS': 'Pierre Gasly', 'PER': 'Sergio Perez', 
+                'ALO': 'Fernando Alonso', 'LEC': 'Charles Leclerc', 'STR': 'Lance Stroll', 
+                'MAG': 'Kevin Magnussen', 'TSU': 'Yuki Tsunoda', 'ALB': 'Alexander Albon',
+                'ZHO': 'Guanyu Zhou', 'HUL': 'Nico Hülkenberg', 'OCO': 'Esteban Ocon',
+                'HAM': 'Lewis Hamilton', 'SAI': 'Carlos Sainz', 'RUS': 'George Russel',
+                'BOT': 'Valteri Bottas', 'PIA': 'Oscar Piastri', 'VRI': 'Nyck de Vries',
+                'SAR': 'Logan Sargeant', 'RIC': 'Daniel Ricciardo'}
+
+drivers_2022 = {'Fastest driver': 'Fastest driver', 'VER': 'Max Verstappen', 
+                'NOR': 'Lando Norris', 'GAS': 'Pierre Gasly', 'PER': 'Sergio Perez', 
+                'ALO': 'Fernando Alonso', 'LEC': 'Charles Leclerc', 'STR': 'Lance Stroll', 
+                'MAG': 'Kevin Magnussen', 'TSU': 'Yuki Tsunoda', 'ALB': 'Alexander Albon',
+                'ZHO': 'Guanyu Zhou', 'HUL': 'Nico Hülkenberg', 'OCO': 'Esteban Ocon',
+                'HAM': 'Lewis Hamilton', 'SAI': 'Carlos Sainz', 'RUS': 'George Russel',
+                'BOT': 'Valteri Bottas', 'VRI': 'Nyck de Vries', 'VET': 'Sebastian Vettel',
+                'RIC': 'Daniel Ricciardo', 'MSC': 'Mick Schumacher',
+                'LAT': 'Nicolas Latifi'}
+
 app_ui = ui.page_fluid(
     shinyswatch.theme.minty(),
     ui.panel_title("Gear usage in fastest lap"),
@@ -23,7 +43,7 @@ app_ui = ui.page_fluid(
         ui.panel_sidebar(
             ui.input_select(
                 "track_select", "Select track:",
-                choices = ["Austria", "Hungary", "Spain", "Bahrain"],
+                choices = ["Austria", "Hungary", "Spain", "Bahrain", "Great-Brittain"],
                 selected = "Austria"
             ),
             ui.input_select(
@@ -55,9 +75,9 @@ def server(input, output, session):
     @reactive.Effect()
     def _():
         if input.year() == "2023":
-            driver_options = ['Fastest driver', 'RUS', 'VER', 'HAM']
+            driver_options = drivers_2023
         elif input.year() == "2022":
-            driver_options = ['Fastest driver', 'LEC', 'NOR', 'GAS', 'VER']
+            driver_options = drivers_2022
             
         ui.update_select("driver_select",
             label="Select driver:",
@@ -67,27 +87,32 @@ def server(input, output, session):
     
     @reactive.Calc
     def get_data():
-        f1_session = ff1.get_session(int(input.year()), input.track_select(), input.session_type())
-        f1_session.load()
-
-        # Check if user input == fastest driver
-        if input.driver_select() == "Fastest driver":
-            lap = f1_session.laps.pick_fastest()
-        else:
-            laps_driver = f1_session.laps.pick_driver(input.driver_select())
-            lap = laps_driver.pick_fastest()
-        
-        tel = lap.get_telemetry()
-        driver = lap['Driver']
+        try:
+            f1_session = ff1.get_session(int(input.year()), input.track_select(), input.session_type())
+            f1_session.load()
 
-        #converting data to numpy data tables
-        x = np.array(tel['X'].values)
-        y = np.array(tel['Y'].values)
+            # Check if user input == fastest driver
+            if input.driver_select() == "Fastest driver":
+                lap = f1_session.laps.pick_fastest()
+            else:
+                laps_driver = f1_session.laps.pick_driver(input.driver_select())
+                lap = laps_driver.pick_fastest()
+            
+            tel = lap.get_telemetry()
+            driver = lap['Driver']
 
-        points = np.array([x, y]).T.reshape(-1, 1, 2)
-        segments = np.concatenate([points[:-1], points[1:]], axis=1)
-        gear = tel['nGear'].to_numpy().astype(float)
-        return segments, gear, driver
+            #converting data to numpy data tables
+            x = np.array(tel['X'].values)
+            y = np.array(tel['Y'].values)
+
+            points = np.array([x, y]).T.reshape(-1, 1, 2)
+            segments = np.concatenate([points[:-1], points[1:]], axis=1)
+            gear = tel['nGear'].to_numpy().astype(float)
+            return segments, gear, driver
+
+        except Exception:
+            pass
+        
     
     @output
     @render.text
@@ -99,22 +124,26 @@ def server(input, output, session):
     @output
     @render.plot
     def gear():
-        segments, gear, driver = get_data()
-        
-        cmap = cm.get_cmap('Paired')
-        lc_comp = LineCollection(segments, norm=plt.Normalize(1, cmap.N+1), cmap=cmap)
-        lc_comp.set_array(gear)
-        lc_comp.set_linewidth(4)
-
-        plt.gca().add_collection(lc_comp)
-        plt.axis('equal')
-        plt.tick_params(labelleft=False, left=False, labelbottom=False, bottom=False)
-
-        cbar = plt.colorbar(mappable=lc_comp, label="Gear", boundaries=np.arange(1, 10))
-        cbar.set_ticks(np.arange(1.5, 9.5))
-        cbar.set_ticklabels(np.arange(1, 9))
-  
-        plt
+        try:
+            segments, gear, driver = get_data()
+            
+            cmap = cm.get_cmap('Paired')
+            lc_comp = LineCollection(segments, norm=plt.Normalize(1, cmap.N+1), cmap=cmap)
+            lc_comp.set_array(gear)
+            lc_comp.set_linewidth(4)
+
+            plt.gca().add_collection(lc_comp)
+            plt.axis('equal')
+            plt.tick_params(labelleft=False, left=False, labelbottom=False, bottom=False)
+
+            cbar = plt.colorbar(mappable=lc_comp, label="Gear", boundaries=np.arange(1, 10))
+            cbar.set_ticks(np.arange(1.5, 9.5))
+            cbar.set_ticklabels(np.arange(1, 9))
+    
+            plt
+
+        except Exception:
+            ui.notification_show("Data not available. Select another track or driver.", duration=10, type = 'error')
 
 
 app = App(app_ui, server)