Include WN relations

app.py

@@ -188,21 +188,24 @@ def get_available_cues(target):
 if st.session_state.actions[-1] == 'cue':
   guessed = False
   write_bot('What do you want to see?', remember=False, blink=False)
-  # 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)
-
-  b1 = st.button("Next letter", key="1")
-  b2 = st.button("Next word", key="2")
-  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')
+
   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)
+        
+    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"]
@@ -214,8 +217,20 @@ if st.session_state.actions[-1] == 'cue':
       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)
+
+    elif st.button('Synonyms'):
+      write_bot(f'Here are synonyms for the current word: {avail_cues["Synonyms"]}', remember=False)
+
+    elif st.button('Hypernyms'):
+      write_bot(f'Here are hypernyms for the current word: {avail_cues["Hypernyms"]}', remember=False)
+
+    elif st.button('Hyponyms'):
+      write_bot(f'Here are hyponyms for the current word: {avail_cues["Hyponyms"]}', remember=False)
+
+    elif st.button('Examples'):   
+      write_bot(f'Here are example contexts for the current word: {avail_cues["Examples"]}', remember=False)
+
     elif b3:
       write_bot(f"Here are all my guesses about your word: {st.session_state.results['results_print']}")