Update app.py

app.py

@@ -46,3 +46,20 @@ iface = gr.Interface(
 
 if __name__ == "__main__":
     iface.launch()
+def detect_emotion_caption(image):
+    results = emotion_model(image)
+    top_emotion = results[0]['label'].lower()
+
+    # Simple caption mapping
+    captions = {
+        "angry": "he is angry",
+        "happy": "he is smiling",
+        "sad": "he is crying",
+        "disgust": "he looks disgusted",
+        "fear": "he looks scared",
+        "surprise": "he is surprised",
+        "neutral": "he looks neutral",
+        "contempt": "he looks contemptuous"
+    }
+
+    return captions.get(top_emotion, f"he looks {top_emotion}")