Update utils.py

utils.py

@@ -300,11 +300,30 @@ def get_facial_attributes(face):
                 confidence = float(emotion[emotion_idx])
                 
                 if confidence > 0.4:  # Only add if confident
+                    
+                    expression_desc = None
+                    
                     if emotion_name == 'happiness':
-                        attributes['expression'] = 'smiling'
-                        attributes['description'].append('smiling')
-                    elif emotion_name not in ['neutral']:
-                        attributes['expression'] = emotion_name
+                        expression_desc = 'smiling'
+                    elif emotion_name == 'surprise':
+                        expression_desc = 'surprised expression'
+                    elif emotion_name == 'sadness':
+                        expression_desc = 'sad expression'
+                    elif emotion_name == 'anger':
+                        expression_desc = 'angry expression'
+                    elif emotion_name == 'neutral':
+                        expression_desc = 'neutral expression'
+                    
+                    # Add other emotions like 'disgust' or 'fear' if desired
+                    
+                    if expression_desc:
+                        attributes['expression'] = expression_desc
+                        
+                        # Only add non-neutral expressions to the prompt description
+                        if emotion_name != 'neutral':
+                            if expression_desc not in attributes['description']:
+                                attributes['description'].append(expression_desc)
+                                
     except (ValueError, TypeError, AttributeError, IndexError) as e:
         # Expression not available in this model
         pass
@@ -506,4 +525,4 @@ def enhance_face_crop(face_crop):
     return face_crop_final
 
 
-print("[OK] Utilities loaded")
+print("[OK] Utilities loaded")