Update app.py

app.py

@@ -48,19 +48,26 @@ def predict_emotion(text):
         st.error(f"⚠️ Prediction failed: {e}")
         return {"Error": f"Prediction failed: {e}"}
 
-# ---- Remedy Suggestions Based on Emotions ----
-def get_remedy_suggestion(emotion):
-    """Suggest activities based on detected emotion."""
-    remedies = {
-        "anger": "🧘‍♂️ Try doing a breathing exercise or meditate for a few minutes to calm down.",
-        "fear": "👣 Go for a walk to clear your mind and release tension. Try mindfulness during your walk.",
-        "joy": "🎉 You're feeling great! Keep up the positive energy and spread it to others around you.",
-        "sadness": "💡 Practice mindfulness or take a short walk outside to refresh your mind.",
-        "surprise": "🤔 Take a deep breath and reflect for a moment. Meditation can help bring focus.",
-        "disgust": "🌬️ Try a breathing exercise to help ease your thoughts and relax.",
-        "neutral": "💆‍♀️ Mindfulness practice could help in centering yourself and finding balance.",
-    }
-    return remedies.get(emotion, "💆‍♂️ Try practicing mindfulness to help center yourself.")
+# ---- Suggesting Activities Based on Emotional State ----
+def suggest_activity(emotion_analysis):
+    # Suggest activities based on emotional state
+    max_emotion = max(emotion_analysis, key=emotion_analysis.get) if emotion_analysis else None
+    if max_emotion == 'sadness':
+        return "It's okay to feel sad sometimes. Try taking a 5-minute mindfulness break or a short walk outside to clear your mind."
+    elif max_emotion == 'joy':
+        return "You’re feeling happy! Keep that positive energy going. How about practicing some deep breathing exercises to maintain your balance?"
+    elif max_emotion == 'fear':
+        return "Feeling anxious? It might help to do a quick breathing exercise or take a walk to ease your mind."
+    elif max_emotion == 'anger':
+        return "It seems like you're angry. Try taking a few deep breaths, or engage in a relaxing mindfulness exercise to calm your nerves."
+    elif max_emotion == 'surprise':
+        return "You’re surprised! Take a moment to breathe deeply and reflect. A walk might help clear your thoughts."
+    elif max_emotion == 'disgust':
+        return "If you’re feeling disgusted, a change of environment might help. Go for a walk or try a mindfulness technique to relax."
+    elif max_emotion == 'sadness':
+        return "It’s okay to feel sad. Try grounding yourself with some mindfulness or a breathing exercise."
+    else:
+        return "Keep doing great! If you feel overwhelmed, consider taking a deep breath or going for a short walk."
 
 # ---- User Input Section ----
 st.write("### 🌺 Let's Get Started!")
@@ -82,21 +89,16 @@ for i, question in enumerate(questions, start=1):
     if user_response:
         with st.spinner("Analyzing emotion... 🎭"):
             analysis = predict_emotion(user_response)
+        responses[question] = {"Response": user_response, "Analysis": analysis}
         
-        # Identify the dominant emotion from the analysis result
-        dominant_emotion = max(analysis, key=analysis.get) if analysis else "neutral"
-        
-        # Get a remedy suggestion based on the dominant emotion
-        remedy_suggestion = get_remedy_suggestion(dominant_emotion)
-        
-        responses[question] = {
-            "Response": user_response,
-            "Analysis": analysis,
-            "Suggested Remedy": remedy_suggestion,
-        }
-        
+        # Display Emotion Analysis
         st.success(f"🎯 Emotion Analysis: {analysis}")
-        st.write(f"🧘‍♀️ Suggested Activity: {remedy_suggestion}")
+        
+        # Display Activity Suggestion
+        if analysis:
+            max_emotion = max(analysis, key=analysis.get)
+            activity_suggestion = suggest_activity(analysis)
+            st.write(f"### 🧘 Suggested Activity: {activity_suggestion}")
 
 # ---- Display Results ----
 if st.button("Submit Responses"):
@@ -106,7 +108,8 @@ if st.button("Submit Responses"):
             st.write(f"#### Question {i}: {question}")
             st.write(f"**Your Response:** {details['Response']}")
             st.write(f"**Emotion Analysis:** {details['Analysis']}")
-            st.write(f"**Suggested Remedy:** {details['Suggested Remedy']}")
+            activity_suggestion = suggest_activity(details["Analysis"])
+            st.write(f"**Suggested Activity:** {activity_suggestion}")
     else:
         st.warning("Please answer at least one question before submitting!")