Update app.py

app.py

@@ -1,64 +1,39 @@
-import gradio as gr
-from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel
-import uvicorn
-import requests
-
-# 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")
-async def add(numbers: Numbers):
-    return {"result": numbers.x + numbers.y}
+def add_numbers_api(numbers: Numbers):
+    return {"result": add_numbers(numbers.x, numbers.y)}
 
 @app.post("/subtract")
-async def subtract(numbers: Numbers):
-    return {"result": numbers.x - numbers.y}
+def subtract_numbers_api(numbers: Numbers):
+    return {"result": subtract_numbers(numbers.x, numbers.y)}
 
 @app.post("/multiply")
-async def multiply(numbers: Numbers):
-    return {"result": numbers.x * numbers.y}
+def multiply_numbers_api(numbers: Numbers):
+    return {"result": multiply_numbers(numbers.x, numbers.y)}
 
 @app.post("/divide")
-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) -> str:
-    # Construct the URL with the correct scheme and host
-    url = f"http://127.0.0.1:8000/{operation}"
+def divide_numbers_api(numbers: Numbers):
     try:
-        response = requests.post(url, json={"x": x, "y": y})
-        response.raise_for_status()  # Raise an error for bad responses
-        result = response.json().get("result")
-        return str(result) if result is not None else "Error: Invalid response"
-    except requests.exceptions.RequestException as e:
-        return f"Error: {e}"
+        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
 
-# 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."
-)
+def subtract_numbers(x: float, y: float) -> float:
+    return x - y
 
-# Mount Gradio app onto FastAPI
-gr.mount_gradio_app(app, demo, path="/")
+def multiply_numbers(x: float, y: float) -> float:
+    return x * y
 
-# Run the FastAPI app with Uvicorn
-if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=8000)
+def divide_numbers(x: float, y: float) -> float:
+    if y == 0:
+        raise ValueError("Cannot divide by zero")
+    return x / y