Update app.py

app.py

@@ -75,23 +75,25 @@ def main():
                        page_icon=":books:")
     st.write(css, unsafe_allow_html=True)
 
-    if "conversation" not in st.session_state:
-        st.session_state.conversation = None
-    if "chat_history" not in st.session_state:
-        st.session_state.chat_history = None
-
+    
     st.header("Chat with multiple PDFs :books:")
 
-    with st.chat_message("Assistant"):
-        st.write("Hello my name is Robert, how can i help you? ")
+    if "messages" not in st.session_state:
+    st.session_state["messages"] = [
+        {"role": "assistant", "content": "Hi, I'm a chatbot who can search the web. How can I help you?"}
+    ]
+
+
     user_question = st.text_input("Ask a question about your documents:")
-    with st.chat_message("User"):
-        st.write(user_question)
+    
+ 
     if user_question:
         handle_userinput(user_question)
 
 
 def handle_userinput(user_question):
+    st.session_state.messages.append({"role": "user", "content": user_question})
+    st.chat_message("user").write(user_question)
     retriever = create_retriever_from_chroma(data_path, vectorstore_path="docs/chroma/", search_type='mmr', k=7, chunk_size=250, chunk_overlap=20)
     docs = retriever.invoke(user_question)
     
@@ -99,8 +101,9 @@ def handle_userinput(user_question):
     
     Rag_chain = create_conversational_rag_chain(retriever)
     response = rag_chain.invoke({"context": doc_txt, "question": user_question})
-    with st.chat_message("Assistant"):
-        st.write(response)
+    st.session_state.messages.append({"role": "assistant", "content": response})
+    st.chat_message("assistant").write(response)
+
 
 
 def create_conversational_rag_chain(retriever):