Update app.py

app.py

@@ -154,14 +154,15 @@ def refresh_chat():
     
                     # If this is the last assistant message, add the "Continue" button
                     # if idx == len(st.session_state.chat_history) - 1 and message['role'] == 'assistant':
-            if st.button("Continue"):
-                st.session_state.need_continue = True
 
 refresh_chat()
 
 # User input area (moved to the bottom)
 st.session_state.user_text = st.chat_input(placeholder="Enter your text here.")
 
+if st.button("Continue"):
+    st.session_state.need_continue = True
+                
 # When the user enters new text
 if st.session_state.user_text:
     # Display the user's message