Update app.py

app.py

@@ -16,7 +16,6 @@ prompt_template = """Answer the question only according to the information provi
 # {}"""
 
 st.session_state.messages = []
-
 if not "stage" in st.session_state:
     st.session_state.stage = "crawl-input"
 elif st.session_state.stage == "remove-crawl-input-widgets":
@@ -37,15 +36,17 @@ 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:
         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]
@@ -62,9 +63,10 @@ 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)