button arrangement

app.py

@@ -182,16 +182,15 @@ def get_available_cues(target):
       available_cues = {}
       synset_target = wn.synsets(target, pos=wn.NOUN)[0]
 
-      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]]
+      #if wn.synonyms(target)[0]:
+       #   available_cues['Synonyms'] = postproc_wn(wn.synonyms(target)[0], syns=True)
       
-      if synset_target.hypernyms():
-          available_cues['Hypernyms'] = postproc_wn(synset_target.hypernyms())
+      #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.hyponyms():
+        #  available_cues['Hyponyms'] = postproc_wn(synset_target.hyponyms())
 
       if synset_target.examples():
           examples = []
@@ -299,65 +298,65 @@ if st.session_state.actions[-1] == 'cue':
   write_bot('What do you want to see?', remember=False, blink=False)
 
   while guessed == False:
-    col1, col2, col3, col4, col5 = st.columns(5)
+    col1, col2, col3, col4 = st.columns(4)
+    col5, col6 = st.columns(2)
+
     with col1:
       b1 = st.button("Next letter", key="1")
     with col2:
-      b2 = st.button("Next word", key="2")
+      b2 = st.button("Related words")
+    with col3:
+      b3 = st.button("Next word", key="2")
+
+    with col4:
+      b4 = st.button("All words", key="3")
 
     # 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 = {cue_type: st.button(cue_type) for cue_type in avail_cues}
-      #for cue_type in avail_cues:
-       # cues_buttons[cue_type] = st.button(cue_type)
+    #if get_available_cues(target):
+     # avail_cues = get_available_cues(target)
+      #cues_buttons = {cue_type: st.button(cue_type) for cue_type in avail_cues}
 
-    with col3:
-      b3 = st.button("All words", key="3")
-    with col4:
-      b4 = st.button("I remembered the word!", key="4", type='primary')
     with col5:
-      b5 = st.button("Exit", key="5", type='primary')
+      b5 = st.button("I remembered the word!", key="4", type='primary')
+    with col6:
+      b6 = st.button("Exit", key="5", type='primary')
 
-    b6 = st.button("Related words")
 
     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:
+      rels = return_top_k(st.session_state.descriptions[-1], word=target, rels=True)
+      write_bot(f'Here are words that are related to your word: {", ".join(rels)}')
+
+    elif b3:
       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 get_available_cues(target) and "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 get_available_cues(target) and "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 get_available_cues(target) and "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 get_available_cues(target) and "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 get_available_cues(target) and "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 get_available_cues(target) and "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 get_available_cues(target) and "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 get_available_cues(target) and "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:
+    elif b4:
       write_bot(f"Here are all my guesses about your word: {st.session_state.results['results_print']}")
 
-    elif b6:
-      sent = f"It is related to '{target}' but not '{target}'."
-      rels = return_top_k(st.session_state.descriptions[-1], word=target, rels=True)
-
-      write_bot(f'Here are words that are related to your word: {", ".join(rels)}')
-
-    elif b4:
+    elif b5:
       write_bot("Yay! I am happy I could be of help!")
       new = st.button('Play again', key=63)
       if new:
@@ -366,7 +365,7 @@ if st.session_state.actions[-1] == 'cue':
     
       break
     
-    elif b5:
+    elif b6:
       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)