Update app.py

app.py

@@ -36,18 +36,24 @@ def handle_chat(user_input, teapot_ai):
     for i, (title, description, url) in enumerate(results, 1):
         documents.append(description.replace('<strong>','').replace('</strong>',''))
     print(documents)
-    teapotai = TeapotAI(documents=documents, settings=TeapotAISettings(rag_num_results=3))
+
+    context="\n".join(documents)
+    
+    response = teapot_ai.query(
+        context=context,
+        query=user_input
+    )
     
-    response = teapot_ai.chat([
-        {
-            "role": "system",
-            "content": "You are Teapot, an open-source AI assistant optimized for running efficiently on low-end devices. You provide short, accurate responses without hallucinating and excel at extracting information and summarizing text."
-        },
-        {
-            "role": "user",
-            "content": user_input
-        }
-    ])
+    # response = teapot_ai.chat([
+    #     {
+    #         "role": "system",
+    #         "content": "You are Teapot, an open-source AI assistant optimized for running efficiently on low-end devices. You provide short, accurate responses without hallucinating and excel at extracting information and summarizing text."
+    #     },
+    #     {
+    #         "role": "user",
+    #         "content": user_input
+    #     }
+    # ])
     return response
 
 def suggestion_button(suggestion_text, teapot_ai):