Update app.py

app.py

@@ -1,8 +1,8 @@
 import streamlit as st
-from transformers import DistilBertTokenizer, DistilBertForSequenceClassification, pipeline
-import os
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+import torch
 
-# Set page config
+# Set page config for a clean interface
 st.set_page_config(page_title="Emotion Prediction & Well-being Suggestions", page_icon="🌺", layout="centered")
 
 # Title and Introduction
@@ -11,36 +11,49 @@ st.markdown("""
 This app uses AI to understand your emotions based on your responses. Afterward, you'll receive well-being suggestions to improve your mood, tailored specifically for you and your cultural context in Hawaii.
 """)
 
-# Load the emotion classification model and tokenizer
-def load_model():
-    model_name = "j-hartmann/emotion-english-distilroberta-base"
-    try:
-        # Attempt to load the model and tokenizer
-        tokenizer = DistilBertTokenizer.from_pretrained(model_name)
-        model = DistilBertForSequenceClassification.from_pretrained(model_name)
-        emotion_classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)
-        return emotion_classifier
-    except Exception as e:
-        st.error(f"Error loading model: {e}")
-        return None
+# Load the tokenizer and model
+tokenizer = AutoTokenizer.from_pretrained("j-hartmann/emotion-english-distilroberta-base")
+model = AutoModelForSequenceClassification.from_pretrained("j-hartmann/emotion-english-distilroberta-base")
 
-emotion_classifier = load_model()
-
-# If the model didn't load, exit the app
-if not emotion_classifier:
-    st.stop()
-
-# Define well-being suggestions based on the emotion
+# Define well-being suggestions with URLs for articles and videos based on the emotion
 def get_wellbeing_suggestions(emotion):
     suggestions = {
-        "joy": ["Keep enjoying life! Consider a walk on the beach, or some hula dancing to feel the rhythm of the island."],
-        "anger": ["Practice deep breathing and mindfulness exercises. Perhaps try surfing or a peaceful walk in nature."],
-        "sadness": ["Try a short meditation session or call a friend to chat. Hawaii offers stunning sunsets perfect for reflection."],
-        "fear": ["Take a calming breath and practice grounding techniques. Explore Hawaiian mindfulness practices to feel centered."],
-        "surprise": ["Embrace the feeling and explore new activities like paddleboarding or hiking."],
-        "disgust": ["Do something relaxing, like yoga or watching the waves crash against the shore."]
+        "joy": [
+            "Keep enjoying life! Consider a walk on the beach, or some hula dancing to feel the rhythm of the island.",
+            "Articles and Resources: [Emotional Wellness Toolkit - NIH](https://www.nih.gov/health-information/emotional-wellness-toolkit)",
+            "Videos: [Hula Dance for Well-being](https://youtu.be/m1vaUGtyo-A)"
+        ],
+        "anger": [
+            "Practice deep breathing and mindfulness exercises. Perhaps try surfing or a peaceful walk in nature.",
+            "Articles and Resources: [Tips for Dealing with Anxiety - HelpGuide](https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety)",
+            "Videos: [Mindful Breathing Meditation](https://www.youtube.com/shorts/Tq49ajl7c8Q?feature=share)"
+        ],
+        "sadness": [
+            "Try a short meditation session or call a friend to chat. Hawaii offers stunning sunsets perfect for reflection.",
+            "Articles and Resources: [Health A-Z - Harvard Health](https://www.health.harvard.edu/health-a-to-z)",
+            "Videos: [Mindful Breathing Meditation](https://youtu.be/MIc299Flibs)"
+        ],
+        "fear": [
+            "Take a calming breath and practice grounding techniques. Explore Hawaiian mindfulness practices to feel centered.",
+            "Articles and Resources: [Mindful Breathing Meditation - HelpGuide](https://www.helpguide.org/mental-health/meditation/mindful-breathing-meditation)",
+            "Videos: [Relaxing Mindfulness Exercise](https://www.youtube.com/shorts/fwH8Ygb0K60?feature=share)"
+        ],
+        "surprise": [
+            "Embrace the feeling and explore new activities like paddleboarding or hiking.",
+            "Articles and Resources: [Emotional Wellness Toolkit - NIH](https://www.nih.gov/health-information/emotional-wellness-toolkit)",
+            "Videos: [Surprising Relaxation Tips](https://youtu.be/yGKKz185M5o)"
+        ],
+        "disgust": [
+            "Do something relaxing, like yoga or watching the waves crash against the shore.",
+            "Articles and Resources: [Tips for Dealing with Anxiety - HelpGuide](https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety)",
+            "Videos: [Hawaiian Mindfulness Practice](https://www.youtube.com/shorts/hTXMi7ZBKdM?feature=share)"
+        ]
     }
-    return suggestions.get(emotion, ["Try some breathing exercises and give yourself time to relax."])
+    return suggestions.get(emotion, [
+        "Try some breathing exercises and give yourself time to relax.",
+        "Articles and Resources: [Health A-Z - Harvard Health](https://www.health.harvard.edu/health-a-to-z)",
+        "Videos: [Calming Breath Exercises](https://youtu.be/Y8HIFRPU6pM)"
+    ])
 
 # Emotional Health Questions
 questions = [
@@ -58,18 +71,26 @@ for question in questions:
 if all(responses):
     user_input = " ".join(responses)
     
-    # Predict emotion based on the input
-    emotion_results = emotion_classifier(user_input)
-    emotion = emotion_results[0]['label']
+    # Tokenize the user input and make predictions
+    inputs = tokenizer(user_input, return_tensors="pt", truncation=True, padding=True)
+    outputs = model(**inputs)
+    predictions = torch.argmax(outputs.logits, dim=-1)  # Get the index of the predicted class
+    emotion_label = model.config.id2label[predictions.item()]  # Get the label of the predicted class
     
     # Show emotion and well-being suggestions
-    st.subheader(f"Your Predicted Emotion: {emotion.capitalize()}")
+    st.subheader(f"Your Predicted Emotion: {emotion_label.capitalize()}")
     st.write(f"Based on your responses, we suggest the following to improve your well-being:")
     
-    wellbeing_suggestions = get_wellbeing_suggestions(emotion.lower())
+    wellbeing_suggestions = get_wellbeing_suggestions(emotion_label.lower())
     for suggestion in wellbeing_suggestions:
         st.write(f"- {suggestion}")
     
+    # Show Summary Button
+    if st.button("Show Summary"):
+        st.subheader("Summary of Well-being Suggestions")
+        for suggestion in wellbeing_suggestions:
+            st.write(f"- {suggestion}")
+    
     st.markdown("---")
     st.write("For more well-being tips and resources, explore the following links:")
     st.markdown("[Hawaii Mindfulness Practice](https://www.hawaiimindfulness.org)")
@@ -78,12 +99,31 @@ if all(responses):
 else:
     st.warning("Please answer all the questions to receive emotional health suggestions.")
 
-# Add custom background image
+# Add custom background image to make it visually appealing
 st.markdown("""
     <style>
     .stApp {
         background-image: url('https://images.unsplash.com/photo-1602231353203-b1e5e2191b68');
         background-size: cover;
+        background-position: center;
+    }
+    .stButton > button {
+        background-color: #F57F17; 
+        color: white;
+        font-size: 18px;
+        padding: 10px;
+    }
+    .stButton > button:hover {
+        background-color: #FF6F00;
+    }
+    .stTextInput > div > input {
+        background-color: rgba(255, 255, 255, 0.7);
+        font-size: 16px;
+        color: #333;
+    }
+    .stMarkdown p {
+        font-size: 16px;
+        color: white;
     }
     </style>
 """, unsafe_allow_html=True)