feat(app): add function: calculate()

app.py

@@ -42,6 +42,32 @@ def get_dad_joke():
         return data.get("joke", "No joke found.")
     else:
         return "Failed to retrieve a joke."
+
+def calculate(n1, op, n2):
+    """
+    Performs basic arithmetic operations between two numbers.
+    This function takes two numbers and an operator as input and returns the result
+    of the arithmetic operation. Supported operations are addition (+), subtraction (-),
+    multiplication (*), and division (/). For division, checks for division by zero.
+    Args:
+        n1 (float): First number
+        op (str): Operator ('+', '-', '*', or '/')
+        n2 (float): Second number
+    Returns:
+        float: Result of the arithmetic operation if valid
+        str: "Error" if invalid operation (like division by zero)
+    Examples:
+        >>> calculate(10, "+", 5)
+        15
+        >>> calculate(10, "/", 0)
+        "Error"
+    """
+
+    if op == "+": return n1 + n2
+    if op == "-": return n1 - n2
+    if op == "*": return n1 * n2
+    if op == "/" and n2 != 0: return n1 / n2
+    return "Error"
     
 # demo = gr.Interface(fn=greet, inputs="text", outputs="text")
 with gr.Blocks() as demo:
@@ -53,4 +79,20 @@ with gr.Blocks() as demo:
     joke_output = gr.Textbox(label="Dad Joke")
     joke_btn.click(fn=get_dad_joke, outputs=joke_output, api_name="get_dad_joke")
 
+    with gr.Row():
+        num1 = gr.Number(label="First Number")
+        operation = gr.Dropdown(
+            choices=["+", "-", "*", "/"],
+            label="Operation"
+        )
+        num2 = gr.Number(label="Second Number")
+    
+    calc_btn = gr.Button("Calculate")
+    calc_output = gr.Number(label="Result")
+    calc_btn.click(
+        fn=calculate,
+        inputs=[num1, operation, num2],
+        outputs=calc_output,
+        api_name="calculate")
+
 demo.launch(mcp_server=True)