Update app.py

app.py

@@ -1,4 +1,5 @@
-import gradio as gr
+
+   import gradio as gr
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 import uvicorn
@@ -32,14 +33,16 @@ async def divide(numbers: Numbers):
     return {"result": numbers.x / numbers.y}
 
 # Define the Gradio interface function
-def perform_operation(operation: str, x: float, y: float) -> float:
+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}"
-    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')}"
+    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}"
 
 # Define the Gradio interface
 demo = gr.Interface(
@@ -60,4 +63,3 @@ 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=8000)
-