f1-mcp-server / app.py
arre99's picture
fixed random import bug in app and changed url for mcp client to my HF space
d4991e4
raw
history blame
1.26 kB
import random
import gradio as gr
def driver_championship_score(driver_name: str) -> str:
"""
Get the championship score for the given driver.
Args:
driver_name (str): The driver's name
Returns:
int: The driver's championship score
"""
return f"Driver {driver_name} has {random.randint(0, 100)} championship points"
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)}"
# Create interfaces for each tool
iface1 = gr.Interface(
fn=driver_championship_score,
inputs="text",
outputs="text",
title="[Dummy] Driver Championship Score"
)
iface2 = gr.Interface(
fn=driver_position,
inputs="text",
outputs="text",
title="[Dummy] Driver Position"
)
# Combine into tabs into server
gradio_server = gr.TabbedInterface(
[iface1, iface2],
tab_names=["Driver Championship Score", "Driver Position"],
title="Formula 1 MCP server"
)
# Launch the interface and MCP server
if __name__ == "__main__":
gradio_server.launch(mcp_server=True)