related word button changes

app.py

@@ -47,12 +47,15 @@ def remove_punctuation(word):
     
     return word_without_punctuation
 
-def return_top_k(sentence, k=10):
+def return_top_k(sentence, k=10, word=None, rels=False):
 
-  if sentence[-1] != ".":
-    sentence = sentence + "."
+  #if sentence[-1] != ".":
+   # sentence = sentence + "."
 
-  inputs = [f"Descripton : {sentence}. Word : "]
+  if rels:
+    inputs = [f"Description : It is related to '{word}' but not '{word}'. Word : "]
+  else:
+    inputs = [f"Description : {sentence}. Word : "]
 
   inputs = tokenizer(
       inputs,
@@ -245,7 +248,8 @@ for message in st.session_state.messages:
 #display user message in chat message container
 prompt = get_text() 
 if prompt:
-  with st.chat_message('user', avatar='simon.jpg'):
+  #JS: would replace Simon by some neutral character 
+  with st.chat_message('user'):
     st.markdown(prompt)
   #add to history
   st.session_state.messages.append({'role': 'user', 'content': prompt})
@@ -348,7 +352,7 @@ if st.session_state.actions[-1] == 'cue':
 
     elif b6:
       sent = f"It is related to '{target}' but not '{target}'."
-      rels = return_top_k(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)}')