Added the bot's response in case the user doesn't say yes or no

pages/1_Descriptive_chatbot.py

@@ -257,15 +257,17 @@ for message in st.session_state.messages:
 #display user message in chat message container
 prompt = get_text() 
 if prompt:
-  #JS: would replace Simon by some neutral character 
   with st.chat_message('user',  avatar='nursulu.jpg'):
     st.markdown(prompt)
   #add to history
   st.session_state.messages.append({'role': 'user', 'content': prompt})
   #TODO: replace it with zero-shot classifier
-  yes = ['yes', 'again', 'Yes', 'sure', 'new word', 'yes!', 'yep', 'yeah']
-  if prompt in yes:
+  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)
@@ -278,6 +280,8 @@ if prompt:
     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:
+    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":
   col1, col2, col3, col4, col5 = st.columns(5)