add actions pie fig

app.py

@@ -90,29 +90,39 @@ if file_name is not None:
 
         
         # 统计各种标签的数量
-        label_counts = {}
+        label_counts_emotions = {}
+        label_counts_actions = {}
         for item in output_list_emotions:
             label = item['label']
-            if label in label_counts:
-                label_counts[label] += 1
+            if label in label_counts_emotions:
+                label_counts_emotions[label] += 1
             else:
-                label_counts[label] = 1
+                label_counts_emotions[label] = 1
+
         for item in output_list_emotions_refined:
             label = item['label']
-            if label in label_counts:
-                label_counts[label] += 1
+            if label in label_counts_actions:
+                label_counts_actions[label] += 1
             else:
-                label_counts[label] = 1
+                label_counts_actions[label] = 1
 
         # 绘制饼状图
-        labels = list(label_counts.keys())
-        sizes = list(label_counts.values())
+        labels_emotions = list(label_counts_emotions.keys())
+        sizes_emotions = list(label_counts_emotions.values())
         
-        pie_fig, pie_ax = plt.subplots()
-        pie_ax.pie(sizes, labels=labels, autopct='%1.1f%%', startangle=140)
-        pie_ax.set_title('Distribution of Emotions')
-        pie_ax.axis('equal')  # 确保饼状图为圆形
+        pie_fig_emotions, pie_ax_emotions = plt.subplots()
+        pie_ax_emotions.pie(sizes_emotions, labels=labels_emotions, autopct='%1.1f%%', startangle=140)
+        pie_ax_emotions.set_title('Distribution of Emotions')
+        pie_ax_emotions.axis('equal')  # 确保饼状图为圆形
 
+        labels_actions = list(label_counts_actions.keys())
+        sizes_actions = list(label_counts_actions.values())
+        
+        pie_fig_actions, pie_ax_actions = plt.subplots()
+        pie_ax_actions.pie(sizes_actions, labels=labels_actions, autopct='%1.1f%%', startangle=140)
+        pie_ax_actions.set_title('Distribution of Actions')
+        pie_ax_actions.axis('equal')  # 确保饼状图为圆形
+        
 
         labels_refined = [item['label'] for item in output_list_emotions_refined]
         label_counts_refined = {label: labels_refined.count(label) for label in set(labels_refined)}
@@ -137,10 +147,9 @@ if file_name is not None:
 
         st.pyplot(fig)  # Display the stitched person images
 
-        st.pyplot(pie_fig)  # Display the pie chart
-
+        st.pyplot(pie_fig_emotions)  # Display the pie chart
+        st.pyplot(pie_fig_actions)  # Display the pie chart
         st.pyplot(bar_fig)  # Display the bar chart
-
         st.pyplot(bar_fig_emotions)  # Display the bar chart
 
     elif file_name.type.startswith('video'):