Update app.py

app.py

@@ -1,42 +1,61 @@
-from pydantic import BaseModel
+
+import gradio as gr
 from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+import uvicorn
 
+# Initialize FastAPI app
 app = FastAPI()
 
+# Define the Pydantic model for input validation
 class Numbers(BaseModel):
     x: float
     y: float
 
+# Define the arithmetic operations
 @app.post("/add")
-def add_numbers_api(numbers: Numbers):
-    return {"result": add_numbers(numbers.x, numbers.y)}
+async def add(numbers: Numbers):
+    return {"result": numbers.x + numbers.y}
 
 @app.post("/subtract")
-def subtract_numbers_api(numbers: Numbers):
-    return {"result": subtract_numbers(numbers.x, numbers.y)}
+async def subtract(numbers: Numbers):
+    return {"result": numbers.x - numbers.y}
 
 @app.post("/multiply")
-def multiply_numbers_api(numbers: Numbers):
-    return {"result": multiply_numbers(numbers.x, numbers.y)}
+async def multiply(numbers: Numbers):
+    return {"result": numbers.x * numbers.y"}
 
 @app.post("/divide")
-def divide_numbers_api(numbers: Numbers):
-    try:
-        result = divide_numbers(numbers.x, numbers.y)
-    except ValueError as e:
-        raise HTTPException(status_code=400, detail=str(e))
-    return {"result": result}
-
-def add_numbers(x: float, y: float) -> float:
-    return x + y
-
-def subtract_numbers(x: float, y: float) -> float:
-    return x - y
-
-def multiply_numbers(x: float, y: float) -> float:
-    return x * y
-
-def divide_numbers(x: float, y: float) -> float:
-    if y == 0:
-        raise ValueError("Cannot divide by zero")
-    return x / y
+async def divide(numbers: Numbers):
+    if numbers.y == 0:
+        raise HTTPException(status_code=400, detail="Cannot divide by zero")
+    return {"result": numbers.x / numbers.y}
+
+# Define the Gradio interface function
+def perform_operation(operation: str, x: float, y: float) -> float:
+    url = f"http://127.0.0.1:8000/{operation}"
+    response = requests.post(url, json={"x": x, "y": y})
+    if response.status_code == 200:
+        return response.json().get("result")
+    else:
+        return f"Error: {response.json().get('detail', 'Unknown error')}"
+
+# Define the Gradio interface
+demo = gr.Interface(
+    fn=perform_operation,
+    inputs=[
+        gr.Radio(choices=["add", "subtract", "multiply", "divide"], label="Operation"),
+        gr.Number(label="X"),
+        gr.Number(label="Y")
+    ],
+    outputs="text",
+    title="Arithmetic Operations",
+    description="Select an operation and enter two numbers to perform arithmetic operations."
+)
+
+# Mount Gradio app onto FastAPI
+gr.mount_gradio_app(app, demo, path="/")
+
+# Run the FastAPI app with Uvicorn
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7680)