Update app.py

app.py

@@ -67,6 +67,15 @@ def video_processing(video_file, encoded_video):
     surprise = 0
     neutral = 0
     emotion_count = 0
+    video_emotions = {
+        'angry': 0,
+        'disgust': 0,
+        'fear': 0,
+        'happy': 0,
+        'sad': 0,
+        'surprise': 0,
+        'neutral':0
+    }
 
     if encoded_video != "":
     
@@ -120,7 +129,10 @@ def video_processing(video_file, encoded_video):
                 resized_face_image = cv2.resize(face_image, (128,128))
 
                 try:
-                    emotion = DeepFace.analyze(color_image,actions=['emotion'],detector_backend = backends[2],enforce_detection = False)# 2,3, 4 works
+                    detected_face_emotion = DeepFace.analyze(color_image,actions=['emotion'],detector_backend = backends[2],enforce_detection = False)# 2,3, 4 works
+                    for emotion in detected_face_emotion:
+                        for key in video_emotions.keys():
+                            video_emotions[key] += emotion['emotion'][key]
                     emotion_count += 1
                 except Exception as e:
                     emotion = 0
@@ -135,13 +147,13 @@ def video_processing(video_file, encoded_video):
                     off_camera += 1
                 total += 1
                 
-                angry += emotion[0]['emotion']['angry']
-                disgust += emotion[0]['emotion']['disgust']
-                fear += emotion[0]['emotion']['fear']
-                happy += emotion[0]['emotion']['happy']
-                sad += emotion[0]['emotion']['sad']
-                surprise += emotion[0]['emotion']['surprise']
-                neutral += emotion[0]['emotion']['neutral']
+                # angry += emotion[0]['emotion']['angry']
+                # disgust += emotion[0]['emotion']['disgust']
+                # fear += emotion[0]['emotion']['fear']
+                # happy += emotion[0]['emotion']['happy']
+                # sad += emotion[0]['emotion']['sad']
+                # surprise += emotion[0]['emotion']['surprise']
+                # neutral += emotion[0]['emotion']['neutral']
 
     try:
         # your processing code here
@@ -167,18 +179,18 @@ def video_processing(video_file, encoded_video):
     sad = sad / emotion_count
     surprise = surprise / emotion_count
     neutral = neutral / emotion_count
-    emotion = {
-        'angry': angry, 
-        'disgust': disgust, 
-        'fear': fear, 
-        'happy': happy, 
-        'sad': sad, 
-        'surprise': surprise, 
-        'neutral': neutral
-     },
+    # emotion = {
+    #     'angry': angry, 
+    #     'disgust': disgust, 
+    #     'fear': fear, 
+    #     'happy': happy, 
+    #     'sad': sad, 
+    #     'surprise': surprise, 
+    #     'neutral': neutral
+    #  },
     final_result_dict = {
         "gaze_percentage" : gaze_percentage,
-        "face_emotion" : emotion,
+        "face_emotion" : video_emotions,
         "text_emotion" : text_emotion,
         "transcription" : transcription,
         "text_sentiment" : text_sentiment