Update app.py

app.py

@@ -281,6 +281,7 @@ def display_dishes_in_grid(dishes, cols=3):
                 st.sidebar.write(dish.replace("_", " ").capitalize())
 
 def display_prediction_graph(class_names, confidences):
+    theme_mode = st.get_option("theme.base")
     #reversing them so graph displays highest predictions at the top
     confidences.reverse()
     class_names.reverse()
@@ -303,7 +304,7 @@ def display_prediction_graph(class_names, confidences):
         if width < min_width:
             width = min_width
         ax.text(width - 0.02, bar.get_y() + bar.get_height()/2, f'{original_width:.1f}%', 
-                va='center', ha='right', color='white', fontweight='bold', fontsize=16)
+                va='center', ha='right', color='white' if theme_mode == "dark" else "black", fontweight='bold', fontsize=16)
 
     ax.set_xticklabels([]) #remove x label
 
@@ -311,13 +312,13 @@ def display_prediction_graph(class_names, confidences):
     max_label_width = 10
     labels = ax.get_yticklabels()
     wrapped_labels = [textwrap.fill(label.get_text(), width=max_label_width) for label in labels] # Wrap the labels if they exceed the max width
-    ax.set_yticklabels(wrapped_labels, fontsize=16, color='white')
+    ax.set_yticklabels(wrapped_labels, fontsize=16, color='white' if theme_mode == "dark" else "black")
     
     #no borders
     for spine in ax.spines.values():
         spine.set_visible(False)
 
-    ax.set_title(class_names[-1], color='white', fontsize=24, fontweight='bold', ha='left', x=0.5)
+    ax.set_title(class_names[-1], color='white' if theme_mode == "dark" else "black", fontsize=24, fontweight='bold', ha='left', x=0.5)
     
     st.pyplot(fig) # Display the plot