fixed the previous commit a bit

pages/1_Descriptive_chatbot.py

@@ -264,23 +264,24 @@ if prompt:
   #TODO: replace it with zero-shot classifier
   yes = ['yes', 'again', 'sure', 'new word', 'yes!', 'yep', 'yeah']
   no = ['no', 'nope', 'nah']
-  if prompt.lower() in yes:
-    write_bot("Please describe your word!")
-  elif prompt.lower() in no:
-    write_bot("Okay, see you next time then! :innocent:")
-  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.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")
-  else:
+  try:
+    if prompt.lower() in yes:
+      write_bot("Please describe your word!")
+    elif prompt.lower() in no:
+      write_bot("Okay, see you next time then! :innocent:")
+    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.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")
+  except:
     write_bot("Sorry, I didn't understand you... I am still learning :sob: For now, could you respond with 'yes' or 'no'? ")
   
 if st.session_state.actions[-1] == "result":