Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import requests
 from MathEngine import calculate, CalculationRequest, app as fastapi_app
 import uvicorn
 from multiprocessing import Process
@@ -29,6 +30,13 @@ def backspace():
     current_input = current_input[:-1]
     return current_input
 
+def test_fastapi(a, b, operation):
+    url = "http://0.0.0.0:8000/calculate"
+    payload = {"a": a, "b": b, "operation": operation}
+    headers = {"Content-Type": "application/json"}
+    response = requests.post(url, json=payload, headers=headers)
+    return response.json()
+
 def create_interface():
     with gr.Blocks() as demo:
         a = gr.Number(label="A", interactive=True)
@@ -52,11 +60,17 @@ def create_interface():
             gr.Button("Multiply").click(lambda a, b: gradio_calculate(a, b, "multiply"), inputs=[a, b], outputs=result)
             gr.Button("Divide").click(lambda a, b: gradio_calculate(a, b, "divide"), inputs=[a, b], outputs=result)
 
+        with gr.Row():
+            gr.Button("Test FastAPI Add").click(lambda a, b: test_fastapi(a, b, "add"), inputs=[a, b], outputs=result)
+            gr.Button("Test FastAPI Subtract").click(lambda a, b: test_fastapi(a, b, "subtract"), inputs=[a, b], outputs=result)
+            gr.Button("Test FastAPI Multiply").click(lambda a, b: test_fastapi(a, b, "multiply"), inputs=[a, b], outputs=result)
+            gr.Button("Test FastAPI Divide").click(lambda a, b: test_fastapi(a, b, "divide"), inputs=[a, b], outputs=result)
+
     return demo
 
 def run_gradio():
     demo = create_interface()
-    demo.launch(share=True)
+    demo.launch(share=True)  # Ensure share=True to make the Gradio interface publicly accessible
 
 def run_fastapi():
     uvicorn.run(fastapi_app, host="0.0.0.0", port=8000)