Update app.py

app.py

@@ -28,9 +28,17 @@ def brave_search(query, count=3):
         print(f"Error: {response.status_code}, {response.text}")
         return []
 
+@traceable 
+def query_teapot(context, query):
+    response = teapot_ai.query(
+        context=context,
+        query=user_input
+    )
+
+    return response
+    
 
 # Function to handle the chat with TeapotAI
-@traceable 
 def handle_chat(user_input, teapot_ai):
     results = brave_search(user_input)
 
@@ -41,10 +49,7 @@ def handle_chat(user_input, teapot_ai):
 
     context="\n".join(documents)
     
-    response = teapot_ai.query(
-        context=context,
-        query=user_input
-    )
+    response = query_teapot(context, query)
     
     # response = teapot_ai.chat([
     #     {