Update app.py

app.py

@@ -37,13 +37,12 @@ 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)
             
@@ -63,10 +62,9 @@ if st.session_state.stage == "prompt-input":
         extra_body={},
         model="deepseek/deepseek-r1:free",
         messages=[{"role":"user","content":prompt}])
-
+        full_response = ""
+        message_placeholder = st.empty()
         with st.chat_message("assistant"):
-            full_response = ""
-            message_placeholder = st.empty()
             for chunk in completion.choices[0].message.content.split():
                 full_response += chunk + " "
                 time.sleep(0.05)