Update app.py

app.py

@@ -108,32 +108,15 @@ def get_top_chunks(query, chunk_embeddings, text_chunks):
 
 client = InferenceClient("Qwen/Qwen2.5-72B-Instruct")
 
-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)
-    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}"
-                    )}]
+def respond(messages, history, mom_type) :
+    mom_type = []
+    if mom_type == "Cool Mom" :
+        top_results = get_top_chunks(message, cool_chunk_embeddings, cleaned_cool_chunks)
+    elif mom_type == "Tutor Mom" :
+        top_results = get_top_chunks(message, tutor_chunk_embeddings, cleaned_tutor_chunks)
     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}"
-                    )}]
+        top_results = get_top_chunks(message, strict_chunk_embeddings, cleaned_strict_chunks)
+        
     if history:
         messages.extend(history)