Update app.py

app.py

@@ -1,34 +1,40 @@
-import gradio as gr
-
-def calculator(num1, operation, num2):
-    if operation == "add":
-        return num1 + num2
-    elif operation == "subtract":
-        return num1 - num2
-    elif operation == "multiply":
-        return num1 * num2
-    elif operation == "divide":
-        if num2 == 0:
-            raise gr.Error("Cannot divide by zero!")
-        return num1 / num2
-
-demo = gr.Interface(
-    calculator,
-    [
-        "number",
-        gr.Radio(["add", "subtract", "multiply", "divide"]),
-        "number"
-    ],
-    "number",
-    examples=[
-        [45, "add", 3],
-        [3.14, "divide", 2],
-        [144, "multiply", 2.5],
-        [0, "subtract", 1.2],
-    ],
-    title="Toy Calculator",
-    description="Here's a sample toy calculator.",
-)
-
-if __name__ == "__main__":
-    demo.launch()
+from pydantic import BaseModel
+from fastapi import FastAPI
+
+app = FastAPI()
+
+
+class Numbers(BaseModel):
+    x: float
+    y: float
+
+@app.post("/add")
+def add_numbers_api(numbers: Numbers):
+    return {"result": add_numbers(numbers.x, numbers.y)}
+
+@app.post("/subtract")
+def subtract_numbers_api(numbers: Numbers):
+    return {"result": subtract_numbers(numbers.x, numbers.y)}
+
+@app.post("/multiply")
+def multiply_numbers_api(numbers: Numbers):
+    return {"result": multiply_numbers(numbers.x, numbers.y)}
+
+@app.post("/divide")
+def divide_numbers_api(numbers: Numbers):
+    return {"result": divide_numbers(numbers.x, numbers.y)}
+
+
+def add_numbers(x, y):
+    return x + y
+
+def subtract_numbers(x, y):
+    return x - y
+
+def multiply_numbers(x, y):
+    return x * y
+
+def divide_numbers(x, y):
+    if y == 0:
+        raise ValueError("Cannot divide by zero")
+    return x / y