Update app.py

app.py

@@ -74,17 +74,32 @@ if st.button("Predict Emotions") and selected_images:
     # Calculate emotion statistics
     emotion_counts = pd.Series(emotions).value_counts()
 
+    # Define a color map that matches the emotions to specific colors
+    color_map = {
+        'Happy': '#ff9999',
+        'Sad': '#66b3ff',
+        'Angry': '#99ff99',
+        'Surprised': '#ffcc99'
+        # Add more emotions and their corresponding colors here
+    }
+
+    # 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
     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=['#ff9999','#66b3ff','#99ff99','#ffcc99'])
+    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.
     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
     st.write("Emotion Distribution (Bar Chart):")
     fig_bar, ax_bar = plt.subplots()
-    emotion_counts.plot(kind='bar', color='skyblue', ax=ax_bar)
+    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')