Update app.py

app.py

@@ -54,21 +54,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_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)
 
-    # 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)