Update app.py

app.py

@@ -10,11 +10,17 @@ sentiment_model = load_model()
 # Define the function to analyze sentiment
 def analyze_sentiment(user_input):
     result = sentiment_model(user_input)[0]
-    sentiment = result['label']
+    sentiment = result['label'].lower()  # Convert to lowercase for easier comparison
     
-    if sentiment in ['NEGATIVE', 'NEUTRAL']:
-        return "Stay positive! 🌟 You can handle anything that comes your way."
-    return "You're on the right track! Keep shining! 🌞"
+    # Customize messages based on detected sentiment
+    if sentiment == 'negative':
+        return "Mood Detected: Negative 😔\n\nStay positive! 🌟 Remember, tough times don't last, but tough people do!"
+    elif sentiment == 'neutral':
+        return "Mood Detected: Neutral 😐\n\nIt's good to reflect on steady days. Keep your goals in mind, and stay motivated!"
+    elif sentiment == 'positive':
+        return "Mood Detected: Positive 😊\n\nYou're on the right track! Keep shining! 🌞"
+    else:
+        return "Mood Detected: Unknown 🤔\n\nKeep going, you're doing great!"
 
 # Gradio UI
 def chatbot_ui():
@@ -32,4 +38,3 @@ def chatbot_ui():
 # Launch the interface
 if __name__ == "__main__":
     chatbot_ui().launch()
-