Update app.py

app.py

@@ -74,14 +74,18 @@ def get_emotion(text: str) -> str:
 '''
 def get_emotion(text: str) -> str:
     try:
+        # Get the classification result
         result = emotion_classifier(text)
-        if result and isinstance(result, list):
-            emotion_label = result[0]['label']
-            return emotion_label.capitalize()
-        return "Unknown emotion"
+        
+        # Check if the result is a list and has at least one item
+        if result and isinstance(result, list) and len(result) > 0:
+            emotion_label = result[0]['label']  # Access the label of the first result
+            return emotion_label.capitalize()  # Capitalize for readability
+        return "Unknown emotion"  # Fallback if no label found
     except Exception as e:
         return f"Error in emotion detection: {e}"
 
+
 # Function to handle long conversations for emotion detection
 def analyze_long_text_emotion(text: str, chunk_size: int = 512) -> str:
     # Split the text into chunks of the specified size