refine page

app.py

@@ -127,11 +127,11 @@ if file_name is not None:
         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)}
 
-        bar_fig, bar_ax = plt.subplots()
-        bar_ax.bar(label_counts_refined.keys(), label_counts_refined.values())
-        bar_ax.set_title('Distribution of Emotions')
-        bar_ax.set_xlabel('Emotions')
-        bar_ax.set_ylabel('Count')
+        bar_fig_actions, bar_ax_actions = plt.subplots()
+        bar_ax_actions.bar(label_counts_refined.keys(), label_counts_refined.values())
+        bar_ax_actions.set_title('Distribution of Actions')
+        bar_ax_actions.set_xlabel('Emotions')
+        bar_ax_actions.set_ylabel('Count')
 
         labels_emotions = [item['label'] for item in output_list_emotions]
         label_counts_emotions = {label: labels_emotions.count(label) for label in set(labels_emotions)}
@@ -147,10 +147,12 @@ if file_name is not None:
 
         st.pyplot(fig)  # Display the stitched person images
 
-        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
+        col1, col2 = st.columns(2)
+        col1.pyplot(pie_fig_emotions)  # Display the pie chart
+        col2.pyplot(bar_fig_emotions)  # Display the bar chart
+
+        col1.pyplot(pie_fig_actions)  # Display the pie chart
+        col2.pyplot(bar_fig_actions)  # Display the bar chart
 
     elif file_name.type.startswith('video'):
         # Save the uploaded video to a temporary file