session results for race, quali, sprint quali and sprint

api_playground.ipynb

@@ -34,234 +34,128 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
-   "id": "2815be5b",
+   "execution_count": 9,
+   "id": "a2ee421b",
    "metadata": {},
    "outputs": [
     {
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "req         WARNING \tDEFAULT CACHE ENABLED! (173.43 MB) /home/arre/.cache/fastf1\n"
+      "core           INFO \tLoading data for Miami Grand Prix - Sprint Qualifying [v3.5.3]\n",
+      "req            INFO \tUsing cached data for session_info\n",
+      "req            INFO \tUsing cached data for driver_info\n",
+      "core        WARNING \tSprint Qualifying is not supported by Ergast! Limited results are calculated from timing data.\n",
+      "req            INFO \tUsing cached data for session_status_data\n",
+      "req            INFO \tUsing cached data for track_status_data\n",
+      "req            INFO \tUsing cached data for _extended_timing_data\n",
+      "req            INFO \tUsing cached data for timing_app_data\n",
+      "core           INFO \tProcessing timing data...\n",
+      "req            INFO \tUsing cached data for weather_data\n",
+      "req            INFO \tUsing cached data for race_control_messages\n",
+      "core           INFO \tFinished loading data for 20 drivers: ['12', '81', '4', '1', '63', '16', '44', '23', '6', '14', '27', '31', '10', '30', '55', '18', '7', '22', '5', '87']\n"
      ]
     },
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "['api_path',\n",
-      " 'car_data',\n",
-      " 'date',\n",
-      " 'drivers',\n",
-      " 'event',\n",
-      " 'f1_api_support',\n",
-      " 'get_circuit_info',\n",
-      " 'get_driver',\n",
-      " 'laps',\n",
-      " 'load',\n",
-      " 'name',\n",
-      " 'pos_data',\n",
-      " 'race_control_messages',\n",
-      " 'results',\n",
-      " 'session_info',\n",
-      " 'session_start_time',\n",
-      " 'session_status',\n",
-      " 't0_date',\n",
-      " 'total_laps',\n",
-      " 'track_status',\n",
-      " 'weather_data']\n"
-     ]
-    }
-   ],
-   "source": [
-    "session = fastf1.get_session(2025, 'Bahrain', 'Race')\n",
-    "attributes = [x for x in dir(session) if not x.startswith(\"_\")]\n",
-    "pprint(attributes)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 21,
-   "id": "1cc4ab17",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "ATTR: api_path: /static/2025/2025-04-13_Bahrain_Grand_Prix/2025-04-13_Race/\n",
-      "\n",
-      "Failed to get attribute: car_data\n",
-      "ATTR: date: 2025-04-13 15:00:00\n",
-      "\n",
-      "Failed to get attribute: drivers\n",
-      "ATTR: event: RoundNumber                                                   4\n",
-      "Country                                                 Bahrain\n",
-      "Location                                                 Sakhir\n",
-      "OfficialEventName    FORMULA 1 GULF AIR BAHRAIN GRAND PRIX 2025\n",
-      "EventDate                                   2025-04-13 00:00:00\n",
-      "EventName                                    Bahrain Grand Prix\n",
-      "EventFormat                                        conventional\n",
-      "Session1                                             Practice 1\n",
-      "Session1Date                          2025-04-11 14:30:00+03:00\n",
-      "Session1DateUtc                             2025-04-11 11:30:00\n",
-      "Session2                                             Practice 2\n",
-      "Session2Date                          2025-04-11 18:00:00+03:00\n",
-      "Session2DateUtc                             2025-04-11 15:00:00\n",
-      "Session3                                             Practice 3\n",
-      "Session3Date                          2025-04-12 15:30:00+03:00\n",
-      "Session3DateUtc                             2025-04-12 12:30:00\n",
-      "Session4                                             Qualifying\n",
-      "Session4Date                          2025-04-12 19:00:00+03:00\n",
-      "Session4DateUtc                             2025-04-12 16:00:00\n",
-      "Session5                                                   Race\n",
-      "Session5Date                          2025-04-13 18:00:00+03:00\n",
-      "Session5DateUtc                             2025-04-13 15:00:00\n",
-      "F1ApiSupport                                               True\n",
-      "Name: 4, dtype: object\n",
+      "   DriverNumber BroadcastName Abbreviation DriverId         TeamName  \\\n",
+      "12           12   K ANTONELLI          ANT                  Mercedes   \n",
+      "81           81     O PIASTRI          PIA                   McLaren   \n",
+      "4             4      L NORRIS          NOR                   McLaren   \n",
+      "1             1  M VERSTAPPEN          VER           Red Bull Racing   \n",
+      "63           63     G RUSSELL          RUS                  Mercedes   \n",
       "\n",
-      "ATTR: f1_api_support: True\n",
+      "   TeamColor TeamId FirstName    LastName        FullName  ... CountryCode  \\\n",
+      "12    27F4D2             Kimi   Antonelli  Kimi Antonelli  ...               \n",
+      "81    FF8000            Oscar     Piastri   Oscar Piastri  ...               \n",
+      "4     FF8000            Lando      Norris    Lando Norris  ...               \n",
+      "1     3671C6              Max  Verstappen  Max Verstappen  ...               \n",
+      "63    27F4D2           George     Russell  George Russell  ...               \n",
       "\n",
-      "ATTR: get_circuit_info: <bound method Session.get_circuit_info of 2025 Season Round 4: Bahrain Grand Prix - Race>\n",
+      "   Position  ClassifiedPosition GridPosition                     Q1  \\\n",
+      "12      1.0                              NaN 0 days 00:01:27.858000   \n",
+      "81      2.0                              NaN 0 days 00:01:27.951000   \n",
+      "4       3.0                              NaN 0 days 00:01:27.890000   \n",
+      "1       4.0                              NaN 0 days 00:01:27.953000   \n",
+      "63      5.0                              NaN 0 days 00:01:27.688000   \n",
       "\n",
-      "ATTR: get_driver: <bound method Session.get_driver of 2025 Season Round 4: Bahrain Grand Prix - Race>\n",
+      "                       Q2                     Q3 Time Status Points  \n",
+      "12 0 days 00:01:27.384000 0 days 00:01:26.482000  NaT           NaN  \n",
+      "81 0 days 00:01:27.354000 0 days 00:01:26.527000  NaT           NaN  \n",
+      "4  0 days 00:01:27.109000 0 days 00:01:26.582000  NaT           NaN  \n",
+      "1  0 days 00:01:27.245000 0 days 00:01:26.737000  NaT           NaN  \n",
+      "63 0 days 00:01:27.666000 0 days 00:01:26.791000  NaT           NaN  \n",
       "\n",
-      "Failed to get attribute: laps\n",
-      "ATTR: load: <bound method Session.load of 2025 Season Round 4: Bahrain Grand Prix - Race>\n",
-      "\n",
-      "ATTR: name: Race\n",
-      "\n",
-      "Failed to get attribute: pos_data\n",
-      "Failed to get attribute: race_control_messages\n",
-      "Failed to get attribute: results\n",
-      "Failed to get attribute: session_info\n",
-      "Failed to get attribute: session_start_time\n",
-      "Failed to get attribute: session_status\n",
-      "Failed to get attribute: t0_date\n",
-      "Failed to get attribute: total_laps\n",
-      "Failed to get attribute: track_status\n",
-      "Failed to get attribute: weather_data\n"
+      "[5 rows x 21 columns]\n"
      ]
     }
    ],
    "source": [
-    "for attr in attributes:\n",
-    "    try:\n",
-    "        print(f\"ATTR: {attr}: {getattr(session, attr)}\\n\")\n",
-    "    except:\n",
-    "        print(f\"Failed to get attribute: {attr}\")"
+    "session = fastf1.get_session(2025, \"Miami\", \"sprint qualifying\")\n",
+    "session.load(telemetry=False)\n",
+    "results = session.results\n",
+    "print(results.head())\n",
+    "#df = results[[\"DriverNumber\", \"Abbreviation\", \"FullName\", \"Position\", \"ClassifiedPosition\", \"GridPosition\", \"Points\", \"Status\", \"Q1\", \"Q2\", \"Q3\"]]"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
-   "id": "44a4b93b",
+   "execution_count": null,
+   "id": "352270a0",
    "metadata": {},
    "outputs": [
     {
-     "name": "stdout",
+     "name": "stderr",
      "output_type": "stream",
      "text": [
-      "RoundNumber                                                   4\n",
-      "Country                                                 Bahrain\n",
-      "Location                                                 Sakhir\n",
-      "OfficialEventName    FORMULA 1 GULF AIR BAHRAIN GRAND PRIX 2025\n",
-      "EventDate                                   2025-04-13 00:00:00\n",
-      "EventName                                    Bahrain Grand Prix\n",
-      "EventFormat                                        conventional\n",
-      "Session1                                             Practice 1\n",
-      "Session1Date                          2025-04-11 14:30:00+03:00\n",
-      "Session1DateUtc                             2025-04-11 11:30:00\n",
-      "Session2                                             Practice 2\n",
-      "Session2Date                          2025-04-11 18:00:00+03:00\n",
-      "Session2DateUtc                             2025-04-11 15:00:00\n",
-      "Session3                                             Practice 3\n",
-      "Session3Date                          2025-04-12 15:30:00+03:00\n",
-      "Session3DateUtc                             2025-04-12 12:30:00\n",
-      "Session4                                             Qualifying\n",
-      "Session4Date                          2025-04-12 19:00:00+03:00\n",
-      "Session4DateUtc                             2025-04-12 16:00:00\n",
-      "Session5                                                   Race\n",
-      "Session5Date                          2025-04-13 18:00:00+03:00\n",
-      "Session5DateUtc                             2025-04-13 15:00:00\n",
-      "F1ApiSupport                                               True\n",
-      "Name: 4, dtype: object\n",
-      "RoundNumber                                                   4\n",
-      "Country                                                 Bahrain\n",
-      "Location                                                 Sakhir\n",
-      "OfficialEventName    FORMULA 1 GULF AIR BAHRAIN GRAND PRIX 2025\n",
-      "EventDate                                   2025-04-13 00:00:00\n",
-      "EventName                                    Bahrain Grand Prix\n",
-      "EventFormat                                        conventional\n",
-      "Session1                                             Practice 1\n",
-      "Session1Date                          2025-04-11 14:30:00+03:00\n",
-      "Session1DateUtc                             2025-04-11 11:30:00\n",
-      "Session2                                             Practice 2\n",
-      "Session2Date                          2025-04-11 18:00:00+03:00\n",
-      "Session2DateUtc                             2025-04-11 15:00:00\n",
-      "Session3                                             Practice 3\n",
-      "Session3Date                          2025-04-12 15:30:00+03:00\n",
-      "Session3DateUtc                             2025-04-12 12:30:00\n",
-      "Session4                                             Qualifying\n",
-      "Session4Date                          2025-04-12 19:00:00+03:00\n",
-      "Session4DateUtc                             2025-04-12 16:00:00\n",
-      "Session5                                                   Race\n",
-      "Session5Date                          2025-04-13 18:00:00+03:00\n",
-      "Session5DateUtc                             2025-04-13 15:00:00\n",
-      "F1ApiSupport                                               True\n",
-      "Name: 4, dtype: object\n"
+      "core           INFO \tLoading data for Japanese Grand Prix - Practice 1 [v3.5.3]\n",
+      "req            INFO \tUsing cached data for session_info\n",
+      "req            INFO \tUsing cached data for driver_info\n",
+      "req            INFO \tUsing cached data for session_status_data\n",
+      "req            INFO \tUsing cached data for track_status_data\n",
+      "req            INFO \tUsing cached data for _extended_timing_data\n",
+      "req            INFO \tUsing cached data for timing_app_data\n",
+      "core           INFO \tProcessing timing data...\n",
+      "req            INFO \tUsing cached data for weather_data\n",
+      "req            INFO \tUsing cached data for race_control_messages\n",
+      "core           INFO \tFinished loading data for 20 drivers: ['1', '4', '5', '6', '10', '12', '14', '16', '18', '22', '23', '27', '30', '31', '44', '55', '62', '63', '81', '87']\n"
      ]
-    }
-   ],
-   "source": [
-    "event1 = fastf1.get_session(2025, 'Bahrain', \"fp1\").event\n",
-    "print(event1)\n",
-    "event2 = fastf1.get_session(2025, 'Bahrain', \"race\").event\n",
-    "print(event2)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "067ab2ac",
-   "metadata": {},
-   "outputs": [
+    },
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "<class 'fastf1.events.Event'>\n",
-      "RoundNumber                                                   4\n",
-      "Country                                                 Bahrain\n",
-      "Location                                                 Sakhir\n",
-      "OfficialEventName    FORMULA 1 GULF AIR BAHRAIN GRAND PRIX 2025\n",
-      "EventDate                                   2025-04-13 00:00:00\n",
-      "EventName                                    Bahrain Grand Prix\n",
-      "EventFormat                                        conventional\n",
-      "Session1                                             Practice 1\n",
-      "Session1Date                          2025-04-11 14:30:00+03:00\n",
-      "Session1DateUtc                             2025-04-11 11:30:00\n",
-      "Session2                                             Practice 2\n",
-      "Session2Date                          2025-04-11 18:00:00+03:00\n",
-      "Session2DateUtc                             2025-04-11 15:00:00\n",
-      "Session3                                             Practice 3\n",
-      "Session3Date                          2025-04-12 15:30:00+03:00\n",
-      "Session3DateUtc                             2025-04-12 12:30:00\n",
-      "Session4                                             Qualifying\n",
-      "Session4Date                          2025-04-12 19:00:00+03:00\n",
-      "Session4DateUtc                             2025-04-12 16:00:00\n",
-      "Session5                                                   Race\n",
-      "Session5Date                          2025-04-13 18:00:00+03:00\n",
-      "Session5DateUtc                             2025-04-13 15:00:00\n",
-      "F1ApiSupport                                               True\n",
-      "Name: 4, dtype: object\n"
+      "{'Abbreviation': 'VER',\n",
+      " 'BroadcastName': 'M VERSTAPPEN',\n",
+      " 'ClassifiedPosition': '',\n",
+      " 'CountryCode': '',\n",
+      " 'DriverId': 'max_verstappen',\n",
+      " 'DriverNumber': '1',\n",
+      " 'FirstName': 'Max',\n",
+      " 'FullName': 'Max Verstappen',\n",
+      " 'GridPosition': nan,\n",
+      " 'HeadshotUrl': 'https://media.formula1.com/d_driver_fallback_image.png/content/dam/fom-website/drivers/M/MAXVER01_Max_Verstappen/maxver01.png.transform/1col/image.png',\n",
+      " 'LastName': 'Verstappen',\n",
+      " 'Points': nan,\n",
+      " 'Position': nan,\n",
+      " 'Q1': NaT,\n",
+      " 'Q2': NaT,\n",
+      " 'Q3': NaT,\n",
+      " 'Status': '',\n",
+      " 'TeamColor': '3671C6',\n",
+      " 'TeamId': 'red_bull',\n",
+      " 'TeamName': 'Red Bull Racing',\n",
+      " 'Time': NaT}\n"
      ]
     }
    ],
    "source": [
-    "event2 = fastf1.get_event(2025, 'Bahrain')\n",
-    "print(type(event2))\n",
-    "print(event2.keys())\n"
+    "session = fastf1.get_session(2025, 3, \"FP1\")\n",
+    "session.load(telemetry=False)\n",
+    "driver_info = session.get_driver(\"VER\").to_dict()\n",
+    "pprint(driver_info)"
    ]
   },
   {

app.py

@@ -5,7 +5,9 @@ import tools
 from utils.constants import (
     DRIVER_NAMES,
     CONSTRUCTOR_NAMES,
-    CURRENT_YEAR
+    CURRENT_YEAR,
+    AVAILABLE_SESSION_TYPES,
+    DROPDOWN_SESSION_TYPES
 )
 
 iface_driver_championship_standings = gr.Interface(
@@ -65,13 +67,26 @@ iface_track_visualization = gr.Interface(
     description="Get the track visualization for the given Grand Prix"
 )
 
+iface_session_results = gr.Interface(
+    fn=tools.get_session_results,
+    inputs=[
+        gr.Number(label="Calendar year", value=CURRENT_YEAR, minimum=1950, maximum=CURRENT_YEAR),
+        gr.Textbox(label="Grand Prix", placeholder="Ex: Monaco", info="The name of the GP/country/location (Fuzzy matching supported)"),
+        gr.Dropdown([session_type for session_type in DROPDOWN_SESSION_TYPES if "practice" not in session_type], label="Session type", value="race")
+    ],
+    outputs=gr.DataFrame(),
+    title="Session Results",
+    description="Get the session results for the given Grand Prix"
+)
+
 
 named_interfaces = {
     "Driver Championship Standings": iface_driver_championship_standings,
     "Constructor Championship Standings": iface_constructor_championship_standings,
     "Event Info": iface_event_info,
     "Season Calendar": iface_season_calendar,
-    "Track Visualizations": iface_track_visualization
+    "Track Visualizations": iface_track_visualization,
+    "Session Results": iface_session_results
 }
 
 # Tab names and interfaces

tools.py

@@ -4,15 +4,14 @@ Here we implement the MCP tools
 Tools to implement
 - session result
 - driver info
-- championship standings
-- driver standings
-- constructor standings
 - compare drivers
 
 """
 
 import json
 import fastf1
+import gradio as gr
+import pandas as pd
 from PIL import Image
 from typing import Union
 from fastf1.core import Session
@@ -71,7 +70,6 @@ def driver_championship_standings(year: int, driver_name: str) -> str:
     standings_string = f"{driver_name} is {driver_standing['position'].iloc[0]}{suffix} with {driver_standing['points'].iloc[0]} points and {driver_standing['wins'].iloc[0]} wins"
     return standings_string
     
-
 def constructor_championship_standings(year: int, constructor_name: str) -> str:
 
     team_mapping = {
@@ -96,8 +94,6 @@ def constructor_championship_standings(year: int, constructor_name: str) -> str:
     standings_string = f"{constructor_name} are {constructor_standing['position'].iloc[0]}{suffix} with {constructor_standing['points'].iloc[0]} points and {constructor_standing['wins'].iloc[0]} wins"
     return standings_string
 
-
-
 def track_visualization(year: int, round: gp, visualization_type: str, driver_name: str) -> Image:
 
     session = get_session(year, round, "race")
@@ -110,6 +106,29 @@ def track_visualization(year: int, round: gp, visualization_type: str, driver_na
     elif visualization_type == "gear":
         return track_utils.create_track_gear_visualization(session)
 
+def get_session_results(year: int, round: gp, session_type: session_type) -> pd.DataFrame:
+
+    try:
+        session = get_session(year, round, session_type)
+        session.load(telemetry=False)
+        results = session.results
+    except ValueError as e:
+        raise gr.Error(f"Session type {session_type} is not supported for the specified round. This Grand Prix most likely did not include a sprint race/quali.")
+
+    df = results[["DriverNumber", "Abbreviation", "FullName", "Position", "GridPosition", "Points", "Status", "Q1", "Q2", "Q3"]]
+    df["Name"] = df.apply(lambda row: f"{row['FullName']} ({row['Abbreviation']} • {row['DriverNumber']})", axis=1)
+    df = df.drop(columns=["FullName", "Abbreviation", "DriverNumber"])
+    df = df.rename(columns={"Position": "Pos", "GridPosition": "Grid Pos"})
+
+    # Process results based on session type
+    if session_type in ["race", "sprint"]:
+        df = df[["Pos", "Name", "Points", "Grid Pos", "Status"]]
+    elif "qualifying" in session_type:
+        df[["Q1", "Q2", "Q3"]] = df[["Q1", "Q2", "Q3"]].apply(lambda x: x.dt.total_seconds().apply(lambda y: f"{int(y//60):02d}:{int(y%60):02d}.{int(y%1*1000):03d}" if pd.notna(y) else "-"))
+        df = df[["Pos", "Name", "Q1", "Q2", "Q3"]]
+    return df
+    
+
 
 if __name__ == "__main__":
     session = get_session(2024, 1, "fp1")

utils/constants.py

@@ -4,12 +4,12 @@ import datetime
 AVAILABLE_SESSION_TYPES = [
     "fp1", "fp2", "fp3", "q", "s", "ss", "sq", "r",
     "practice 1", "practice 2", "practice 3", "sprint",
-    "sprint shootout", "sprint qualifying", "qualifying", "race"
+    "sprint qualifying", "qualifying", "race"
     ]
 
 DROPDOWN_SESSION_TYPES = [
     "practice 1", "practice 2", "practice 3", "sprint",
-    "sprint shootout", "sprint qualifying", "qualifying", "race"
+    "sprint qualifying", "qualifying", "race"
     ]
 
 # Load in driver names