second graph

app.py

@@ -81,12 +81,13 @@ app_ui = ui.page_fluid(
             ui.row(
                 ui.column(
                     6,
-                    ui.output_plot("gear"),
-                    ui.output_text("fastest_driver")
+                    ui.output_plot("gear_1"),
+                    ui.output_text("fastest_driver_1")
                 ),
                 ui.column(
                     6,
-                    ui.output_text("column_2")
+                    ui.output_plot("gear_2"),
+                    ui.output_text("fastest_driver_2")
                 )
             ),
         ),
@@ -115,7 +116,7 @@ def server(input, output, session):
             selected=input.driver2_select()
         )
     
-    # Get required data for both drivers based on selection
+    # Get required data for driver 2 based on selection
     @reactive.Calc
     def get_data_1():
         try:
@@ -144,12 +145,48 @@ def server(input, output, session):
         except Exception:
             ui.notification_show("Data not available. Select another track or driver.", duration=10, type = 'error')
 
+    # Get required data for driver 2 based on selection
+    @reactive.Calc
+    def get_data_2():
+        try:
+            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.driver2_select() == "Fastest driver":
+                lap = f1_session.laps.pick_fastest()
+            else:
+                laps_driver = f1_session.laps.pick_driver(input.driver2_select())
+                lap = laps_driver.pick_fastest()
+            
+            tel = lap.get_telemetry()
+            driver = lap['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:
+            ui.notification_show("Data not available. Select another track or driver.", duration=10, type = 'error')
+    
     @output
     @render.text
     def fastest_driver_1():
         segments, gear, driver = get_data_1()
         #print(f"The driver of the fastest lap this session is: {driver}")
         return f"Graph shows the fastest lap of: {driver}"
+    
+    @output
+    @render.text
+    def fastest_driver_2():
+        segments, gear, driver = get_data_2()
+        #print(f"The driver of the fastest lap this session is: {driver}")
+        return f"Graph shows the fastest lap of: {driver}"
 
     @output
     @render.plot
@@ -175,4 +212,28 @@ def server(input, output, session):
         except Exception:
             pass
 
+    @output
+    @render.plot
+    def gear_2():
+        try:
+            segments, gear, driver = get_data_2()
+            
+            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:
+            pass
+
 app = App(app_ui, server)

cache/fastf1_http_cache.sqlite

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1b369a473451488571ed3c82190ba1cd424e0bea56d0af4b02b2a31088a3c945
+oid sha256:e42b17a02523d5d41be6dc5a492b5c5b4b0fec7cf167e2d1f2af55c975f1bfae
 size 287285248