driver live standings

api_playground.ipynb

@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": 2,
    "id": "b7d2515e",
    "metadata": {},
    "outputs": [
@@ -264,6 +264,127 @@
     "print(event2.keys())\n"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": 24,
+   "id": "e6de8707",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/html": [
+       "<div>\n",
+       "<style scoped>\n",
+       "    .dataframe tbody tr th:only-of-type {\n",
+       "        vertical-align: middle;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe tbody tr th {\n",
+       "        vertical-align: top;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe thead th {\n",
+       "        text-align: right;\n",
+       "    }\n",
+       "</style>\n",
+       "<table border=\"1\" class=\"dataframe\">\n",
+       "  <thead>\n",
+       "    <tr style=\"text-align: right;\">\n",
+       "      <th></th>\n",
+       "      <th>position</th>\n",
+       "      <th>positionText</th>\n",
+       "      <th>points</th>\n",
+       "      <th>wins</th>\n",
+       "      <th>driverId</th>\n",
+       "      <th>driverNumber</th>\n",
+       "      <th>driverCode</th>\n",
+       "      <th>driverUrl</th>\n",
+       "      <th>givenName</th>\n",
+       "      <th>familyName</th>\n",
+       "      <th>dateOfBirth</th>\n",
+       "      <th>driverNationality</th>\n",
+       "      <th>constructorIds</th>\n",
+       "      <th>constructorUrls</th>\n",
+       "      <th>constructorNames</th>\n",
+       "      <th>constructorNationalities</th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <th>0</th>\n",
+       "      <td>1</td>\n",
+       "      <td>1</td>\n",
+       "      <td>575.0</td>\n",
+       "      <td>19</td>\n",
+       "      <td>max_verstappen</td>\n",
+       "      <td>33</td>\n",
+       "      <td>VER</td>\n",
+       "      <td>http://en.wikipedia.org/wiki/Max_Verstappen</td>\n",
+       "      <td>Max</td>\n",
+       "      <td>Verstappen</td>\n",
+       "      <td>1997-09-30</td>\n",
+       "      <td>Dutch</td>\n",
+       "      <td>[red_bull]</td>\n",
+       "      <td>[http://en.wikipedia.org/wiki/Red_Bull_Racing]</td>\n",
+       "      <td>[Red Bull]</td>\n",
+       "      <td>[Austrian]</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>1</th>\n",
+       "      <td>2</td>\n",
+       "      <td>2</td>\n",
+       "      <td>285.0</td>\n",
+       "      <td>2</td>\n",
+       "      <td>perez</td>\n",
+       "      <td>11</td>\n",
+       "      <td>PER</td>\n",
+       "      <td>http://en.wikipedia.org/wiki/Sergio_P%C3%A9rez</td>\n",
+       "      <td>Sergio</td>\n",
+       "      <td>Pérez</td>\n",
+       "      <td>1990-01-26</td>\n",
+       "      <td>Mexican</td>\n",
+       "      <td>[red_bull]</td>\n",
+       "      <td>[http://en.wikipedia.org/wiki/Red_Bull_Racing]</td>\n",
+       "      <td>[Red Bull]</td>\n",
+       "      <td>[Austrian]</td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n",
+       "</div>"
+      ],
+      "text/plain": [
+       "   position positionText  points  wins        driverId  driverNumber  \\\n",
+       "0         1            1   575.0    19  max_verstappen            33   \n",
+       "1         2            2   285.0     2           perez            11   \n",
+       "\n",
+       "  driverCode                                       driverUrl givenName  \\\n",
+       "0        VER     http://en.wikipedia.org/wiki/Max_Verstappen       Max   \n",
+       "1        PER  http://en.wikipedia.org/wiki/Sergio_P%C3%A9rez    Sergio   \n",
+       "\n",
+       "   familyName dateOfBirth driverNationality constructorIds  \\\n",
+       "0  Verstappen  1997-09-30             Dutch     [red_bull]   \n",
+       "1       Pérez  1990-01-26           Mexican     [red_bull]   \n",
+       "\n",
+       "                                  constructorUrls constructorNames  \\\n",
+       "0  [http://en.wikipedia.org/wiki/Red_Bull_Racing]       [Red Bull]   \n",
+       "1  [http://en.wikipedia.org/wiki/Red_Bull_Racing]       [Red Bull]   \n",
+       "\n",
+       "  constructorNationalities  \n",
+       "0               [Austrian]  \n",
+       "1               [Austrian]  "
+      ]
+     },
+     "execution_count": 24,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "ergastandings = fastf1.ergast.Ergast()\n",
+    "driver_standings = ergastandings.get_driver_standings(2023).content[0]\n",
+    "driver_standings.head(2)"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 2,

app.py

@@ -22,7 +22,7 @@ with open("assets/constructor_details.json") as f:
 
 
 iface_driver_championship_standings = gr.Interface(
-    fn=tools.driver_championship_score,
+    fn=tools.driver_championship_standings,
     inputs=[
         gr.Number(label="Calendar year", value=CURRENT_YEAR, minimum=1950, maximum=CURRENT_YEAR),
         gr.Dropdown(driver_names)
@@ -37,7 +37,7 @@ iface_event_info = gr.Interface(
     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.Radio(["human", "LLM"], label="Format", value="human")
+        gr.Radio(["human", "LLM"], label="Display format", value="human")
     ],
     outputs="text",
     title="Event Info",
@@ -63,21 +63,21 @@ iface_track_visualization = gr.Interface(
         gr.Dropdown(driver_names)
     ],
     outputs="image",
-    title="[WIP] Track Visualization",
+    title="Track Visualizations",
     description="Get the track visualization for the given Grand Prix"
 )
 
 
-# TODO swap to this format
 named_interfaces = {
-    "driver_championship_standings": iface_driver_championship_standings,
-    "event_info": iface_event_info,
-    "season_calendar": iface_season_calendar,
-    "track_visualization": iface_track_visualization
+    "Driver Championship Standings": iface_driver_championship_standings,
+    "Event Info": iface_event_info,
+    "Season Calendar": iface_season_calendar,
+    "Track Visualizations": iface_track_visualization
 }
 
-interface_list = [iface_driver_championship_standings, iface_event_info, iface_season_calendar, iface_track_visualization]
-tab_names = ["Driver Championship Standings", "Event Info", "Season Calendar", "Track Visualization"]
+# Tab names and interfaces
+tab_names = list(named_interfaces.keys())
+interface_list = list(named_interfaces.values())
 
 
 # Combine all the interfaces into a single TabbedInterface

tools.py

@@ -6,34 +6,27 @@ Tools to implement
 - driver info
 - championship standings
 - driver standings
+- constructor standings
 - compare drivers
 
 """
 
-
-## Here we implement the MCP tools
+import json
 import fastf1
-from utils.constants import AVAILABLE_SESSION_TYPES
-
-from fastf1.core import Session
+from PIL import Image
 from typing import Union
+from fastf1.core import Session
 
+# Local modules
 from utils import parser_utils, track_utils
-from PIL import Image
+from utils.constants import AVAILABLE_SESSION_TYPES
 
 # Custom types
 gp = Union[str, int]
 session_type = Union[str, int, None]
 
-def get_season_calendar(year: int) -> str:
-    """ Returns the season calendar for the given year """
-    season_calendar = fastf1.get_event_schedule(year)
-    return parser_utils.parse_season_calendar(season_calendar)
-
 def get_session(year: int, round: gp, session_type: session_type) -> Session:
-    """
-    Returns a session specified by year, round and session type
-    """
+    """ Returns a session specified by year, round and session type """
 
     # Check if session type is valid
     if isinstance(session_type, str):
@@ -42,6 +35,11 @@ def get_session(year: int, round: gp, session_type: session_type) -> Session:
 
     return fastf1.get_session(year, round, session_type)
 
+def get_season_calendar(year: int) -> str:
+    """ Returns the season calendar for the given year """
+    season_calendar = fastf1.get_event_schedule(year)
+    return parser_utils.parse_season_calendar(season_calendar)
+
 def get_event_info(year: int, round: gp, format: str) -> str:
     """ Returns the event info for the specified session """
 
@@ -57,24 +55,27 @@ def get_event_info(year: int, round: gp, format: str) -> str:
 def get_constructor_standings(year: int) -> str:
     pass
 
-def get_end_of_season_standings(year): 
+def get_driver_standings(year: int) -> str: 
     pass
 
+def driver_championship_standings(year: int, driver_name: str) -> str:
+
+    with open("assets/driver_abbreviations.json") as f:
+        driver_abbreviations = json.load(f)
+        driver_abbreviation = driver_abbreviations[driver_name]
+    ergast = fastf1.ergast.Ergast()
+    driver_standings = ergast.get_driver_standings(year).content[0]
+    driver_standing = driver_standings[["position", "points", "wins", "driverCode"]].reset_index(drop=True)
+    driver_standing = driver_standing[driver_standing["driverCode"] == driver_abbreviation]
+    suffix = "st" if driver_standing['position'].iloc[0] == 1 else "nd" if driver_standing['position'].iloc[0] == 2 else "rd" if driver_standing['position'].iloc[0] == 3 else "th"
+    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:
+    pass
 
-def driver_position(driver_name: str) -> str:
-    """
-    Get the current position of the given driver.
-
-    Args:
-        driver_name (str): The driver's name
-
-    Returns:
-        str: The driver's current position
-    """
-    return f"Driver {driver_name} is in position {random.randint(1, 20)}"
 
-def driver_championship_score(year: int, driver_name: str) -> str:
-    pass
 
 def track_visualization(year: int, round: gp, visualization_type: str, driver_name: str) -> Image: