Bug fix WN

app.py

@@ -147,8 +147,11 @@ 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]
+def postproc_wn(related_words, syns=False):
+    if syns:
+        related_words = [word.split('.')[0] if word[0] != "." else word.split('.')[1] for word in related_words]
+    else:
+        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
@@ -163,7 +166,8 @@ def get_available_cues(target):
       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]]
+          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]]
       
       if synset_target.hypernyms():
           available_cues['Hypernyms'] = postproc_wn(synset_target.hypernyms())