Update app.py

app.py

@@ -48,6 +48,20 @@ 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.")
+
 # ---- User Input Section ----
 st.write("### 🌺 Let's Get Started!")
 questions = [
@@ -68,8 +82,21 @@ 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,
+        }
+        
         st.success(f"🎯 Emotion Analysis: {analysis}")
+        st.write(f"🧘‍♀️ Suggested Activity: {remedy_suggestion}")
 
 # ---- Display Results ----
 if st.button("Submit Responses"):
@@ -79,6 +106,7 @@ 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']}")
     else:
         st.warning("Please answer at least one question before submitting!")