added constructors standings and tidied up the UI for driver standings

api_playground.ipynb

@@ -385,6 +385,207 @@
     "driver_standings.head(2)"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": 33,
+   "id": "feeb4592",
+   "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>constructorId</th>\n",
+       "      <th>constructorUrl</th>\n",
+       "      <th>constructorName</th>\n",
+       "      <th>constructorNationality</th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <th>0</th>\n",
+       "      <td>1</td>\n",
+       "      <td>1</td>\n",
+       "      <td>362.0</td>\n",
+       "      <td>7</td>\n",
+       "      <td>mclaren</td>\n",
+       "      <td>http://en.wikipedia.org/wiki/McLaren</td>\n",
+       "      <td>McLaren</td>\n",
+       "      <td>British</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>1</th>\n",
+       "      <td>2</td>\n",
+       "      <td>2</td>\n",
+       "      <td>165.0</td>\n",
+       "      <td>0</td>\n",
+       "      <td>ferrari</td>\n",
+       "      <td>http://en.wikipedia.org/wiki/Scuderia_Ferrari</td>\n",
+       "      <td>Ferrari</td>\n",
+       "      <td>Italian</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>2</th>\n",
+       "      <td>3</td>\n",
+       "      <td>3</td>\n",
+       "      <td>159.0</td>\n",
+       "      <td>0</td>\n",
+       "      <td>mercedes</td>\n",
+       "      <td>http://en.wikipedia.org/wiki/Mercedes-Benz_in_...</td>\n",
+       "      <td>Mercedes</td>\n",
+       "      <td>German</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>3</th>\n",
+       "      <td>4</td>\n",
+       "      <td>4</td>\n",
+       "      <td>144.0</td>\n",
+       "      <td>2</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>4</th>\n",
+       "      <td>5</td>\n",
+       "      <td>5</td>\n",
+       "      <td>54.0</td>\n",
+       "      <td>0</td>\n",
+       "      <td>williams</td>\n",
+       "      <td>http://en.wikipedia.org/wiki/Williams_Grand_Pr...</td>\n",
+       "      <td>Williams</td>\n",
+       "      <td>British</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>5</th>\n",
+       "      <td>6</td>\n",
+       "      <td>6</td>\n",
+       "      <td>28.0</td>\n",
+       "      <td>0</td>\n",
+       "      <td>rb</td>\n",
+       "      <td>http://en.wikipedia.org/wiki/RB_Formula_One_Team</td>\n",
+       "      <td>RB F1 Team</td>\n",
+       "      <td>Italian</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>6</th>\n",
+       "      <td>7</td>\n",
+       "      <td>7</td>\n",
+       "      <td>26.0</td>\n",
+       "      <td>0</td>\n",
+       "      <td>haas</td>\n",
+       "      <td>http://en.wikipedia.org/wiki/Haas_F1_Team</td>\n",
+       "      <td>Haas F1 Team</td>\n",
+       "      <td>American</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>7</th>\n",
+       "      <td>8</td>\n",
+       "      <td>8</td>\n",
+       "      <td>16.0</td>\n",
+       "      <td>0</td>\n",
+       "      <td>sauber</td>\n",
+       "      <td>http://en.wikipedia.org/wiki/Sauber_Motorsport</td>\n",
+       "      <td>Sauber</td>\n",
+       "      <td>Swiss</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>8</th>\n",
+       "      <td>9</td>\n",
+       "      <td>9</td>\n",
+       "      <td>16.0</td>\n",
+       "      <td>0</td>\n",
+       "      <td>aston_martin</td>\n",
+       "      <td>http://en.wikipedia.org/wiki/Aston_Martin_in_F...</td>\n",
+       "      <td>Aston Martin</td>\n",
+       "      <td>British</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>9</th>\n",
+       "      <td>10</td>\n",
+       "      <td>10</td>\n",
+       "      <td>11.0</td>\n",
+       "      <td>0</td>\n",
+       "      <td>alpine</td>\n",
+       "      <td>http://en.wikipedia.org/wiki/Alpine_F1_Team</td>\n",
+       "      <td>Alpine F1 Team</td>\n",
+       "      <td>French</td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n",
+       "</div>"
+      ],
+      "text/plain": [
+       "   position positionText  points  wins constructorId  \\\n",
+       "0         1            1   362.0     7       mclaren   \n",
+       "1         2            2   165.0     0       ferrari   \n",
+       "2         3            3   159.0     0      mercedes   \n",
+       "3         4            4   144.0     2      red_bull   \n",
+       "4         5            5    54.0     0      williams   \n",
+       "5         6            6    28.0     0            rb   \n",
+       "6         7            7    26.0     0          haas   \n",
+       "7         8            8    16.0     0        sauber   \n",
+       "8         9            9    16.0     0  aston_martin   \n",
+       "9        10           10    11.0     0        alpine   \n",
+       "\n",
+       "                                      constructorUrl constructorName  \\\n",
+       "0               http://en.wikipedia.org/wiki/McLaren         McLaren   \n",
+       "1      http://en.wikipedia.org/wiki/Scuderia_Ferrari         Ferrari   \n",
+       "2  http://en.wikipedia.org/wiki/Mercedes-Benz_in_...        Mercedes   \n",
+       "3       http://en.wikipedia.org/wiki/Red_Bull_Racing        Red Bull   \n",
+       "4  http://en.wikipedia.org/wiki/Williams_Grand_Pr...        Williams   \n",
+       "5   http://en.wikipedia.org/wiki/RB_Formula_One_Team      RB F1 Team   \n",
+       "6          http://en.wikipedia.org/wiki/Haas_F1_Team    Haas F1 Team   \n",
+       "7     http://en.wikipedia.org/wiki/Sauber_Motorsport          Sauber   \n",
+       "8  http://en.wikipedia.org/wiki/Aston_Martin_in_F...    Aston Martin   \n",
+       "9        http://en.wikipedia.org/wiki/Alpine_F1_Team  Alpine F1 Team   \n",
+       "\n",
+       "  constructorNationality  \n",
+       "0                British  \n",
+       "1                Italian  \n",
+       "2                 German  \n",
+       "3               Austrian  \n",
+       "4                British  \n",
+       "5                Italian  \n",
+       "6               American  \n",
+       "7                  Swiss  \n",
+       "8                British  \n",
+       "9                 French  "
+      ]
+     },
+     "execution_count": 33,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "ergastandings = fastf1.ergast.Ergast()\n",
+    "constructor_standings = ergastandings.get_constructor_standings(2025).content[0]\n",
+    "constructor_standings.head(10)"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 2,

app.py

@@ -1,35 +1,33 @@
-import json
-import random
-import datetime
 import gradio as gr
 
 # Local modules
 import tools
 from utils.constants import (
-    AVAILABLE_SESSION_TYPES,
-    DROPDOWN_SESSION_TYPES
+    DRIVER_NAMES,
+    CONSTRUCTOR_NAMES,
+    CURRENT_YEAR
 )
 
-# Variables
-CURRENT_YEAR = datetime.datetime.now().year
-
-
-# Load in driver names
-with open("assets/driver_names.json") as f:
-    driver_names = json.load(f)["drivers"]
-with open("assets/constructor_details.json") as f:
-    constructor_names = [constructor["constructor"] for constructor in json.load(f)["constructors"]]
-
-
 iface_driver_championship_standings = gr.Interface(
     fn=tools.driver_championship_standings,
     inputs=[
         gr.Number(label="Calendar year", value=CURRENT_YEAR, minimum=1950, maximum=CURRENT_YEAR),
-        gr.Dropdown(driver_names)
+        gr.Dropdown(label="Driver", choices=DRIVER_NAMES)
     ],
     outputs="text",
     title="Driver Championship Standings",
-    description="Get the championship standings for the given driver"
+    description="Get the driver championship standings"
+)
+
+iface_constructor_championship_standings = gr.Interface(
+    fn=tools.constructor_championship_standings,
+    inputs=[
+        gr.Number(label="Calendar year", value=CURRENT_YEAR, minimum=1950, maximum=CURRENT_YEAR),
+        gr.Dropdown(label="Constructor", choices=CONSTRUCTOR_NAMES)
+    ],
+    outputs="text",
+    title="Constructor Championship Standings",
+    description="Get the constructor championship standings"
 )
 
 iface_event_info = gr.Interface(
@@ -60,7 +58,7 @@ iface_track_visualization = gr.Interface(
         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(["speed", "corners", "gear"], label="Visualization type", value="speed"),
-        gr.Dropdown(driver_names)
+        gr.Dropdown(DRIVER_NAMES)
     ],
     outputs="image",
     title="Track Visualizations",
@@ -70,6 +68,7 @@ iface_track_visualization = gr.Interface(
 
 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

tools.py

@@ -73,7 +73,28 @@ def driver_championship_standings(year: int, driver_name: str) -> str:
     
 
 def constructor_championship_standings(year: int, constructor_name: str) -> str:
-    pass
+
+    team_mapping = {
+    "McLaren": "McLaren",
+    "Ferrari": "Ferrari",
+    "Red Bull Racing": "Red Bull",
+    "Mercedes": "Mercedes",
+    "Aston Martin": "Aston Martin",
+    "Alpine": "Alpine F1 Team",
+    "Haas": "Haas F1 Team",
+    "Racing Bulls": "RB F1 Team",
+    "Williams": "Williams",
+    "Kick Sauber": "Sauber"
+    }
+
+    ergast = fastf1.ergast.Ergast()
+    constructor_standings = ergast.get_constructor_standings(year).content[0]
+    constructor_standing = constructor_standings[["position", "points", "wins", "constructorName"]].reset_index(drop=True)
+    mapped_name = team_mapping[constructor_name]
+    constructor_standing = constructor_standing[constructor_standing["constructorName"] == mapped_name]
+    suffix = "st" if constructor_standing['position'].iloc[0] == 1 else "nd" if constructor_standing['position'].iloc[0] == 2 else "rd" if constructor_standing['position'].iloc[0] == 3 else "th"
+    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
 
 
 

utils/constants.py

@@ -1,3 +1,5 @@
+import json
+import datetime
 
 AVAILABLE_SESSION_TYPES = [
     "fp1", "fp2", "fp3", "q", "s", "ss", "sq", "r",
@@ -8,4 +10,15 @@ AVAILABLE_SESSION_TYPES = [
 DROPDOWN_SESSION_TYPES = [
     "practice 1", "practice 2", "practice 3", "sprint",
     "sprint shootout", "sprint qualifying", "qualifying", "race"
-    ]
+    ]
+
+# Load in driver names
+with open("assets/driver_names.json") as f:
+    DRIVER_NAMES: list[str] = json.load(f)["drivers"]
+
+# Load in constructor names
+with open("assets/constructors.json") as f:
+    CONSTRUCTOR_NAMES: list[str] = json.load(f)["constructors"]
+
+# Variables
+CURRENT_YEAR = datetime.datetime.now().year