added new interface/tools

api_playground.ipynb

@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 1,
    "id": "b7d2515e",
    "metadata": {},
    "outputs": [
@@ -19,6 +19,7 @@
     "import pandas as pd\n",
     "from urllib.request import urlopen\n",
     "from pprint import pprint\n",
+    "from utils.parser_utils import parse_event_info, parse_season_calendar\n",
     "import json\n",
     "print(\"Successfully loaded in libraries\")"
    ]
@@ -33,41 +34,302 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 20,
-   "id": "8109aed6",
+   "execution_count": 3,
+   "id": "2815be5b",
    "metadata": {},
    "outputs": [
     {
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "req         WARNING \tDEFAULT CACHE ENABLED! (107.63 KB) /home/arre/.cache/fastf1\n",
-      "core           INFO \tLoading data for Bahrain Grand Prix - Race [v3.5.3]\n",
-      "req            INFO \tNo cached data found for session_info. Loading data...\n",
-      "_api           INFO \tFetching session info data...\n",
-      "req            INFO \tData has been written to cache!\n",
-      "req            INFO \tNo cached data found for driver_info. Loading data...\n",
-      "_api           INFO \tFetching driver list...\n",
-      "req            INFO \tData has been written to cache!\n",
-      "req            INFO \tNo cached data found for race_control_messages. Loading data...\n",
-      "_api           INFO \tFetching race control messages...\n",
-      "req            INFO \tData has been written to cache!\n",
-      "core           INFO \tFinished loading data for 20 drivers: ['81', '63', '4', '16', '44', '1', '10', '31', '22', '87', '12', '23', '6', '7', '14', '30', '18', '5', '55', '27']\n"
+      "req         WARNING \tDEFAULT CACHE ENABLED! (173.43 MB) /home/arre/.cache/fastf1\n"
      ]
     },
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Lewis: Bono my Tyres are dead, oh wait I'm in a Ferrari now\n"
+      "['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",
-    "session.load(telemetry=False, laps=False, weather=False)\n",
-    "driver = session.get_driver('HAM')\n",
-    "print(f\"{driver['FirstName']}: Bono my Tyres are dead, oh wait I'm in a Ferrari now\")"
+    "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",
+      "\n",
+      "ATTR: f1_api_support: True\n",
+      "\n",
+      "ATTR: get_circuit_info: <bound method Session.get_circuit_info of 2025 Season Round 4: Bahrain Grand Prix - Race>\n",
+      "\n",
+      "ATTR: get_driver: <bound method Session.get_driver of 2025 Season Round 4: Bahrain Grand Prix - Race>\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"
+     ]
+    }
+   ],
+   "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}\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 10,
+   "id": "44a4b93b",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "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"
+     ]
+    }
+   ],
+   "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"
+     ]
+    }
+   ],
+   "source": [
+    "event2 = fastf1.get_event(2025, 'Bahrain')\n",
+    "print(type(event2))\n",
+    "print(event2.keys())\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "id": "32f180c1",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "req         WARNING \tDEFAULT CACHE ENABLED! (173.47 MB) /home/arre/.cache/fastf1\n"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "RoundNumber                              1\n",
+      "Country                          Australia\n",
+      "Location                         Melbourne\n",
+      "OfficialEventName                         \n",
+      "EventDate              2007-03-18 03:00:00\n",
+      "EventName            Australian Grand Prix\n",
+      "EventFormat                   conventional\n",
+      "Session1                        Practice 1\n",
+      "Session1Date                          None\n",
+      "Session1DateUtc        2007-03-16 00:00:00\n",
+      "Session2                        Practice 2\n",
+      "Session2Date                          None\n",
+      "Session2DateUtc        2007-03-16 00:00:00\n",
+      "Session3                        Practice 3\n",
+      "Session3Date                          None\n",
+      "Session3DateUtc        2007-03-17 00:00:00\n",
+      "Session4                        Qualifying\n",
+      "Session4Date                          None\n",
+      "Session4DateUtc        2007-03-17 00:00:00\n",
+      "Session5                              Race\n",
+      "Session5Date                          None\n",
+      "Session5DateUtc        2007-03-18 03:00:00\n",
+      "F1ApiSupport                         False\n",
+      "Name: 0, dtype: object\n",
+      "Season calendar:\n",
+      "Round 1 : Australian Grand Prix - Melbourne, Australia (2007-03-16 - 2007-03-18)\n",
+      "Round 2 : Malaysian Grand Prix - Kuala Lumpur, Malaysia (2007-04-06 - 2007-04-08)\n",
+      "Round 3 : Bahrain Grand Prix - Sakhir, Bahrain (2007-04-13 - 2007-04-15)\n",
+      "Round 4 : Spanish Grand Prix - Montmeló, Spain (2007-05-11 - 2007-05-13)\n",
+      "Round 5 : Monaco Grand Prix - Monte-Carlo, Monaco (2007-05-25 - 2007-05-27)\n",
+      "Round 6 : Canadian Grand Prix - Montreal, Canada (2007-06-08 - 2007-06-10)\n",
+      "Round 7 : United States Grand Prix - Indianapolis, USA (2007-06-15 - 2007-06-17)\n",
+      "Round 8 : French Grand Prix - Magny Cours, France (2007-06-29 - 2007-07-01)\n",
+      "Round 9 : British Grand Prix - Silverstone, UK (2007-07-06 - 2007-07-08)\n",
+      "Round 10 : European Grand Prix - Nürburg, Germany (2007-07-20 - 2007-07-22)\n",
+      "Round 11 : Hungarian Grand Prix - Budapest, Hungary (2007-08-03 - 2007-08-05)\n",
+      "Round 12 : Turkish Grand Prix - Istanbul, Turkey (2007-08-24 - 2007-08-26)\n",
+      "Round 13 : Italian Grand Prix - Monza, Italy (2007-09-07 - 2007-09-09)\n",
+      "Round 14 : Belgian Grand Prix - Spa, Belgium (2007-09-14 - 2007-09-16)\n",
+      "Round 15 : Japanese Grand Prix - Oyama, Japan (2007-09-28 - 2007-09-30)\n",
+      "Round 16 : Chinese Grand Prix - Shanghai, China (2007-10-05 - 2007-10-07)\n",
+      "Round 17 : Brazilian Grand Prix - São Paulo, Brazil (2007-10-19 - 2007-10-21)\n"
+     ]
+    }
+   ],
+   "source": [
+    "schedule = fastf1.get_event_schedule(2007)\n",
+    "print(schedule.iloc[0])\n",
+    "print(parse_season_calendar(schedule))"
    ]
   },
   {
@@ -424,7 +686,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": ".venv",
+   "display_name": "hackaton",
    "language": "python",
    "name": "python3"
   },
@@ -438,7 +700,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.12.3"
+   "version": "3.12.9"
   }
  },
  "nbformat": 4,

app.py

@@ -1,6 +1,18 @@
+import json
 import random
+import datetime
 import gradio as gr
 
+# Local modules
+import tools
+from utils.constants import (
+    AVAILABLE_SESSION_TYPES,
+    DROPDOWN_SESSION_TYPES
+)
+
+# Variables
+CURRENT_YEAR = datetime.datetime.now().year
+
 def driver_championship_score(driver_name: str) -> str:
     """
     Get the championship score for the given driver.
@@ -27,26 +39,57 @@ def driver_position(driver_name: str) -> str:
     return f"Driver {driver_name} is in position {random.randint(1, 20)}"
 
 
+# Load in driver names
+with open("assets/driver_names.json") as f:
+    driver_names = json.load(f)["drivers"]
 
-# Create interfaces for each tool
 iface1 = gr.Interface(
     fn=driver_championship_score,
-    inputs="text",
+    inputs=gr.Dropdown(driver_names),
     outputs="text",
-    title="[Dummy] Driver Championship Score"
+    title="[WIP] Driver Championship Score",
+    description="Get the championship score for the given driver"
 )
 
 iface2 = gr.Interface(
     fn=driver_position,
-    inputs="text",
+    inputs=gr.Dropdown(driver_names),
+    outputs="text",
+    title="[WIP] Driver Position",
+    description="Get the current position of the given driver"
+)
+
+iface3 = gr.Interface(
+    fn=tools.get_event_info,
+    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")
+    ],
     outputs="text",
-    title="[Dummy] Driver Position"
+    title="Event Info",
+    description="Get the Grand Prix event info for a specific race week"
 )
 
-# Combine into tabs into server
+iface4 = gr.Interface(
+    fn=tools.get_season_calendar,
+    inputs=[
+        gr.Number(label="Calendar year", value=CURRENT_YEAR, minimum=1950, maximum=CURRENT_YEAR),
+    ],
+    outputs="text",
+    title="Season Calendar",
+    description="Get the season calendar information for the given year"
+)
+
+
+interface_list = [iface1, iface2, iface3, iface4]
+tab_names = ["[WIP] Driver Championship Score", "[WIP] Driver Position", "Event Info", "Season Calendar"]
+
+
+# Combine all the interfaces into a single TabbedInterface
 gradio_server = gr.TabbedInterface(
-    [iface1, iface2],
-    tab_names=["Driver Championship Score", "Driver Position"],
+    interface_list,
+    tab_names=tab_names,
     title="Formula 1 MCP server"
 )
 

tools.py

@@ -1 +1,66 @@
-## Here we implement the MCP tools
+"""
+Here we implement the MCP tools
+
+Tools to implement
+- session result
+- driver info
+- championship standings
+- driver standings
+- compare drivers
+
+"""
+
+
+## Here we implement the MCP tools
+import fastf1
+from utils.constants import AVAILABLE_SESSION_TYPES
+
+from fastf1.core import Session
+from typing import Union
+
+from utils import parser_utils
+
+# 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
+    """
+
+    # Check if session type is valid
+    if isinstance(session_type, str):
+        if session_type.lower() not in AVAILABLE_SESSION_TYPES:
+            return f"Session type {session_type} is not available. Supported session types: {AVAILABLE_SESSION_TYPES}"
+
+    return fastf1.get_session(year, round, session_type)
+
+def get_event_info(year: int, round: gp, format: str) -> str:
+    """ Returns the event info for the specified session """
+
+    event = fastf1.get_session(year, round, "race").event # Event object is the same for all sessions, so hardcode "race"
+    if format == "human":
+        data_interval = f"{event['Session1DateUtc'].date()} - {event['Session5DateUtc'].date()}"
+        event_string = f"Round {event['RoundNumber']} : {event['EventName']} - {event['Location']}, {event['Country']} ({data_interval})"
+        return event_string
+    elif format == "LLM":
+        return parser_utils.parse_event_info(event)
+
+
+def get_constructor_standings(year: int) -> str:
+    pass
+
+def get_driver_standings(year): 
+    pass
+
+
+
+if __name__ == "__main__":
+    session = get_session(2024, 1, "fp1")
+    session.load()