button bug fix

app.py

@@ -124,7 +124,7 @@ def ask_if_helped():
     write_bot("I am happy to help!")
   elif n:
     st.session_state.actions.append('cue')
-    cue_generation()
+    #cue_generation()
   elif new:
     write_bot("Please describe your word!")
     # st.session_state.is_helpful['ask'] = False
@@ -278,13 +278,80 @@ if st.session_state.actions[-1] == "result":
     #write_bot(f'The first letter is {st.session_state.results["results"][0][0]}.')
     #time.sleep(1)
     st.session_state.actions.append('cue')
-    cue_generation()
+    #cue_generation()
     #write_bot('Does it help you remember the word?', remember=False)
     #st.session_state.is_helpful['ask'] = True
     
 if st.session_state.is_helpful['ask']:
     ask_if_helped()
 
+if st.session_state.actions[-1] == 'cue':
+  guessed = False
+  write_bot('What do you want to see?', remember=False, blink=False)
+
+  while guessed == False:
+    b1 = st.button("Next letter", key="1")
+    b2 = st.button("Next word", key="2")
+
+    # JS
+    word_count = st.session_state.counters["word_count"]
+    target = st.session_state.results["results"][word_count]
+    if get_available_cues(target):
+      avail_cues = get_available_cues(target)
+      cues_buttons = {}
+      for cue_type, cues in avail_cues.items():
+        #st.button(cue_type)
+        cues_buttons[cue_type] = st.button(cue_type)
+
+    b3 = st.button("All words", key="3")
+    b4 = st.button("I remembered the word!", key="4", type='primary')
+    b5 = st.button("Exit", key="5", type='primary')
+
+    if b1:
+      st.session_state.counters["letter_count"] += 1
+      word_count = st.session_state.counters["word_count"]
+      letter_count = st.session_state.counters["letter_count"]
+      write_bot(f'The word starts with {st.session_state.results["results"][word_count][:letter_count]}', remember=False)
+
+    elif b2:
+      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)
+
+    elif "Synonyms" in cues_buttons and cues_buttons['Synonyms']:
+      write_bot(f'Here are synonyms for the current word: {", ".join(avail_cues["Synonyms"])}', remember=False)
+
+    elif "Hypernyms" in cues_buttons and cues_buttons['Hypernyms']:
+      write_bot(f'Here are hypernyms for the current word: {", ".join(avail_cues["Hypernyms"])}', remember=False)
+
+    elif "Hyponyms" in cues_buttons and cues_buttons['Hyponyms']:
+      write_bot(f'Here are hyponyms for the current word: {", ".join(avail_cues["Hyponyms"])}', remember=False)
+
+    elif "Examples" in cues_buttons and cues_buttons['Examples']:   
+      write_bot(f'Here are example contexts for the current word: {", ".join(avail_cues["Examples"])}', remember=False)
+
+    elif b3:
+      write_bot(f"Here are all my guesses about your word: {st.session_state.results['results_print']}")
+
+    elif b4:
+      write_bot("Yay! I am happy I could be of help!")
+      new = st.button('Play again', key=63)
+      if new:
+        write_bot("Please describe your word!")
+      guessed = True
+    
+      break
+    
+    elif b5:
+      write_bot("I am sorry I couldn't help you this time. See you soon!")
+    st.session_state.actions.append('cue')
+    new = st.button('Play again', key=64)
+    if new:
+      write_bot("Please describe your word!")
+    break
+
   # elif prompt == 'results':
   #   st.text("results")
   #   st.write("results")