Update app.py

app.py

@@ -56,13 +56,6 @@ for message in st.session_state.messages:
     with st.chat_message(message["role"]):
         st.markdown(message["content"])
 
-
-if st.button("Reset"):
-        st.session_state.knowledge_base = None
-        st.session_state.messages = []
-        st.rerun()  # Rerun the script
-
-
 # Accept user input
 if question := st.chat_input("What is up?"):
     # Add user message to chat history
@@ -75,4 +68,10 @@ if question := st.chat_input("What is up?"):
     with st.chat_message("assistant"):
         response = docqa.run(question)
         st.markdown(response)
-    st.session_state.messages.append({"role": "assistant", "content": response})
+    st.session_state.messages.append({"role": "assistant", "content": response})
+
+
+if st.button("Reset"):
+        st.session_state.knowledge_base = None
+        st.session_state.messages = []
+        st.rerun()  # Rerun the script