Include wordnet relations test

app.py

@@ -146,10 +146,55 @@ if st.session_state.actions[-1] == "result":
 if st.session_state.is_helpful['ask'] == True:
     ask_if_helped()
 
+# JS
+def postproc_wn(related_words):
+    related_words = [word.name().split('.')[0] if word.name()[0] != "." else word.name().split('.')[1] for word in related_words]
+    related_words = [word.replace("_", " ") for word in related_words]
+
+    return related_words
+
+# JS
+def get_available_cues(target):
+    wn_nouns = [word.name() for word in wn.all_synsets(pos='n')]
+    wn_nouns = [word.split('.')[0] if word[0] != "." else word.split('.')[1] for word in wn_nouns]
+
+    if target in wn_nouns:
+      available_cues = {}
+      synset_target = wn.synsets(target, pos=wn.NOUN)[0]
+
+      if wn.synonyms(target):
+          available_cues['Synonyms'] = [word.split('.')[0] if word[0] != "." else word.split('.')[1] for word in wn.synonyms(target)[0]]
+      
+      if synset_target.hypernyms():
+          available_cues['Hypernyms'] = postproc_wn(synset_target.hypernyms())
+
+
+      if synset_target.hyponyms():
+          available_cues['Hyponyms'] = postproc_wn(synset_target.hyponyms())
+
+      if synset_target.examples():
+          examples = []
+          
+          for example in synset_target.examples():
+              examples.append(example.replace(target, "..."))
+                  
+          available_cues['Examples'] = examples
+
+      return available_cues
+
+    else:
+      return None 
 
 if st.session_state.actions[-1] == 'cue':
   guessed = False
   write_bot('What do you want to see?', remember=False, blink=False)
+  # JS
+  if get_available_cues():
+    avail_cues = get_available_cues()
+    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")