Update app.py

app.py

@@ -38,15 +38,15 @@ if st.session_state.stage == "crawl-input":
 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["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]