Update app.py

app.py

@@ -1,68 +1,71 @@
 from mcp.server.fastmcp import FastMCP
 import gradio as gr
 
-mcp = FastMCP("My First MCP server")
+# 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 Degree Fahrenheit"
-
-@mcp.tool()
-def add_two_numbers(a: int, b: int) -> int:
-    """
-    Adds two integers
-
-    Args:
-        a (int): The first number
-        b (int): The second number
-
-    Returns:
-        int: The sum of two numbers
-    """
-    return a + b
+    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 current weather for the given location
-
+    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 the weather information.
+        str: The weather resource string containing weather information.
     """
-    return f"The weather for the given {location}: Sunny, 72 degree Fahrenheit."
+    return f"The weather for the given {location} : Sunny, 72 F"
 
 @mcp.prompt()
-def get_weather_prompt(location: str) -> str:
+def weather_prompt(location: str) -> str:
     """
-    Create a weather report prompt
-
-    Args: 
-        location (str): The name of the location to get the weather for.
-
+    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 "You are a weather expert and you are supposed to provide the weather report"
+    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.
-
+    
+    Args:
+        location (str): Location name.
+    
     Returns:
         str: Weather info string.
     """
@@ -71,31 +74,32 @@ def weather_ui(location: str) -> str:
 def math_ui(a: int, b: int) -> int:
     """
     Wrapper for math UI to add numbers.
-
+    
     Args:
-        a (int): The first number.
-        b (int): The second number.
-
+        a (int): First number.
+        b (int): Second number.
+    
     Returns:
-        int: The sum of two numbers
+        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")
+    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 = 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")
-        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=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)