changes in conversational flow

app.py

@@ -128,7 +128,7 @@ def ask_if_helped():
     write_bot("Please describe your word!")
     # st.session_state.is_helpful['ask'] = False
 
-if st.session_state.actions[-1] == "result":
+'''if st.session_state.actions[-1] == "result":
   a1 = st.button('Results', key=10)
   a2 = st.button('Cue', key=11)
   if a1:
@@ -144,7 +144,7 @@ if st.session_state.actions[-1] == "result":
     st.session_state.is_helpful['ask'] = True
     
 if st.session_state.is_helpful['ask'] == True:
-    ask_if_helped()
+    ask_if_helped()'''
 
 # JS
 def postproc_wn(related_words, syns=False):
@@ -165,7 +165,7 @@ def get_available_cues(target):
       available_cues = {}
       synset_target = wn.synsets(target, pos=wn.NOUN)[0]
 
-      if wn.synonyms(target):
+      if wn.synonyms(target)[0]:
           available_cues['Synonyms'] = postproc_wn(wn.synonyms(target)[0], syns=True)
           #available_cues['Synonyms'] = [word.split('.')[0] if word[0] != "." else word.split('.')[1] for word in wn.synonyms(target)[0]]
       
@@ -189,7 +189,117 @@ def get_available_cues(target):
     else:
       return None 
 
-if st.session_state.actions[-1] == 'cue':
+def cue_generation():
+  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
+  
+#display user message in chat message container
+prompt = get_text() 
+if prompt:
+  with st.chat_message('user', avatar='simon.jpg'):
+    st.markdown(prompt)
+  #add to history
+  st.session_state.messages.append({'role': 'user', 'content': prompt})
+  yes = ['yes', 'again', 'Yes', 'sure', 'new word', 'yes!', 'yep', 'yeah']
+  if prompt in yes:
+    write_bot("Please describe your word!")
+  elif prompt == 'It is similar to the best place on earth':
+    write_bot("Great! Let me think what it could be...")
+    time.sleep(3)
+    write_bot("Do you mean Saarland?")
+  #if previously we asked to give a prompt
+  elif (st.session_state.messages[-2]['content'] == "Please describe your word!") & (st.session_state.messages[-1]['content'] != "no"):
+    write_bot("Great! Let me think what it could be...")
+    st.session_state.descriptions.append(prompt)
+    st.session_state.results['results'] = return_top_k(st.session_state.descriptions[-1])
+    st.session_state.results['results_print'] = dict(zip(range(1, 11), st.session_state.results['results']))
+    write_bot("I think I have some ideas. Do you want to see my guesses or do you want a cue?")
+    st.session_state.actions.append("result")
+
+if st.session_state.actions[-1] == "result":
+  a1 = st.button('Results', key=10)
+  a2 = st.button('Cue', key=11)
+  if a1:
+    write_bot("Here are my guesses about your word:")
+    st.write(st.session_state.results['results_print'])
+    time.sleep(1)
+    write_bot('Does it help you remember the word?', remember=False)
+    st.session_state.is_helpful['ask'] = True
+  elif a2:
+    #write_bot(f'The first letter is {st.session_state.results["results"][0][0]}.')
+    #time.sleep(1)
+    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)
 
@@ -254,32 +364,7 @@ if st.session_state.actions[-1] == 'cue':
     new = st.button('Play again', key=64)
     if new:
       write_bot("Please describe your word!")
-    break
-  
-#display user message in chat message container
-prompt = get_text() 
-if prompt:
-  with st.chat_message('user', avatar='simon.jpg'):
-    st.markdown(prompt)
-  #add to history
-  st.session_state.messages.append({'role': 'user', 'content': prompt})
-  yes = ['yes', 'again', 'Yes', 'sure', 'new word', 'yes!', 'yep', 'yeah']
-  if prompt in yes:
-    write_bot("Please describe your word!")
-  elif prompt == 'It is similar to the best place on earth':
-    write_bot("Great! Let me think what it could be...")
-    time.sleep(3)
-    write_bot("Do you mean Saarland?")
-  #if previously we asked to give a prompt
-  elif (st.session_state.messages[-2]['content'] == "Please describe your word!") & (st.session_state.messages[-1]['content'] != "no"):
-    write_bot("Great! Let me think what it could be...")
-    st.session_state.descriptions.append(prompt)
-    st.session_state.results['results'] = return_top_k(st.session_state.descriptions[-1])
-    st.session_state.results['results_print'] = dict(zip(range(1, 11), st.session_state.results['results']))
-    write_bot("I think I have some ideas. Do you want to see my guesses or do you want a cue?")
-    st.session_state.actions.append("result")
-
-    
+    break''' 
 
   # elif prompt == 'results':
   #   st.text("results")