Update app.py

app.py

@@ -5,14 +5,15 @@ from transformers import pipeline
 sentiment_analysis = pipeline("sentiment-analysis", model="distilbert/distilbert-base-uncased-finetuned-sst-2-english")
 
 # Function to analyze user's mood based on input
-def analyze_mood(user_input):
+    def analyze_mood(user_input):
     # Analyze the mood from input text
     result = sentiment_analysis(user_input)[0]
     # Define mood and suggestion based on sentiment analysis
-    if result["label"] == "POSITIVE":
+    score = result["score"]
+    if result["label"] == "POSITIVE" and score > 0.85:
         mood = "Happy"
         suggestion = "Keep doing what you're doing! 😊"
-    elif result["label"] == "NEGATIVE":
+    elif result["label"] == "NEGATIVE" and score > 0.85:
         mood = "Sad"
         suggestion = "Try to talk to someone, or take a break 💡"
     else: