Update app.py

app.py

@@ -108,7 +108,7 @@ def get_top_chunks(query, chunk_embeddings, text_chunks):
 
 client = InferenceClient("Qwen/Qwen2.5-72B-Instruct")
 
-def respond_cool(message, history):
+def respond_cool(message, history, mom_type):
     top_cool_results = get_top_chunks(message, cool_chunk_embeddings, cleaned_cool_chunks) # Complete this line
     #str_chunks = "\n".join(best_chunks)
 
@@ -130,11 +130,11 @@ def respond_cool(message, history):
     )
     return response['choices'][0]['message']['content'].strip()
 
-def respond_tutor(message, history):
+def respond_tutor(message, history, mom_type):
     top_tutor_results = get_top_chunks(message, tutor_chunk_embeddings, cleaned_tutor_chunks)
     #str_chunks = "\n".join(best_chunks)
 
-    messages = [{"role": "system", "content": f"You are chatbot that plays the role of the user's extremely studious, tutor-like mom. Respond in full sentences, don't cut yourself off. Base your response on the provided context: {top_tutor_results}"}, 
+    messages = [{"role": "system", "content": f"You are chatbot that plays the role of the user's extremely studious, tutor-like mom. Respond in full sentences, don't cut yourself off. Base your response on the provided context: {mom_type}"}, 
                {"role": "user", 
                "content": (
                    f"Context:\n{top_tutor_results}\n\n"
@@ -174,37 +174,14 @@ def respond_strict(message, history):
     )
     return response['choices'][0]['message']['content'].strip()
 
-
-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")
-
-    def set_mom_cool(): return "Cool Mom"
-    def set_mom_tutor(): return "Tutor Mom"
-    def set_mom_strict(): return "Strict Mom"
+    with gr.Blocks() as demo:
         
-    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, chat_history, mom_type):
-        if mom_type == "Cool Mom":
-            return respond_cool(message, chat_history)
-        elif mom_type == "Tutor Mom":
-            return respond_tutor(message, chat_history)
-        else:
-            return respond_strict(message, chat_history)
+        with gr.Row():
+            mom_type = gr.CheckboxGroup(['Cool Mom', 'Tutor Mom', 'Strict Mom'],label='Choose Your Mom')
             
     gr.ChatInterface(
         fn=route_message,
-        additional_inputs=[chat_history,mom_type],
+        additional_inputs=[mom_type],
         title="StudyMama"
     )