trynna fix the remember word -yes/no bug again

pages/2_Context-based_chatbot.py

@@ -261,32 +261,30 @@ if st.session_state.is_helpful_context['ask'] == True:
 if st.session_state.actions_context[-1] == 'cue':
   guessed = False
   write_bot('What do you want to see?', remember=False, blink=False)
+  col1, col2, col3, col4, col5 = st.columns(5)
+
 
+  with col1:
+    b1 = st.button("Next letter", key="1")
+  with col2:
+    b2 = st.button("Related words")
+  with col3:
+    b3 = st.button("Next word", key="2")
+  with col4:
+    b4 = st.button("All words", key="3")
+
+  # JS
+  #if get_available_cues(target):
+    # avail_cues = get_available_cues(target)
+    #cues_buttons = {cue_type: st.button(cue_type) for cue_type in avail_cues}
+
+  b5 = st.button("I remembered the word!", key="4", type='primary')
+  b6 = st.button("Exit", key="5", type='primary')
+  new = st.button('Play again', key=64, type='primary')
   while guessed == False:
     # JS
     word_count = st.session_state.counter_context["word_count"]
     target = st.session_state.results_context["results_context"][word_count]
-    
-    col1, col2, col3, col4, col5 = st.columns(5)
-  
-
-    with col1:
-      b1 = st.button("Next letter", key="1")
-    with col2:
-      b2 = st.button("Related words")
-    with col3:
-      b3 = st.button("Next word", key="2")
-    with col4:
-      b4 = st.button("All words", key="3")
-
-    # JS
-    #if get_available_cues(target):
-     # avail_cues = get_available_cues(target)
-      #cues_buttons = {cue_type: st.button(cue_type) for cue_type in avail_cues}
-
-    b5 = st.button("I remembered the word!", key="4", type='primary')
-    b6 = st.button("Exit", key="5", type='primary')
-    new = st.button('Play again', key=64, type='primary')
 
     if b1:
       st.session_state.counter_context["letter_count"] += 1