Update app.py

app.py

@@ -72,6 +72,16 @@ def get_emotion(text: str) -> str:
     except Exception as e:
         return f"Error in emotion detection: {e}"
 
+def get_emotion(text: str) -> str:
+    try:
+        result = emotion_classifier(text)
+        if result and isinstance(result, list):
+            emotion_label = result[0]['label']
+            return emotion_label.capitalize()
+        return "Unknown emotion"
+    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