Initial commit

hooks/mood_mapping.py

@@ -0,0 +1,114 @@
+"""
+Mood Mapping for Reachy Mini Emotion Clusters
+
+Maps high-level moods to clusters of emotion moves for ambient presence during TTS.
+"""
+
+# Mood clusters - each mood contains a list of related emotions
+MOOD_CLUSTERS = {
+    'thoughtful': [
+        'thoughtful1', 'thoughtful2', 'curious1', 'inquiring1', 'inquiring2',
+        'inquiring3', 'attentive1', 'attentive2', 'understanding1', 'understanding2'
+    ],
+
+    'energetic': [
+        'cheerful1', 'enthusiastic1', 'enthusiastic2', 'electric1', 'success1',
+        'success2', 'proud1', 'proud2', 'proud3', 'amazed1', 'yes1'
+    ],
+
+    'playful': [
+        'laughing1', 'laughing2', 'dance1', 'dance2', 'dance3', 'come1',
+        'electric1', 'oops1', 'oops2'
+    ],
+
+    'calm': [
+        'calming1', 'serenity1', 'relief1', 'relief2', 'understanding1',
+        'understanding2', 'welcoming1', 'welcoming2', 'grateful1'
+    ],
+
+    'confused': [
+        'confused1', 'uncertain1', 'lost1', 'oops1', 'oops2',
+        'incomprehensible2', 'uncomfortable1'
+    ],
+
+    'frustrated': [
+        'frustrated1', 'impatient1', 'impatient2', 'irritated1', 'irritated2',
+        'exhausted1', 'tired1', 'resigned1', 'displeased1', 'displeased2'
+    ],
+
+    'sad': [
+        'sad1', 'sad2', 'downcast1', 'lonely1', 'no_sad1', 'yes_sad1',
+        'resigned1', 'uncomfortable1'
+    ],
+
+    'surprised': [
+        'surprised1', 'surprised2', 'amazed1', 'oops1', 'oops2',
+        'incomprehensible2', 'fear1', 'scared1'
+    ],
+
+    'angry': [
+        'furious1', 'rage1', 'frustrated1', 'irritated1', 'irritated2',
+        'contempt1', 'disgusted1', 'reprimand1', 'reprimand2', 'reprimand3'
+    ],
+
+    'helpful': [
+        'helpful1', 'helpful2', 'welcoming1', 'welcoming2', 'grateful1',
+        'understanding1', 'understanding2', 'attentive1', 'attentive2', 'yes1'
+    ],
+
+    'shy': [
+        'shy1', 'uncertain1', 'uncomfortable1', 'downcast1', 'anxiety1'
+    ],
+
+    'sleepy': [
+        'sleep1', 'tired1', 'exhausted1', 'boredom1', 'boredom2', 'resigned1'
+    ],
+
+    'affectionate': [
+        'loving1', 'grateful1', 'welcoming1', 'welcoming2', 'cheerful1',
+        'shy1', 'come1'
+    ],
+
+    'defiant': [
+        'no1', 'no_excited1', 'go_away1', 'contempt1', 'reprimand1',
+        'reprimand2', 'reprimand3', 'indifferent1'
+    ],
+
+    'neutral': [
+        'attentive1', 'attentive2', 'thoughtful1', 'curious1', 'yes1',
+        'understanding1', 'calming1', 'serenity1'
+    ]
+}
+
+def get_mood_emotions(mood_name):
+    """
+    Get the list of emotions for a given mood.
+
+    Args:
+        mood_name: Name of the mood (e.g., 'thoughtful', 'energetic')
+
+    Returns:
+        List of emotion names, or None if mood not found
+    """
+    return MOOD_CLUSTERS.get(mood_name.lower())
+
+def get_all_moods():
+    """
+    Get list of all available mood names.
+
+    Returns:
+        List of mood names
+    """
+    return list(MOOD_CLUSTERS.keys())
+
+def validate_mood(mood_name):
+    """
+    Check if a mood name is valid.
+
+    Args:
+        mood_name: Name to validate
+
+    Returns:
+        True if valid, False otherwise
+    """
+    return mood_name.lower() in MOOD_CLUSTERS