Update app.py

app.py

@@ -74,11 +74,13 @@ def plot_confusion_matrix_beamPred(cm, classes, title, save_path, theme=None):
     # Choose the color scheme based on the user's mode
     if theme == 'Dark':
         plt.style.use('dark_background')  # Use dark mode styling
-        text_color = 'white'
+        #text_color = 'white'
+        text_color = 'gray'
         cmap = 'cividis'  # Dark-mode-friendly colormap
     else:
         plt.style.use('default')  # Use default (light) mode styling
-        text_color = 'black'
+        #text_color = 'black'
+        text_color = 'gray'
         cmap = 'Blues'  # Light-mode-friendly colormap
 
     plt.figure(figsize=(10, 10))