Update app.py

app.py

@@ -5,25 +5,28 @@ 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
-    score = result["score"]
-    if result["label"] == "POSITIVE" and score > 0.85:
+    mood = "Neutral"  # Default mood
+    suggestion = "You're doing okay! Stay calm 🌸"  # Default suggestion
+    
+    # Adjust mood based on analysis
+    if result["label"] == "POSITIVE" and result["score"] > 0.85:
         mood = "Happy"
         suggestion = "Keep doing what you're doing! 😊"
-    elif result["label"] == "NEGATIVE" and score > 0.85:
+    elif result["label"] == "NEGATIVE" and result["score"] > 0.85:
         mood = "Sad"
         suggestion = "Try to talk to someone, or take a break 💡"
-    else:
-        mood = "Neutral"
-        suggestion = "You're doing okay! Stay calm 🌸"
     
     # Return mood and suggestion
     return "Your mood is: " + mood, suggestion
 
+# Set up Gradio inputs and outputs
 inputs = gr.Textbox(label="How are you feeling today?", placeholder="Type your thoughts here...")
 outputs = gr.Textbox(label="Mood and Suggestion")
 
-gr.Interface(fn=analyze_mood, inputs=inputs, outputs=outputs, title="Mood Analyzer").launch()
+# Launch the Gradio interface
+gr.Interface(fn=analyze_mood, inputs=inputs, outputs=outputs, title="Mood Analyzer").launch()