Update app.py

app.py

@@ -31,11 +31,13 @@ def brave_search(query, count=3):
 # Function to handle the chat with TeapotAI
 def handle_chat(user_input, teapot_ai):
     results = brave_search(user_input)
-    
-    for i, (title, description, url) in enumerate(results, 1):
-      print(f"{i}. {title}\n   {description.replace('<strong>','').replace('</strong>','')}\n   {url}\n")
-      st.write(f"{i}. {title}\n   {description.replace('<strong>','').replace('</strong>','')}\n   {url}\n")
 
+    documents = []
+    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))
+    
     response = teapot_ai.chat([
         {
             "role": "system",