Update app.py

app.py

@@ -1,117 +1,144 @@
 import streamlit as st
-from transformers import AutoTokenizer, AutoModelForSequenceClassification
 from transformers import pipeline
 
-# Ensure this is the first Streamlit command
-st.set_page_config(page_title="Emotion Detection and Well-Being Suggestions", layout="wide")
-
-# Load pre-trained model and tokenizer
+# Load emotion classification model
 @st.cache_resource
 def load_model():
-    tokenizer = AutoTokenizer.from_pretrained("j-hartmann/emotion-english-distilroberta-base")
-    model = AutoModelForSequenceClassification.from_pretrained("j-hartmann/emotion-english-distilroberta-base")
-    return tokenizer, model
+    try:
+        emotion_classifier = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base")
+        return emotion_classifier
+    except Exception as e:
+        st.error(f"Error loading model: {str(e)}")
+        return None
+
+emotion_classifier = load_model()
 
-tokenizer, model = load_model()
+# Well-being suggestions based on emotions
+def get_well_being_suggestions(emotion):
+    suggestions = {
+        "joy": {
+            "text": "You're feeling joyful! Keep the positivity going.",
+            "links": [
+                "https://www.nih.gov/health-information/emotional-wellness-toolkit",
+                "https://www.health.harvard.edu/health-a-to-z",
+                "https://www.helpguide.org/mental-health/meditation/mindful-breathing-meditation"
+            ],
+            "videos": [
+                "https://youtu.be/m1vaUGtyo-A",
+                "https://youtu.be/MIc299Flibs"
+            ]
+        },
+        "anger": {
+            "text": "You're feeling angry. Take a moment to calm down.",
+            "links": [
+                "https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety",
+                "https://www.helpguide.org/mental-health/meditation/mindful-breathing-meditation"
+            ],
+            "videos": [
+                "https://youtu.be/m1vaUGtyo-A",
+                "https://www.youtube.com/shorts/fwH8Ygb0K60?feature=share"
+            ]
+        },
+        "sadness": {
+            "text": "You're feeling sad. It's okay to take a break.",
+            "links": [
+                "https://www.nih.gov/health-information/emotional-wellness-toolkit",
+                "https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety"
+            ],
+            "videos": [
+                "https://youtu.be/-e-4Kx5px_I",
+                "https://youtu.be/Y8HIFRPU6pM"
+            ]
+        },
+        "fear": {
+            "text": "You're feeling fearful. Try some relaxation techniques.",
+            "links": [
+                "https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety",
+                "https://www.health.harvard.edu/health-a-to-z"
+            ],
+            "videos": [
+                "https://www.youtube.com/shorts/Tq49ajl7c8Q?feature=share",
+                "https://youtu.be/yGKKz185M5o"
+            ]
+        },
+        "disgust": {
+            "text": "You're feeling disgusted. Take a deep breath and refocus.",
+            "links": [
+                "https://www.health.harvard.edu/health-a-to-z",
+                "https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety"
+            ],
+            "videos": [
+                "https://youtu.be/MIc299Flibs",
+                "https://youtu.be/-e-4Kx5px_I"
+            ]
+        },
+    }
+    return suggestions.get(emotion, {
+        "text": "Feeling neutral? That's okay! Take care of your mental health.",
+        "links": [],
+        "videos": []
+    })
 
-# Add a background image (use your own URL or file path here)
-st.markdown(
-    """
+# Streamlit UI
+def main():
+    # Set the background image
+    st.markdown("""
     <style>
-    body {
-        background-image: url('https://drive.google.com/uc?export=view&id=1TydbjDRL5fXLMxQgaPkIg9oGOJlU2IWt');
+    .stApp {
+        background-image: url('https://www.example.com/your-image.jpg');
         background-size: cover;
-        background-repeat: no-repeat;
-        background-attachment: fixed;
-        color: white;
-    }
-    .stButton button {
-        background-color: #6c63ff;
-        color: white;
-        font-size: 20px;
+        background-position: center;
     }
     </style>
-    """,
-    unsafe_allow_html=True,
-)
-
-# Display header
-st.title("Emotion Detection and Well-Being Suggestions")
-
-# User input for text (emotion detection)
-user_input = st.text_area("How are you feeling today?", "Enter your thoughts here...")
-
-# Model prediction
-if user_input:
-    pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
-    result = pipe(user_input)
-
-    # Extracting the emotion from the model's result
-    emotion = result[0]['label']
-
-    # Display emotion
-    st.write(f"**Emotion Detected:** {emotion}")
+    """, unsafe_allow_html=True)
     
-    # Provide suggestions based on the detected emotion
-    if emotion == 'joy':
-        st.write("You're feeling happy! Keep up the great mood!")
-        st.write("Useful Resources:")
-        st.markdown("[Relaxation Techniques](https://www.helpguide.org/mental-health/meditation/mindful-breathing-meditation)")
-        st.write("[Dealing with Stress](https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety)")
-        st.write("[Emotional Wellness Toolkit](https://www.nih.gov/health-information/emotional-wellness-toolkit)")
-        
-        st.write("Relaxation Videos:")
-        st.markdown("[Watch on YouTube](https://youtu.be/m1vaUGtyo-A)")
-
-    elif emotion == 'anger':
-        st.write("You're feeling angry. It's okay to feel this way. Let's try to calm down.")
-        st.write("Useful Resources:")
-        st.markdown("[Emotional Wellness Toolkit](https://www.nih.gov/health-information/emotional-wellness-toolkit)")
-        st.write("[Stress Management Tips](https://www.health.harvard.edu/health-a-to-z)")
-        st.write("[Dealing with Anger](https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety)")
-        
-        st.write("Relaxation Videos:")
-        st.markdown("[Watch on YouTube](https://youtu.be/MIc299Flibs)")
-
-    elif emotion == 'fear':
-        st.write("You're feeling fearful. Take a moment to breathe and relax.")
-        st.write("Useful Resources:")
-        st.markdown("[Mindfulness Practices](https://www.helpguide.org/mental-health/meditation/mindful-breathing-meditation)")
-        st.write("[Coping with Anxiety](https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety)")
-        st.write("[Emotional Wellness Toolkit](https://www.nih.gov/health-information/emotional-wellness-toolkit)")
+    # Title of the app
+    st.title("Emotion Prediction and Well-being Suggestions")
+    
+    # User input for emotional state
+    st.header("Tell us how you're feeling today!")
+    
+    user_input = st.text_area("Enter a short sentence about your current mood:", "")
+    
+    if user_input:
+        # Clean the input text (stripping unnecessary spaces, lowercasing)
+        clean_input = user_input.strip().lower()
         
-        st.write("Relaxation Videos:")
-        st.markdown("[Watch on YouTube](https://youtu.be/yGKKz185M5o)")
+        # Use the model to predict emotion
+        try:
+            result = emotion_classifier(clean_input)
+            st.write(f"Raw Model Result: {result}")  # Debug output to see raw result
+            
+            emotion = result[0]['label'].lower()
+            
+            st.subheader(f"Emotion Detected: {emotion.capitalize()}")
+            
+            # Get well-being suggestions based on emotion
+            suggestions = get_well_being_suggestions(emotion)
+            
+            # Display text suggestions
+            st.write(suggestions["text"])
 
-    elif emotion == 'sadness':
-        st.write("You're feeling sad. It's okay to take a break.")
-        st.write("Useful Resources:")
-        st.markdown("[Emotional Wellness Toolkit](https://www.nih.gov/health-information/emotional-wellness-toolkit)")
-        st.write("[Dealing with Anxiety](https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety)")
-        
-        st.write("Relaxation Videos:")
-        st.markdown("[Watch on YouTube](https://youtu.be/-e-4Kx5px_I)")
+            # Display links
+            if suggestions["links"]:
+                st.write("Useful Resources:")
+                for link in suggestions["links"]:
+                    st.markdown(f"[{link}]({link})")
 
-    elif emotion == 'surprise':
-        st.write("You're feeling surprised. It's okay to feel neutral!")
-        st.write("Useful Resources:")
-        st.markdown("[Managing Stress](https://www.health.harvard.edu/health-a-to-z)")
-        st.write("[Coping Strategies](https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety)")
-        
-        st.write("Relaxation Videos:")
-        st.markdown("[Watch on YouTube](https://youtu.be/m1vaUGtyo-A)")
+            # Display video links
+            if suggestions["videos"]:
+                st.write("Relaxation Videos:")
+                for video in suggestions["videos"]:
+                    st.markdown(f"[Watch here]({video})")
+            
+            # Add a button for a summary
+            if st.button('Summary'):
+                st.write(f"Emotion detected: {emotion.capitalize()}. Here are your well-being suggestions to enhance your mood.")
+                st.write("Explore the links and videos to improve your emotional health!")
+            
+        except Exception as e:
+            st.error(f"Error predicting emotion: {str(e)}")
 
-    # Add a button for summary
-    if st.button("Show Summary"):
-        st.write(f"Emotion Detected: {emotion}")
-        st.write("Useful Resources based on your mood:")
-        if emotion == 'joy':
-            st.write("[Relaxation Techniques](https://www.helpguide.org/mental-health/meditation/mindful-breathing-meditation)")
-        elif emotion == 'anger':
-            st.write("[Stress Management Tips](https://www.health.harvard.edu/health-a-to-z)")
-        elif emotion == 'fear':
-            st.write("[Mindfulness Practices](https://www.helpguide.org/mental-health/meditation/mindful-breathing-meditation)")
-        elif emotion == 'sadness':
-            st.write("[Dealing with Anxiety](https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety)")
-        elif emotion == 'surprise':
-            st.write("[Managing Stress](https://www.health.harvard.edu/health-a-to-z)")
+# Run the Streamlit app
+if __name__ == "__main__":
+    main()