Update app.py

app.py

@@ -39,14 +39,14 @@ if st.session_state.stage == "prompt-input":
     
     prompt = st.chat_input("Ask the documentation a question!")
     
+    for message in st.session_state.messages:
+            with st.chat_message(message["role"]):
+                st.markdown(message)
+                
     if prompt:
         with st.chat_message("user"):
             st.markdown(prompt)
-            
-        for message in st.session_state.messages:
-            with st.chat_message(message["role"]):
-                st.markdown(message)    
-                
+        
         st.session_state.messages.append({"role":"user","message":prompt})
         results = st.session_state.collection.query(query_texts=[prompt],n_results=5)
         infos = results["documents"][0]