Update app.py

app.py

@@ -13,7 +13,7 @@ st.set_page_config(
 
 # ---- App Title ----
 st.title("🌟 Emotion Prediction App 🌈")
-st.subheader("Understand your emotions better with AI-powered predictions!")
+st.subheader("Get emotional insights with tailored resources!")
 
 # ---- Function to Load Emotion Analysis Model ----
 @st.cache_resource
@@ -49,35 +49,64 @@ def predict_emotion(text):
         st.error(f"⚠️ Prediction failed: {e}")
         return {"Error": f"Prediction failed: {e}"}
 
-# ---- Suggesting Activities Based on Emotional State ----
+# ---- Enhanced Suggestion Database ----
+suggestion_database = {
+    "sadness": {
+        "suggestions": ["Try a guided meditation", "Take a walk in nature", "Connect with a friend"],
+        "articles": [
+            {"title": "Overcoming Sadness", "url": "https://example.com/sadness1"},
+            {"title": "Understanding Depression", "url": "https://example.com/sadness2"},
+        ],
+        "videos": [
+            {"title": "Mindfulness for Sadness", "url": "https://www.youtube.com/watch?v=sadnessvideo1"},
+            {"title": "Coping with Grief", "url": "https://www.youtube.com/watch?v=sadnessvideo2"},
+        ],
+    },
+    "joy": {
+        "suggestions": ["Practice gratitude", "Engage in a hobby", "Spend time with loved ones"],
+        "articles": [
+            {"title": "The Benefits of Joy", "url": "https://example.com/joy1"},
+            {"title": "Maintaining Positive Emotions", "url": "https://example.com/joy2"},
+        ],
+        "videos": [
+            {"title": "Boosting Your Happiness", "url": "https://www.youtube.com/watch?v=joyvideo1"},
+            {"title": "Practicing Gratitude", "url": "https://www.youtube.com/watch?v=joyvideo2"},
+        ],
+    },
+    "neutral": {
+        "suggestions": ["Take a break", "Engage in a relaxing activity", "Spend time in nature"],
+        "articles": [
+            {"title": "Importance of Self-Care", "url": "https://example.com/selfcare1"},
+            {"title": "Stress Management Techniques", "url": "https://example.com/stress1"}
+        ],
+        "videos": [
+            {"title": "Relaxation Techniques", "url": "https://www.youtube.com/watch?v=relaxvideo1"},
+            {"title": "Mindfulness Exercises", "url": "https://www.youtube.com/watch?v=mindfulnessvideo1"}
+        ]
+    }
+}
+
+# Function to fetch relevant resources
+def get_relevant_resources(emotion):
+    resources = suggestion_database.get(emotion, {})
+    return resources.get("suggestions", []), resources.get("articles", []), resources.get("videos", [])
+
+# Enhanced Suggestion Function
 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."
+    max_emotion = max(emotion_analysis, key=emotion_analysis.get) if emotion_analysis else "neutral"
+    suggestions, articles, videos = get_relevant_resources(max_emotion)
+    return {
+        "suggestions": suggestions,
+        "articles": articles,
+        "videos": videos,
+    }
 
 # ---- User Input Section ----
-st.write("### 🌺 Let's Get Started!")
+st.write("### 🌺 Answer 3 Simple Questions!")
 questions = [
-    "How are you feeling today?",
-    "Describe your mood in a few words.",
-    "What was the most significant emotion you felt this week?",
-    "How do you handle stress or challenges?",
-    "What motivates you the most right now?",
+    "How are you feeling right now?",
+    "What was the most significant emotion you experienced today?",
+    "What helps you feel better when you’re stressed?"
 ]
 
 responses = {}
@@ -99,24 +128,31 @@ for i, question in enumerate(questions, start=1):
         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"):
-    st.write("### 📊 Emotion Analysis Results")
-
-    # Prepare the data for the summary table
+            st.write(f"### 🧘 Suggested Activity: {activity_suggestion['suggestions'][0]}")
+            st.write(f"📖 Suggested Articles:")
+            for article in activity_suggestion['articles']:
+                st.markdown(f"- [{article['title']}]({article['url']})")
+            st.write(f"🎥 Suggested Videos:")
+            for video in activity_suggestion['videos']:
+                st.markdown(f"- [{video['title']}]({video['url']})")
+
+# ---- Display Summary Button ----
+if st.button("Get Summary"):
+    st.write("### 📊 Your Emotional Insights Summary")
+
+    # Prepare the summary data
     summary_data = []
     for i, (question, details) in enumerate(responses.items(), start=1):
-        # Activity suggestion for each response
+        max_emotion = max(details["Analysis"], key=details["Analysis"].get)
         activity_suggestion = suggest_activity(details["Analysis"])
         
-        # Add data to the summary list
         summary_data.append({
             "Question": question,
             "Your Response": details["Response"],
-            "Emotion Analysis": details["Analysis"],
-            "Suggested Activity": activity_suggestion
+            "Emotion Analysis": max_emotion,
+            "Suggested Activity": activity_suggestion["suggestions"][0],
+            "Suggested Articles": ", ".join([article['title'] for article in activity_suggestion['articles']]),
+            "Suggested Videos": ", ".join([video['title'] for video in activity_suggestion['videos']]),
         })
 
     # Create a DataFrame from the summary data
@@ -127,7 +163,7 @@ if st.button("Submit Responses"):
 
     # Provide additional suggestions at the end of all questions
     st.write("### 🌟 Final Suggestions")
-    st.write("It’s great that you're exploring your emotions! Regular mindfulness practices, deep breathing exercises, and physical activities like walking can help keep your emotions in balance and improve overall well-being. 🌱")
+    st.write("Incorporating mindfulness, deep breathing, and spending time outdoors can help with emotional balance. Keep exploring activities that promote well-being! 🌱")
 
 # ---- Footer ----
 st.markdown(
@@ -137,11 +173,3 @@ st.markdown(
     Designed for a fun and intuitive experience! 🌟  
     """
 )
-
-# ---- Error Handling and User Suggestions ----
-if emotion_analyzer is None:
-    st.error("⚠️ We couldn't load the emotion analysis model. Please check your internet connection or try again later.")
-    st.markdown("🔧 **Troubleshooting Steps:**")
-    st.markdown("1. Ensure you have a stable internet connection.")
-    st.markdown("2. If the issue persists, please refresh the page and try again.")
-    st.markdown("3. Check if the model has been updated or is temporarily unavailable.")