alok nath ji

app.py

@@ -1,13 +1,10 @@
-# app.py
-
+import gradio as gr
 from fastapi import FastAPI, HTTPException
 from fastapi.exceptions import RequestValidationError
 from fastapi.responses import JSONResponse
-import gradio as gr
-import requests
-import os
-from contextlib import asynccontextmanager
+from fastapi.middleware.wsgi import WSGIMiddleware
 
+# FastAPI setup
 app = FastAPI()
 
 @app.exception_handler(RequestValidationError)
@@ -38,21 +35,26 @@ async def divide(a: float, b: float):
   if b == 0:
       raise HTTPException(status_code=400, detail="Cannot divide by zero")
   return {"result": a / b}
-# Gradio Interface
-
-# HUGGING FACE URL
-BACKEND_URL = "https://scruffykay2-100x-assignment-3.hf.space"
 
+# Gradio interface
 def calculate(operation, num1, num2):
   try:
       num1, num2 = float(num1), float(num2)
-      response = requests.get(f"{BACKEND_URL}/{operation}/{num1}/{num2}")
-      if response.status_code == 200:
-          return response.json()["result"]
+      if operation == "add":
+          result = add(num1, num2)
+      elif operation == "subtract":
+          result = subtract(num1, num2)
+      elif operation == "multiply":
+          result = multiply(num1, num2)
+      elif operation == "divide":
+          result = divide(num1, num2)
       else:
-          return f"Error: {response.json()['detail']}"
-  except requests.RequestException:
-      return "Error: Unable to connect to the backend"
+          return "Invalid operation"
+      return result["result"]
+  except HTTPException as e:
+      return f"Error: {e.detail}"
+  except Exception as e:
+      return f"Error: {str(e)}"
 
 iface = gr.Interface(
   fn=calculate,
@@ -64,13 +66,10 @@ iface = gr.Interface(
   outputs="text"
 )
 
-@asynccontextmanager
-async def lifespan(app: FastAPI):
-  iface.launch()
-  yield
-
-app.router.lifespan_context = lifespan
+# Mount Gradio app to FastAPI
+app = gr.mount_gradio_app(app, iface, path="/")
 
+# For running locally
 if __name__ == "__main__":
   import uvicorn
-  uvicorn.run(app, host="0.0.0.0", port=8000)
+  uvicorn.run(app, host="0.0.0.0", port=7860)