Update app.py

app.py

@@ -1,32 +1,22 @@
-from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel
+import gradio as gr
+from MathEngine import calculate, CalculationRequest
+# Gradio Interface
+def gradio_calculate(a, b, operation):
+    request = CalculationRequest(a=a, b=b, operation=operation)
+    response = calculate(request)
+    return response["result"]
 
-app = FastAPI()
+iface = gr.Interface(
+    fn=gradio_calculate,
+    inputs=[
+        gr.Number(label="A"),
+        gr.Number(label="B"),
+        gr.Textbox(label="Operation")
+    ],
+    outputs="text",
+    title="Calculator",
+    description="A simple calculator using FastAPI and Gradio"
+)
 
-class CalculationRequest(BaseModel):
-    a: float
-    b: float
-    operation: str
-
-@app.post("/calculate")
-def calculate(request: CalculationRequest):
-    a = request.a
-    b = request.b
-    operation = request.operation.lower()
-
-    if operation == "add":
-        result = a + b
-    elif operation == "subtract":
-        result = a - b
-    elif operation == "multiply":
-        result = a * b
-    elif operation == "divide":
-        if b == 0:
-            raise HTTPException(status_code=400, detail="Division by zero is not allowed")
-        result = a / b
-    else:
-        raise HTTPException(status_code=400, detail="Invalid operation")
-
-    return {"result": result}
-
-# To run the application, use the command: uvicorn app.main:app --reload
+if __name__ == "__main__":
+    iface.launch()