Update app.py

app.py

@@ -22,10 +22,23 @@ def respond(user_input):
     else:
         st.session_state['chat_history_ids'] = torch.cat([st.session_state['chat_history_ids'], new_user_input_ids], dim=-1)
 
-    # Check for specific requests and provide specific answers
-    if "facts about blue" in user_input.lower():
-        response = "Here are some happy facts about blue:\n1. Blue is often associated with calmness and tranquility.\n2. Studies show that blue can help reduce stress and promote relaxation.\n3. The sky and ocean are blue, reminding us of nature's beauty.\n4. Blue is the color of many happy brands, symbolizing trust and dependability."
+    # Check for specific emotional support requests
+    if "happy" in user_input.lower():
+        response = "Wow, that's great to hear! What made you feel happy today?"
+    elif "sad" in user_input.lower():
+        response = "I'm really sorry to hear that. Would you like to share what’s making you feel sad? I'm here to listen."
+    elif "angry" in user_input.lower():
+        response = "It's okay to feel angry sometimes. Would you like to talk about what's bothering you?"
+    elif "facts about blue" in user_input.lower():
+        response = (
+            "Here are some happy facts about blue:\n"
+            "1. Blue is often associated with calmness and tranquility.\n"
+            "2. Studies show that blue can help reduce stress and promote relaxation.\n"
+            "3. The sky and ocean are blue, reminding us of nature's beauty.\n"
+            "4. Blue is the color of many happy brands, symbolizing trust and dependability."
+        )
     else:
+        # Generate a response from the model for general queries
         chat_history_ids = model.generate(
             st.session_state['chat_history_ids'],
             max_length=1000,
@@ -45,8 +58,8 @@ def respond(user_input):
     return response
 
 # Streamlit app layout
-st.title("Emotional Health Chatbot")
-st.write("Hello! I am here to support you emotionally. How are you feeling today?")
+st.title("Emotional Health & General Knowledge Chatbot")
+st.write("Hello! I'm here to support you emotionally and answer any questions. How are you feeling today?")
 
 # Display chat history
 if st.session_state['chat_history']: