Update app.py

app.py

@@ -67,10 +67,12 @@ def perform_emotion_detection(text):
 
         emotion_results = emotion_classifier(text)
         print(f"Raw emotion classifier output: {emotion_results}")
-        if not emotion_results or not isinstance(emotion_results, list):
-            st.error("Emotion classifier returned invalid results.")
+        if not emotion_results or not isinstance(emotion_results, list) or not emotion_results[0]:
+            st.error("Emotion classifier returned invalid or empty results.")
             return {}, "neutral", {}, "NEUTRAL"
 
+        # Access the first inner list, which contains the emotion dictionaries
+        emotion_results = emotion_results[0]
         emotion_map = {
             "joy": "😊", "anger": "😡", "disgust": "🤢", "fear": "😨",
             "sadness": "😭", "surprise": "😲"
@@ -81,7 +83,14 @@ def perform_emotion_detection(text):
 
         emotions_dict = {}
         for result in emotion_results:
-            emotions_dict[result['label']] = result['score']
+            if isinstance(result, dict) and 'label' in result and 'score' in result:
+                emotions_dict[result['label']] = result['score']
+            else:
+                print(f"Invalid result format: {result}")
+
+        if not emotions_dict:
+            st.error("No valid emotions detected.")
+            return {}, "neutral", {}, "NEUTRAL"
 
         filtered_emotions = {k: v for k, v in emotions_dict.items() if v > 0.01}