Update app.py

app.py

@@ -51,6 +51,10 @@ def handle_chat(user_input, teapot_ai):
     documents = [desc.replace('<strong>','').replace('</strong>','') for _, desc, _ in results]
     print(documents)
 
+    for document in documents:
+        st.sidebar.write(f"## {documents}")
+        st.write("---")
+
     context = "\n".join(documents)
     prompt = "You are Teapot, an open-source AI assistant optimized for low-end devices, providing short, accurate responses without hallucinating while excelling at information extraction and text summarization."
     response = query_teapot(prompt, context, user_input, teapot_ai)
@@ -68,7 +72,7 @@ def hash_documents(documents):
 def main():
     st.set_page_config(page_title="TeapotAI Chat", page_icon=":robot_face:", layout="wide")
     
-    st.sidebar.header("Document Input (for RAG)")
+    st.sidebar.header("Retrieval Augmented Generation")
     user_documents = st.sidebar.text_area("Enter documents, each on a new line", value="\n".join(default_documents))
     
     documents = [doc.strip() for doc in user_documents.split("\n") if doc.strip()]