Update app.py

app.py

@@ -77,8 +77,13 @@ def main():
     
     st.header("Chat with multiple PDFs :books:")
 
-    
+    if "messages" not in st.session_state:
+        st.session_state["messages"] = [
+        {"role": "assistant", "content": "Hi, I'm a chatbot who is based on lithuanian law documents. 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:")
 
@@ -90,13 +95,9 @@ def main():
         handle_userinput(user_question,retriever)
 
 
-    if "messages" not in st.session_state:
-        st.session_state["messages"] = [
-        {"role": "assistant", "content": "Hi, I'm a chatbot who is based on lithuanian law documents. How can I help you?"}
-    ]
+    
 
-    for msg in st.session_state.messages:
-        st.chat_message(msg["role"]).write(msg["content"])
+    
 
 
 def handle_userinput(user_question,retriever):