def map_to_mood(label): mapping = { 'joy': 'Happy', 'neutral': 'Neutral', 'sadness': 'Sad', 'anger': 'Angry', 'disgust': 'Angry', 'fear': 'Sad', 'surprise': 'Neutral' } return mapping.get(label, 'Neutral')