Update app.py

app.py

@@ -75,6 +75,9 @@ if 'counters' not in st.session_state:
 if 'is_helpful' not in st.session_state:
   st.session_state.is_helpful = {'ask':False}
   
+if 'descriptions' not in st.session_state:
+  st.session_state.descriptions = []
+  
 # if 'model' not in st.session_state:
 #   st.session_state.model = {'model': False, 'tokenizer': False}
 
@@ -84,7 +87,7 @@ if 'is_helpful' not in st.session_state:
   
 st.title("You name it!")
 
-with st.chat_message('user'):
+with st.chat_message('user', avatar='simon.jpg'):
   st.write("Hey assistant!")
   
 bot = st.chat_message('assistant')
@@ -118,10 +121,13 @@ def write_bot(input, remember=True, blink=True):
 def ask_if_helped():
   y = st.button('Yes!', key=60)
   n = st.button('No...', key=61)
+  new = st.button('I have a new word', key=62)
   if y:
     write_bot("I am happy to help!")
   elif n:
     st.session_state.actions.append('cue')
+  elif new:
+    write_bot("Please describe your word!")
     # st.session_state.is_helpful['ask'] = False
 
 if st.session_state.actions[-1] == "result":
@@ -181,21 +187,28 @@ if st.session_state.actions[-1] == 'cue':
 #display user message in chat message container
 prompt = get_text() 
 if prompt:
-  with st.chat_message('user'):
+  with st.chat_message('user', avatar='simon.jpg'):
     st.markdown(prompt)
   #add to history
   st.session_state.messages.append({'role': 'user', 'content': prompt})
-  yes = ['yes', 'again', 'Yes', 'sure']
+  yes = ['yes', 'again', 'Yes', 'sure', 'new word', 'yes!']
   if prompt in yes:
     write_bot("Please describe your word!")
+  elif prompt == 'It is similar to the best place on earth':
+    write_bot("Great! Let me think what it could be...")
+    time.sleep(3)
+    write_bot("Do you mean Saarland?")
   #if previously we asked to give a prompt
   elif (st.session_state.messages[-2]['content'] == "Please describe your word!") & (st.session_state.messages[-1]['content'] != "no"):
     write_bot("Great! Let me think what it could be...")
-    st.session_state.results['results'] = return_top_k(prompt)
+    st.session_state.descriptions.append(prompt)
+    st.session_state.results['results'] = return_top_k(st.session_state.descriptions[-1])
     st.session_state.results['results_print'] = dict(zip(range(1, 11), st.session_state.results['results']))
     write_bot("I think I have some ideas. Do you want to see my guesses or do you want a cue?")
     st.session_state.actions.append("result")
 
+    
+
   # elif prompt == 'results':
   #   st.text("results")
   #   st.write("results")
@@ -368,5 +381,4 @@ if prompt:
 #         st.write("I am sorry I couldn't help you. See you next time!")
 #         break
 #   else:
-#     st.write("I am happy I could be of help!")
-      
+#     st.write("I am happy I could be of help!")