Update app.py

app.py

@@ -24,6 +24,20 @@ def check_for_smile(face_landmarks):
     return False
 
 # Display emotion with post-processing to check for smiles
+def display_emotion(emotion):
+    """Map predicted emotion index to a label"""
+    emotion_map = {
+        0: "Anger",
+        1: "Disgust",
+        2: "Fear",
+        3: "Happiness",
+        4: "Sadness",
+        5: "Surprise",
+        6: "Neutral"
+    }
+    return emotion_map.get(emotion, "Unknown")
+
+# Display emotion with smile detection
 def display_emotion_with_smile(emotion, face_landmarks=None):
     if emotion == 6 and face_landmarks:  # 'Neutral' is typically 6 in the emotion_map
         if check_for_smile(face_landmarks):