Update app.py

app.py

@@ -171,6 +171,7 @@ refresh_chat()
 st.session_state.user_text = st.chat_input(placeholder="Enter your text here.")
                 
 # When the user enters new text
+can_continue = False
 if st.session_state.user_text:
     # Display the user's message
     with st.chat_message("user", avatar=st.session_state.avatars['user']):
@@ -188,12 +189,16 @@ if st.session_state.user_text:
                 continue_last=False
             )
             st.markdown(response)
+            can_continue = True
 
     # Clear the user input
     st.session_state.user_text = None
 
-if st.button("Continue"):
-    need_continue = True
+if can_continue:
+    if st.button("Continue"):
+        need_continue = True
+else:
+    need_continue = False
 
 # If "Continue" button was pressed
 if need_continue: