Update app.py

app.py

@@ -4,7 +4,7 @@ import pandas as pd
 from functools import partial
 from ai_classroom_suite.UIBaseComponents import *
 
-# TODO: 
+# Testing purpose
 class EchoingTutor(SlightlyDelusionalTutor):
     def add_user_message(self, user_message):
         self.conversation_memory.append([user_message, None])
@@ -23,13 +23,16 @@ class EchoingTutor(SlightlyDelusionalTutor):
 
 ### Chatbot Functions ###
 def add_user_message(user_message, chat_tutor):
-  """Display user message and update chat history to include it."""
   chat_tutor.add_user_message(user_message)
   return chat_tutor.conversation_memory, chat_tutor
-
+"""
 def get_tutor_reply(user_message, chat_tutor):
   chat_tutor.get_tutor_reply(user_message)
   return gr.update(value="", interactive=True), chat_tutor.conversation_memory, chat_tutor
+"""
+def get_tutor_reply(chat_tutor):
+  chat_tutor.get_tutor_reply(input_kwargs={'question':''})
+  return gr.update(value="", interactive=True), chat_tutor.conversation_memory, chat_tutor
 
 def get_conversation_history(chat_tutor):
     return chat_tutor.conversation_memory, chat_tutor
@@ -78,7 +81,7 @@ def embed_prompt(instructor_prompt):
 with gr.Blocks() as demo:
     
     #initialize tutor (with state)
-    study_tutor = gr.State(EchoingTutor())
+    study_tutor = gr.State(SlightlyDelusionalTutor())
 
     # Student interface
     with gr.Tab("For Students"): 
@@ -95,11 +98,18 @@ with gr.Blocks() as demo:
               with gr.Row():
                 user_chat_input = gr.Textbox(label="User input", scale=9)
                 user_chat_submit = gr.Button("Ask/answer model", scale=1)
-        
-        user_chat_submit.click(add_user_message, 
-                               [user_chat_input, study_tutor], 
-                               [chatbot, study_tutor], queue=False).then(
-            get_tutor_reply, [user_chat_input, study_tutor], [user_chat_input, chatbot, study_tutor], queue=True)
+
+        user_chat_submit.click(
+            add_user_message,
+            [user_chat_input, study_tutor],
+            [chatbot, study_tutor], 
+            queue=False
+        ).then(
+            get_tutor_reply, 
+            [study_tutor], 
+            [user_chat_input, chatbot, study_tutor], 
+            queue=True)
+
 
         # Testing purpose
         test_btn = gr.Button("View your chat history")