Update app.py

app.py

@@ -176,17 +176,36 @@ def respond_strict(message, history):
 
 
 with gr.Blocks() as demo:
+    mom_type = gr.State("Cool Mom")   # default state
+    history = gr.State([])
+    
+    with gr.Row():
+        gr.Markdown("### Choose your Mom Mode")
     with gr.Row():
         cool_button = gr.Button("Cool Mom")
         tutor_button = gr.Button("Tutor Mom")
         strict_button = gr.Button("Strict Mom")
-    cool_button.click(fn=respond_cool, inputs=[], outputs=[])
-    tutor_button.click(fn=respond_tutor, inputs=[], outputs=[])
-    strict_button.click(fn=respond_strict, inputs=[], outputs=[])
-    if cool_button.click():
-        chatbot = gr.ChatInterface(respond_cool, type = 'messages')
-    elif tutor_button.click():
-        chatbot = gr.ChatInterface(respond_tutor, type = 'messages')
-    else:
-        chatbot = gr.ChatInterface(respond_strict, type = 'messages')
+
+    def set_mom_cool(): return "Cool Mom"
+    def set_mom_tutor(): return "Tutor Mom"
+    def set_mom_strict(): return "Strict Mom"
+        
+    cool_button.click(fn=set_mom_cool, inputs=[], outputs=mom_type)
+    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):
+        if mom_type == "Cool Mom":
+            return respond_cool(message, history)
+        elif mom_type == "Tutor Mom":
+            return respond_tutor(message, history)
+        else:
+            return respond_strict(message, history)
+            
+    gr.ChatInterface(
+        fn=route_message,
+        additional_inputs=[history, mom_type],
+        title="StudyMama"
+    )
+
 demo.launch()