Update app.py

app.py

@@ -86,31 +86,31 @@ if st.button("Predict Emotions") and selected_images:
         # Add more emotions and their corresponding colors here
     }
 
-    #https://link.springer.com/article/10.3758/s13428-015-0598-8#:~:text=The%20color%20red%20was%20most%20associated%20with%20anger%2C%20green%20with,%E2%80%9D%20or%20%E2%80%9Cfeeling%20blue.%E2%80%9D
+    # Calculate the total number of faces analyzed
+    total_faces = len(selected_images)
 
-    #The color red was most associated with anger, green with disgust, black with fear, yellow with happiness, blue with sadness, 
-    #and bright with surprise. These associations may be a result of various expressions containing color terms that are used in the 
-    #English language—for example, “seeing red” or “feeling blue.”May 19, 2015
-    
     # Use the color map to assign colors to the pie chart
     pie_colors = [color_map.get(emotion, '#999999') for emotion in emotion_counts.index]  # Default to grey if not found
 
-    # Plot pie chart
+    # Plot pie chart with total faces in the title
     st.write("Emotion Distribution (Pie Chart):")
     fig_pie, ax_pie = plt.subplots()
     ax_pie.pie(emotion_counts, labels=emotion_counts.index, autopct='%1.1f%%', startangle=140, colors=pie_colors)
     ax_pie.axis('equal')  # Equal aspect ratio ensures that pie is drawn as a circle.
+    # Add total faces to the title
+    ax_pie.set_title(f"Total Faces Analyzed: {total_faces}")
     st.pyplot(fig_pie)
 
     # Use the same color map for the bar chart
     bar_colors = [color_map.get(emotion, '#999999') for emotion in emotion_counts.index]  # Default to grey if not found
 
-    # Plot bar chart
+    # Plot bar chart with total faces in the title
     st.write("Emotion Distribution (Bar Chart):")
     fig_bar, ax_bar = plt.subplots()
     emotion_counts.plot(kind='bar', color=bar_colors, ax=ax_bar)
     ax_bar.set_xlabel('Emotion')
     ax_bar.set_ylabel('Count')
-    ax_bar.set_title('Emotion Distribution')
+    # Add total faces to the title
+    ax_bar.set_title(f"Emotion Distribution - Total Faces Analyzed: {total_faces}")
     ax_bar.yaxis.set_major_locator(plt.MaxNLocator(integer=True))  # Ensure integer ticks on Y-axis
     st.pyplot(fig_bar)