Update app.py

app.py

@@ -45,21 +45,21 @@ if uploaded_image:
         'Surprise': '#FFA500',
         'Fear': '#000000'
     }
+    pie_colors = [color_map.get(emotion, '#999999') for emotion in emotion_counts.index]  
+    # Pie chart visualization
+    fig_pie, ax_pie = plt.subplots(figsize=(4, 4))
+    ax_pie.pie(emotion_counts, labels=emotion_counts.index, autopct='%1.1f%%', startangle=140, colors=pie_colors)
+    ax_pie.axis('equal')
+    ax_pie.set_title("Emotion Distribution")
+    st.pyplot(fig_pie)
 
-    # # Pie chart visualization
-    # fig_pie, ax_pie = plt.subplots(figsize=(4, 4))
-    # ax_pie.pie(emotion_counts, labels=emotion_counts.index, autopct='%1.1f%%', startangle=140, colors=pie_colors)
-    # ax_pie.axis('equal')
-    # ax_pie.set_title("Emotion Distribution")
-    # st.pyplot(fig_pie)
-
-    # # Bar chart visualization
-    # fig_bar, ax_bar = plt.subplots(figsize=(4, 4))
-    # emotion_counts.plot(kind='bar', color=pie_colors, ax=ax_bar)
-    # ax_bar.set_xlabel('Emotion')
-    # ax_bar.set_ylabel('Count')
-    # ax_bar.set_title("Emotion Count")
-    # ax_bar.yaxis.set_major_locator(plt.MaxNLocator(integer=True))
-    # for i in ax_bar.patches:
-    #     ax_bar.text(i.get_x() + i.get_width() / 2, i.get_height() + 0.1, str(int(i.get_height())), ha='center', va='bottom')
-    # st.pyplot(fig_bar)
+    # Bar chart visualization
+    fig_bar, ax_bar = plt.subplots(figsize=(4, 4))
+    emotion_counts.plot(kind='bar', color=pie_colors, ax=ax_bar)
+    ax_bar.set_xlabel('Emotion')
+    ax_bar.set_ylabel('Count')
+    ax_bar.set_title("Emotion Count")
+    ax_bar.yaxis.set_major_locator(plt.MaxNLocator(integer=True))
+    for i in ax_bar.patches:
+        ax_bar.text(i.get_x() + i.get_width() / 2, i.get_height() + 0.1, str(int(i.get_height())), ha='center', va='bottom')
+    st.pyplot(fig_bar)