Update app.py

app.py

@@ -1,144 +1,117 @@
 import streamlit as st
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
 from transformers import pipeline
 
-# Load emotion classification model
+# Set page config as the very first Streamlit command
+st.set_page_config(page_title="Emotion Detection and Well-Being Suggestions", layout="wide")
+
+# Load pre-trained model and tokenizer
 @st.cache_resource
 def load_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 = AutoTokenizer.from_pretrained("j-hartmann/emotion-english-distilroberta-base")
+    model = AutoModelForSequenceClassification.from_pretrained("j-hartmann/emotion-english-distilroberta-base")
+    return tokenizer, 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": []
-    })
+tokenizer, model = load_model()
 
-# Streamlit UI
-def main():
-    # Set the background image
-    st.markdown("""
+# Add a background image (use your own URL or file path here)
+st.markdown(
+    """
     <style>
-    .stApp {
-        background-image: url('https://www.example.com/your-image.jpg');
+    body {
+        background-image: url('https://drive.google.com/uc?export=view&id=1TydbjDRL5fXLMxQgaPkIg9oGOJlU2IWt');
         background-size: cover;
-        background-position: center;
+        background-repeat: no-repeat;
+        background-attachment: fixed;
+        color: white;
+    }
+    .stButton button {
+        background-color: #6c63ff;
+        color: white;
+        font-size: 20px;
     }
     </style>
-    """, unsafe_allow_html=True)
-    
-    # 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:", "")
+    """,
+    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}")
     
-    if user_input:
-        # Clean the input text (stripping unnecessary spaces, lowercasing)
-        clean_input = user_input.strip().lower()
+    # 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)")
         
-        # 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"])
+        st.write("Relaxation Videos:")
+        st.markdown("[Watch on YouTube](https://youtu.be/m1vaUGtyo-A)")
 
-            # Display links
-            if suggestions["links"]:
-                st.write("Useful Resources:")
-                for link in suggestions["links"]:
-                    st.markdown(f"[{link}]({link})")
+    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)")
 
-            # 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)}")
+    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)")
+        
+        st.write("Relaxation Videos:")
+        st.markdown("[Watch on YouTube](https://youtu.be/yGKKz185M5o)")
+
+    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)")
+
+    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)")
 
-# Run the Streamlit app
-if __name__ == "__main__":
-    main()
+    # 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)")