Update app.py

app.py

@@ -19,8 +19,7 @@ def get_models():
   
   st.session_state.model['model'] = model
   st.session_state.model['tokenizer'] = tokenizer
-
-  st.write("The assistant is loaded and ready to use!")
+  
 
 def return_top_k(sentence, k=10):
   
@@ -79,12 +78,12 @@ if 'model' not in st.session_state:
 
 if st.session_state.model['model'] == False:
   get_models()
+  st.write("The assistant is loaded and ready to use!")
   
 st.title("You name it!")
 
-if st.session_state.model['model'] != False:
-    with st.chat_message('user'):
-        st.write("Hey assistant! All good?")
+with st.chat_message('user'):
+  st.write("Hey assistant!")
   
 bot = st.chat_message('assistant')
 bot.write("Hello human! Wanna practice naming some words?")
@@ -98,17 +97,18 @@ def get_text():
   input_text = st.chat_input()
   return input_text
 
-def write_bot(input, remember=True):
+def write_bot(input, remember=True, blink=True):
     with st.chat_message('assistant'):
       message_placeholder = st.empty()
       full_response = input
-      response = ''
-      for chunk in full_response.split():
-        response += chunk + " "
-        time.sleep(0.05)
-        # Add a blinking cursor to simulate typing
-        message_placeholder.markdown(response + "▌")
-      time.sleep(0.5)
+      if blink == True:
+        response = ''
+        for chunk in full_response.split():
+          response += chunk + " "
+          time.sleep(0.05)
+          # Add a blinking cursor to simulate typing
+          message_placeholder.markdown(response + "▌")
+        time.sleep(0.5)
       message_placeholder.markdown(full_response)
     if remember == True:
       st.session_state.messages.append({'role': 'assistant', 'content': full_response})
@@ -121,6 +121,7 @@ def ask_if_helped():
       write_bot("I am happy to help!")
     elif n:
       st.session_state.actions.append('cue')
+  st.session_state.is_helpful['ask'] = False
 
 if st.session_state.actions[-1] == "result":
   a1 = st.button('Results', key=10)
@@ -128,12 +129,12 @@ if st.session_state.actions[-1] == "result":
   if a1:
     write_bot("Here are my guesses about your word:")
     st.write(st.session_state.results['results_print'])
-    time.sleep(2)
+    time.sleep(1)
     write_bot('Does it help you remember the word?', remember=False)
     st.session_state.is_helpful['ask'] = True
   elif a2:
     write_bot(f'The first letter is {st.session_state.results["results"][0][0]}.')
-    time.sleep(2)
+    time.sleep(1)
     write_bot('Does it help you remember the word?', remember=False)
     st.session_state.is_helpful['ask'] = True
     
@@ -143,7 +144,7 @@ ask_if_helped()
 
 if st.session_state.actions[-1] == 'cue':
   guessed = False
-  write_bot('What do you want to see?', remember=False)
+  write_bot('What do you want to see?', remember=False, blink=False)
   b1 = st.button("Next letter", key="1")
   b2 = st.button("Next word", key="2")
   b3 = st.button("All words", key="3")
@@ -154,14 +155,14 @@ if st.session_state.actions[-1] == 'cue':
       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]}')
+      write_bot(f'The word starts with {st.session_state.results["results"][word_count][:letter_count]}', remember=False)
 
     elif b2:
       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]}')      
+      write_bot(f'The next word starts with {st.session_state.results["results"][word_count][:letter_count]}', remember=False)      
       
     elif b3:
       write_bot(f"Here are all my guesses about your word: {st.session_state.results['results_print']}")
@@ -183,8 +184,8 @@ if prompt:
     st.markdown(prompt)
   #add to history
   st.session_state.messages.append({'role': 'user', 'content': prompt})
-  yes = "YesyesYes!Sure!sure"
-  if prompt == "yes":
+  yes = ['yes', 'again', 'Yes', 'sure']
+  if prompt in yes:
     write_bot("Please describe your word!")
   #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"):
@@ -367,10 +368,4 @@ if prompt:
 #         break
 #   else:
 #     st.write("I am happy I could be of help!")
-      
-                 
-      
-        
-        
-      
-  
+