Update app.py

app.py

@@ -171,6 +171,8 @@ def video_processing(video_file, encoded_video):
     
     # Modify 'sad' key to 'sadness'
     video_emotions['sadness'] = video_emotions.pop('sad')
+
+
     
     final_result_dict = {
         "gaze_percentage" : gaze_percentage,
@@ -180,6 +182,17 @@ def video_processing(video_file, encoded_video):
         "text_sentiment" : text_sentiment
     }
     
+    for item in final_result_dict['data']:
+        emotions = item['face_emotion']
+        top_emotions = dict(sorted(emotions.items(), key=lambda x: x[1], reverse=True)[:3])
+        item['face_emotion'] = top_emotions
+
+    # Update the text_emotion data
+    for item in final_result_dict['data']:
+        emotions = item['text_emotion']
+        top_emotions = sorted(emotions, key=lambda x: x['score'], reverse=True)[:3]
+        item['text_emotion'] = top_emotions
+    
     return final_result_dict