Update app.py

app.py

@@ -125,14 +125,23 @@ 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
+     }, 
     
-    angry = 'total anger percentage' + str(angry) 
-    disgust = 'total disgust percentage' + str(disgust) 
-    fear = 'total fear percentage' + str(fear) 
-    happy = 'total happy percentage' + str(happy) 
-    sad = 'total sad percentage' + str(sad) 
-    surprise = 'total surprise percentage' + str(surprise)
-    neutral = 'total neutral percentage' + str(neutral) 
+    # angry = 'total anger percentage' + str(angry) 
+    # disgust = 'total disgust percentage' + str(disgust) 
+    # fear = 'total fear percentage' + str(fear) 
+    # happy = 'total happy percentage' + str(happy) 
+    # sad = 'total sad percentage' + str(sad) 
+    # surprise = 'total surprise percentage' + str(surprise)
+    # neutral = 'total neutral percentage' + str(neutral) 
     print(f'total anger percentage = {angry}')
     print(f'total disgust percentage = {disgust}')
     print(f'total fear percentage = {fear}')
@@ -140,7 +149,7 @@ def video_processing(video_file, encoded_video):
     print(f'total sad percentage = {sad}')
     print(f'total surprise percentage = {surprise}')
     print(f'total neutral percentage = {neutral}')
-    return str(gaze_percentage)
+    return str(gaze_percentage,emotion)
 
 
 demo = gr.Interface(fn=video_processing,