Update app.py

app.py

@@ -37,14 +37,16 @@ if st.session_state.stage == "crawl-input":
         st.session_state.stage = "remove-crawl-input-widgets"
 
 if st.session_state.stage == "prompt-input":
-    for message in st.session_state.messages:
-        with st.chat_message(message["role"]):
-            st.markdown(message)
     prompt = st.chat_input("Ask the documentation a question!")
     
     if prompt:
+        for message in st.session_state.messages:
+            with st.chat_message(message["role"]):
+                st.markdown(message)
+                
         with st.chat_message("user"):
             st.markdown(prompt)
+            
         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]