count states change

app.py

@@ -139,13 +139,14 @@ def ask_if_helped():
   new = st.button('I have a new word', key=62)
   if y:
     write_bot("I am happy to help!")
+    st.session_state.is_helpful['ask'] = False
   elif n:
     st.session_state.actions.append('cue')
     st.session_state.is_helpful['ask'] = False
     #cue_generation()
   elif new:
     write_bot("Please describe your word!")
-    # st.session_state.is_helpful['ask'] = False
+    st.session_state.is_helpful['ask'] = False
 
 ## removed: if st.session_state.actions[-1] == "result":
 
@@ -338,17 +339,23 @@ if st.session_state.actions[-1] == 'cue':
         write_bot(f'This is my predicted word: "{target}". Does this match your query?')
         ask_if_helped()
 
-    
     elif b2:
       rels = return_top_k(st.session_state.descriptions[-1], word=target, rels=True)
-      write_bot(f'Here are words that are related to your word: {", ".join(rels)}')
+      write_bot(f'Here are words that are related to your word: {", ".join(rels)}. \n Does this help you remember the word?', remember=False)
+      ask_if_helped()
 
     elif b3:
       st.session_state.counters["letter_count"] = 1
       letter_count = st.session_state.counters["letter_count"]
       st.session_state.counters["word_count"] += 1
       word_count = st.session_state.counters["word_count"]
-      write_bot(f'The next word starts with {st.session_state.results["results"][word_count][:letter_count]}', remember=False)
+      #write_bot(f'The next word starts with {st.session_state.results["results"][word_count][:letter_count]}', remember=False)
+      if letter_count < len(target):
+        write_bot(f'The next word starts with {st.session_state.results["results"][word_count][:letter_count]}. \n Does this help you remember the word?', remember=False)
+        ask_if_helped()
+      else:
+        write_bot(f'This is my predicted word: "{target}". Does this match your query?')
+        ask_if_helped()
 
     #elif get_available_cues(target) and "Synonyms" in cues_buttons and cues_buttons['Synonyms']:
       #write_bot(f'Here are synonyms for the current word: {", ".join(avail_cues["Synonyms"])}', remember=False)