Update app.py

app.py

@@ -5,10 +5,6 @@ from functools import partial
 from ai_classroom_suite.UIBaseComponents import *
 
 ### Chatbot Functions ###
-def add_user_message(user_message, chat_tutor):
-  chat_tutor.add_user_message(user_message)
-  return chat_tutor.conversation_memory, chat_tutor
-
 def get_tutor_reply(chat_tutor):
   chat_tutor.get_tutor_reply()
   return gr.update(value="", interactive=True), chat_tutor.conversation_memory, chat_tutor
@@ -52,7 +48,7 @@ with gr.Blocks() as demo:
         # First add user's message to the conversation history
         # Then get reply from the tutor and add that to the conversation history
         user_chat_submit.click(
-            fn = add_user_message, inputs = [user_chat_input, study_tutor], outputs = [chatbot, study_tutor], queue=False
+            fn = add_user_message, inputs = [user_chat_input, study_tutor], outputs = [user_chat_input, chatbot, study_tutor], queue=False
         ).then(
             fn = get_tutor_reply, inputs = [study_tutor], outputs = [user_chat_input, chatbot, study_tutor], queue=True
         )