Update app.py

app.py

@@ -194,17 +194,17 @@ with gr.Blocks() as demo:
     tutor_button.click(fn=set_mom_tutor, inputs=[], outputs=mom_type)
     strict_button.click(fn=set_mom_strict, inputs=[], outputs=mom_type)
 
-    def route_message(message, history, mom_type):
+    def route_message(message, chat_history, mom_type):
         if mom_type == "Cool Mom":
-            return respond_cool(message, history)
+            return respond_cool(message, chat_history)
         elif mom_type == "Tutor Mom":
-            return respond_tutor(message, history)
+            return respond_tutor(message, chat_history)
         else:
-            return respond_strict(message, history)
+            return respond_strict(message, chat_history)
             
     gr.ChatInterface(
         fn=route_message,
-        additional_inputs=[history, mom_type],
+        additional_inputs=[mom_type],
         title="StudyMama"
     )