updated

app.py

@@ -45,11 +45,33 @@ def get_top_chunk(query):
 
     return best_chunk
 
-print(get_top_chunk("How can I manage test anxiety?"))
+mental_health_keywords = [ # we implement a list of keywords to ensure chatbot only assists with mental health
+    "stress", "stressed", "anxiety", "anxious",
+    "sad", "sadness", "depressed", "depression",
+    "lonely", "alone", "angry", "anger",
+    "overwhelmed", "burnout", "burned out",
+    "school", "grades", "homework", "test", "exam",
+    "family", "parents", "siblings",
+    "friend", "friends", "friendship",
+    "relationship", "breakup",
+    "sleep", "tired", "exhausted",
+    "panic", "worried", "worry",
+    "mental", "emotion", "emotional",
+    "feel", "feeling", "feelings",
+    "coping", "calm",
+    "therapy", "therapist", "counselor",
+    "motivation", "confidence", "self-esteem",
+    "crying", "numb", "overthinking",
+    "focus", "concentrate", "concentration"
+]
 
 def respond(message, history):
 
+    if not any(word in message.lower() for word in mental_health_keywords):
+        return "I'm mainly here to support teen mental wellness. Try asking me about stress, anxiety, school pressure, friendships, emotions, sleep, or coping strategies."
+
     relevant_chunk = get_top_chunk(message)
+    print("Retrieved chunk:", relevant_chunk) # we are using this line to test RAG system (the terminal will showwhat knowledge chunk chatbot used)
     
     messages = [
     {