Update app.py

app.py

@@ -1,198 +1,117 @@
 import streamlit as st
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
 from transformers import pipeline
 
-# Load emotion classification model
+# 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)}")  # Error handling
-        return None
+    tokenizer = AutoTokenizer.from_pretrained("j-hartmann/emotion-english-distilroberta-base")
+    model = AutoModelForSequenceClassification.from_pretrained("j-hartmann/emotion-english-distilroberta-base")
+    return tokenizer, model
 
-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"
-            ]
-        },
-        # New addition for "Surprise"
-        "surprise": {
-            "text": "You're feeling surprised. Take a deep breath and ground yourself.",
-            "links": [
-                "https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety",
-                "https://www.health.harvard.edu/health-a-to-z",
-                "https://www.psychologytoday.com/us/blog/mindful-anger/201908/5-ways-to-deal-with-unexpected-surprises"
-            ],
-            "videos": [
-                "https://youtu.be/MIc299Flibs",
-                "https://www.youtube.com/shorts/Tq49ajl7c8Q?feature=share",
-                "https://youtu.be/m1vaUGtyo-A"
-            ]
-        },
-    }
-    return suggestions.get(emotion, {
-        "text": "Feeling neutral? That's okay! Take care of your mental health.",
-        "links": [],
-        "videos": []
-    })
+# Set page configuration
+st.set_page_config(page_title="Emotion Detection and Well-Being Suggestions", layout="wide")
 
-# Streamlit UI
-def main():
-    # Add the background image
-    st.markdown("""
+# Add a background image (use your own URL or file path here)
+st.markdown(
+    """
     <style>
-    .stApp {
+    body {
         background-image: url('https://drive.google.com/uc?export=view&id=1TydbjDRL5fXLMxQgaPkIg9oGOJlU2IWt');
         background-size: cover;
-        background-position: center;
-        color: white;  /* Set text color to white */
-    }
-    
-    h1, h2, h3 {
+        background-repeat: no-repeat;
+        background-attachment: fixed;
         color: white;
-        font-family: 'Arial', sans-serif;
     }
-    
-    .stTextArea textarea {
-        background-color: #f2f2f2;  /* Light grey text area */
-        border-radius: 8px;
-        color: #333333;
-    }
-    
     .stButton button {
-        background-color: #9b4dca;  /* Purple button */
+        background-color: #6c63ff;
         color: white;
-        font-weight: bold;
-        border-radius: 5px;
+        font-size: 20px;
     }
+    </style>
+    """,
+    unsafe_allow_html=True,
+)
 
-    .stButton button:hover {
-        background-color: #7a33a2;  /* Darker purple on hover */
-    }
+# Display header
+st.title("Emotion Detection and Well-Being Suggestions")
 
-    .stMarkdown, .stWrite {
-        color: white !important;
-    }
+# User input for text (emotion detection)
+user_input = st.text_area("How are you feeling today?", "Enter your thoughts here...")
 
-    /* Style for links to make them more visible */
-    a {
-        color: #ffffff;
-        text-decoration: underline;
-    }
-    a:hover {
-        color: #ffccff;  /* Light purple hover effect */
-    }
-    </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:", "")
+# 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:
-        # Display Enter button only after user has entered input
-        enter_button = st.button("Enter")
+    # 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)")
         
-        if enter_button:
-            # Clean the input text (stripping unnecessary spaces, lowercasing)
-            clean_input = user_input.strip().lower()
-            
-            # 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})", unsafe_allow_html=True)
+    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})", unsafe_allow_html=True)
-            
-                # 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)")