Update app.py

app.py

@@ -50,6 +50,9 @@ if "system_message" not in st.session_state:
 if "starter_message" not in st.session_state:
     st.session_state.starter_message = "Hello, there! How can I help you today?"
 
+if "can_continue" not in st.session_state:
+    st.session_state.can_continue = False
+
 # Initialize state for continue action
 need_continue = False
 
@@ -171,7 +174,6 @@ 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']):
@@ -189,12 +191,12 @@ if st.session_state.user_text:
                 continue_last=False
             )
             st.markdown(response)
-            can_continue = True
+            st.session_state.can_continue = True
 
     # Clear the user input
     st.session_state.user_text = None
 
-if can_continue:
+if st.session_state.can_continue:
     if st.button("Continue"):
         need_continue = True
 else: