Update app.py

app.py

@@ -1,28 +1,38 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load pre-trained sentiment analysis model
+# Load sentiment analysis pipeline
 sentiment_analysis = pipeline("sentiment-analysis", model="distilbert/distilbert-base-uncased-finetuned-sst-2-english")
 
-# analyze user's mood from text
+# Function to analyze user's mood based on input
 def analyze_mood(user_input):
-    result = sentiment_analysis(user_input)[0]
-    
-    # assign mood based on sentiment
-    if result["label"] == "POSITIVE":
-        mood = "Happy"
+    # Analyze the mood from input text
+    results = sentiment_analysis(user_input)
+    mood_summary = {"POSITIVE": 0, "NEGATIVE": 0, "NEUTRAL": 0}
+    suggestions = []
+
+    # Loop through all results and summarize sentiments
+    for result in results:
+        label = result["label"]
+        score = result["score"]
+        mood_summary[label] += score
+
+    # Determine the dominant mood
+    dominant_mood = max(mood_summary, key=mood_summary.get)
+
+    # Provide suggestions based on mood
+    if dominant_mood == "POSITIVE":
         suggestion = "Keep enjoying your day 😊"
-    elif result["label"] == "NEGATIVE":
-        mood = "Sad"
-        suggestion = "Try playing a game you like or practice some deep breathing exercises it might help!🍃"
+    elif dominant_mood == "NEGATIVE":
+        suggestion = "Try playing a game you like or practice some deep breathing exercises. It might help! 🍃"
     else:
-        mood = "Neutral"
         suggestion = "You're doing well! Stay calm 🌸"
-    
-    # Output mood and suggestion
-    return "Your mood is " + mood + ". " + suggestion
+
+    # Return mood and suggestion
+    return f"Your mood seems mostly {dominant_mood.lower()}.", suggestion
 
 inputs = gr.Textbox(label="How are you feeling today?", placeholder="Type your thoughts here...")
 outputs = gr.Textbox(label="Mood and Suggestion")
+interface = gr.Interface(fn=analyze_mood, inputs=inputs, outputs=outputs, title="Mood Analyzer with Suggestions")
 
-gr.Interface(fn=analyze_mood, inputs=inputs, outputs=outputs, title="Mood Analyzer with Suggestions").launch()
+interface.launch()