Update app.py

app.py

@@ -81,14 +81,17 @@ def main():
         st.session_state["messages"] = [
         {"role": "assistant", "content": "Hi, I'm a chatbot who can search the web. How can I help you?"}
     ]
-    
+
+    for msg in st.session_state.messages:
+    st.chat_message(msg["role"]).write(msg["content"])
+
 
     user_question = st.text_input("Ask a question about your documents:")
 
     retriever = create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type='mmr', k=7, chunk_size=250, chunk_overlap=20)
  
     if user_question:
-        handle_userinput(user_question)
+        handle_userinput(user_question,retriever)
 
 
 def handle_userinput(user_question,retriever):