Update app.py

app.py

@@ -108,17 +108,48 @@ def get_top_chunks(query, chunk_embeddings, text_chunks):
 
 client = InferenceClient("Qwen/Qwen2.5-72B-Instruct")
 
+with gr.Blocks() as chatbot:
+    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=[]: handle_click("Button 1"), inputs=[], outputs=respond(message, history, "Cool Mom"))
+    tutor_button.click(fn=[]: handle_click("Button 2"), inputs=[], outputs=respond(message, history, "Tutor Mom"))
+    strict_button.click(fn=[]: handle_click("Button 3"), inputs=[], outputs=respond(message, history, "Strict Mom"))
+    
+    gr.ChatInterface(
+        #fn=respond,
+        #additional_inputs=[mom_type],
+        title="StudyMama"
+    )
+
 def respond(message, history, mom_type):
     top_cool_results = get_top_chunks(message, cool_chunk_embeddings, cleaned_cool_chunks) # Complete this line
+    top_tutor_results = get_top_chunks(message, tutor_chunk_embeddings, cleaned_tutor_chunks)
+    top_strict_results = get_top_chunks(message, strict_chunk_embeddings, cleaned_strict_chunks)
     #str_chunks = "\n".join(best_chunks)
-
-    messages = [{"role": "system", "content": f"You are chatbot that plays the role of the user's cool and super chill mom. Respond in full sentences, don't cut yourself off. Use responses from this text file: {top_cool_results} and respond very kindly. Do not be mean or strict at all"}, 
-               {"role": "user", 
-               "content": (
-                   f"Context:\n{top_cool_results}\n\n"
-                   f"Question{message}"
-               )}]
-
+    if mom_type == "Cool Mom":
+        messages = [{"role": "system", "content": f"You are chatbot that plays the role of the user's cool and super chill mom. Respond in full sentences, don't cut yourself off. Use responses from this text file: {top_cool_results} and respond very kindly. Do not be mean or strict at all"}, 
+                   {"role": "user", 
+                   "content": (
+                       f"Context:\n{top_cool_results}\n\n"
+                       f"Question{message}"
+                   )}]
+    elif mom_type == "Tutor Mom":
+        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"
+                    f"Question{message}"
+                    )}]
+    elif mom_type == "Strict Mom":
+        messages = [{"role": "system", "content": f"You are chatbot that plays the role of the user's extremely strict mom. Respond in full sentences, don't cut yourself off. Base your response on the provided context: {top_strict_results}"}, 
+                    {"role": "user", 
+                    "content": (
+                    f"Context:\n{top_strict_results}\n\n"
+                    f"Question{message}"
+                    )}]
     if history:
         messages.extend(history)
 
@@ -132,59 +163,50 @@ def respond(message, history, mom_type):
 
 chatbot = gr.ChatInterface(respond, type="messages")
 
-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)
+#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: {mom_type}"}, 
-                {"role": "user", 
-                "content": (
-                    f"Context:\n{top_tutor_results}\n\n"
-                    f"Question{message}"
-                )}]
+#    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"
+#                    f"Question{message}"
+#                )}]
 
-    if history:
-        messages.extend(history)
+#    if history:
+#        messages.extend(history)
 
-    messages.append({"role": "user", "content": message})
+#    messages.append({"role": "user", "content": message})
 
-    response = client.chat_completion(
-        messages, 
-        temperature = 0.2
+#    response = client.chat_completion(
+#        messages, 
+#        temperature = 0.2
      )
-    return response['choices'][0]['message']['content'].strip()
+#    return response['choices'][0]['message']['content'].strip()
 
-def respond_strict(message, history):
-    top_strict_results = get_top_chunks(message, strict_chunk_embeddings, cleaned_strict_chunks)
+#def respond_strict(message, history):
+#    top_strict_results = get_top_chunks(message, strict_chunk_embeddings, cleaned_strict_chunks)
     #str_chunks = "\n".join(best_chunks)
 
-    messages = [{"role": "system", "content": f"You are chatbot that plays the role of the user's extremely strict mom. Respond in full sentences, don't cut yourself off. Base your response on the provided context: {top_strict_results}"}, 
-            {"role": "user", 
-                "content": (
-                    f"Context:\n{top_strict_results}\n\n"
-                    f"Question{message}"
-                )}]
+#    messages = [{"role": "system", "content": f"You are chatbot that plays the role of the user's extremely strict mom. Respond in full sentences, don't cut yourself off. Base your response on the provided context: {top_strict_results}"}, 
+#            {"role": "user", 
+#                "content": (
+#                    f"Context:\n{top_strict_results}\n\n"
+#                    f"Question{message}"
+#                )}]
 
-    if history:
-        messages.extend(history)
+#    if history:
+ #       messages.extend(history)
+#
+#    messages.append({"role": "user", "content": message})
 
-    messages.append({"role": "user", "content": message})
+#    response = client.chat_completion(
+#        messages, 
+#        temperature = 0.2
+#     )
+#    return response['choices'][0]['message']['content'].strip()
 
-    response = client.chat_completion(
-        messages, 
-        temperature = 0.2
-     )
-    return response['choices'][0]['message']['content'].strip()
 
-with gr.Blocks() as chatbot:
-        
-    with gr.Row():
-        mom_type = gr.CheckboxGroup(['Cool Mom', 'Tutor Mom', 'Strict Mom'],label='Choose Your Mom')
-            
-    gr.ChatInterface(
-        fn=respond,
-        additional_inputs=[mom_type],
-        title="StudyMama"
-    )
 
 chatbot.launch()