from mcp.server.fastmcp import FastMCP
import gradio as gr

# Create a MCP Server
mcp = FastMCP("My first MCP server")

# Defining weather tools
@mcp.tool()
def get_weather(location: str) -> str:
    """
    Get the current weather for a given location.
    
    Args:
        location (str): The name of the location to get the weather for.
    
    Returns:
        str: The current weather information for the specified location.
    """
    return f"The weather for the given {location} : Sunny, 72 F"

@mcp.resource("weather://{location}")
def get_weather_resource(location: str) -> str:
    """
    Resource to get the current weather for the given location.
    
    Args:
        location (str): The name of the location to get the weather for.
    
    Returns:
        str: The weather resource string containing weather information.
    """
    return f"The weather for the given {location} : Sunny, 72 F"

@mcp.prompt()
def weather_prompt(location: str) -> str:
    """
    Create a weather report prompt.
    
    Args:
        location (str): The name of the location for which the weather report prompt should be created.
    
    Returns:
        str: A formatted prompt string to instruct the model to act as a weather expert.
    """
    return f"You are a weather expert and you are supposed to provide weather information for the given location {location}."

@mcp.tool()
def add_two_numbers(a: int, b: int) -> int:
    """
    Add two numbers.
    
    Args:
        a (int): The first number to be added.
        b (int): The second number to be added.
    
    Returns:
        int: The sum of the two numbers.
    """
    return a + b

# Here will be defining the Gradio UI
def weather_ui(location: str) -> str:
    """
    Wrapper for weather UI to fetch weather.
    
    Args:
        location (str): Location name.
    
    Returns:
        str: Weather info string.
    """
    return get_weather(location)

def math_ui(a: int, b: int) -> int:
    """
    Wrapper for math UI to add numbers.
    
    Args:
        a (int): First number.
        b (int): Second number.
    
    Returns:
        int: Sum of the two numbers.
    """
    return add_two_numbers(a, b)

with gr.Blocks(title="My first MCP server") as demo:
    gr.Markdown("## Weather + Math MCP Server")

    with gr.Tab("Weather Service"):
        location_input = gr.Textbox(label="Enter Location")
        weather_output = gr.Textbox(label="Weather Result")
        weather_btn = gr.Button("Get Weather")
        weather_btn.click(fn=weather_ui, inputs=location_input, outputs=weather_output)

    with gr.Tab("Math Service"):
        num_1_input = gr.Number(label="Enter First Number")
        num_2_input = gr.Number(label="Enter Second Number")
        math_output = gr.Number(label="Sum")
        math_btn = gr.Button("Add Numbers")
        math_btn.click(fn=math_ui, inputs=[num_1_input, num_2_input], outputs=math_output)
    
# Running the MCP server
if __name__ == "__main__":
    demo.launch(mcp_server=True)