utils/mapping.py

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')