Update main.py

main.py

@@ -262,17 +262,13 @@ async def session_drivers(year: int, event: str | int, session: str) -> any:
     
     # get drivers available for a given year, event and session
     f1session = fastf1.get_session(year, event, session)
-    f1session.load(telemetry=False, weather=False, messages=False)
-    laps = f1session.laps
-
-    team_colors = available_data.team_colors(year)    
-    
-    drivers = laps.Driver.unique()
-    # for each driver in drivers, get the Team column from laps and get the color from team_colors dict
-    drivers = [{"color": team_colors[laps[laps.Driver ==
-                                            driver].Team.iloc[0]], "label": driver, "value": driver} for driver in drivers]
-    
-    
+    api_path = f1session.api_path
+    drivers_raw = fastf1.api.driver_info(api_path)
+    drivers = [{
+        "color": available_data.team_colors(year)[driver[1]['TeamName']],
+        "label": driver[1]['Tla'],
+        "value": driver[1]['Tla']
+    } for driver in drivers_raw.items()]
     
     return {"drivers": drivers}