Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import requests
 from fastapi import FastAPI
+from fastapi.testclient import TestClient
 from routes import query_handler, purchase, order_management, account_management, customer_support, search_products
 
 app = FastAPI()
@@ -17,6 +18,8 @@ app.include_router(search_products.router,
 app.include_router(query_handler.router,
                    prefix="/query-handler", tags=["query-handler"])
 
+# Initialize the TestClient with the FastAPI app
+client = TestClient(app)
 
 def fastlane_agent(message, history):
 
@@ -28,8 +31,7 @@ def fastlane_agent(message, history):
             {"role": "assistant", "content": assistant})
     history_openai_format.append({"role": "user", "content": message})
 
-    response = requests.post(
-        "http://localhost:8000/query-handler/", json={"text": message, "history": history_openai_format})
+    response = client.post("/query-handler/", json={"text": message, "history": history_openai_format})
     if response.status_code == 200:
         return  response.json().get("generative response")
     else: