Update main.py

main.py

@@ -13,7 +13,7 @@ from pydantic import BaseModel
 import functools
 import math
 import numpy as np
-
+import concurrent.futures
 
 
 import available_data
@@ -257,13 +257,16 @@ 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)
-    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()]
+    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]
+    
     
     
     return {"drivers": drivers}