Update app.py

app.py

@@ -143,9 +143,10 @@ with chat_interface:
                 st.markdown(message['content'])
 
                 # 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
+                # if idx == len(st.session_state.chat_history) - 1 and message['role'] == 'assistant':
+
+if st.button("Continue"):
+    st.session_state.need_continue = True
 
 # User input area (moved to the bottom)
 st.session_state.user_text = st.chat_input(placeholder="Enter your text here.")