Update app.py

app.py

@@ -24,7 +24,11 @@ def beam_prediction_task(data_percentage, task_complexity, theme):
     raw_cm = compute_average_confusion_matrix(raw_folder)
     if raw_cm is not None:
         raw_cm_path = os.path.join(raw_folder, "confusion_matrix_raw.png")
-        plot_confusion_matrix_beamPred(raw_cm, classes=np.arange(raw_cm.shape[0]), title=f"Raw Confusion Matrix\n({data_percentage}% data, {task_complexity} beams)", save_path=raw_cm_path, theme=theme)
+        plot_confusion_matrix_beamPred(raw_cm, 
+                                       classes=np.arange(raw_cm.shape[0]), 
+                                       title=f"Raw Confusion Matrix\n({data_percentage}% data, {task_complexity} beams)", 
+                                       save_path=raw_cm_path, 
+                                       theme=theme)
         raw_img = Image.open(raw_cm_path)
     else:
         raw_img = None
@@ -33,7 +37,11 @@ def beam_prediction_task(data_percentage, task_complexity, theme):
     embeddings_cm = compute_average_confusion_matrix(embeddings_folder)
     if embeddings_cm is not None:
         embeddings_cm_path = os.path.join(embeddings_folder, "confusion_matrix_embeddings.png")
-        plot_confusion_matrix_beamPred(embeddings_cm, classes=np.arange(embeddings_cm.shape[0]), title=f"Embeddings Confusion Matrix\n({data_percentage}% data, {task_complexity} beams)", save_path=embeddings_cm_path, theme=theme)
+        plot_confusion_matrix_beamPred(embeddings_cm, 
+                                       classes=np.arange(embeddings_cm.shape[0]), 
+                                       title=f"Embeddings Confusion Matrix\n({data_percentage}% data, {task_complexity} beams)", 
+                                       save_path=embeddings_cm_path, 
+                                       theme=theme)
         embeddings_img = Image.open(embeddings_cm_path)
     else:
         embeddings_img = None
@@ -474,20 +482,24 @@ def process_hdf5_file(uploaded_file, percentage):
         sys.stdout = sys.__stdout__  # Reset print statements
 
 ######################## Define the Gradio interface ###############################
-# JavaScript to detect dark/light mode and pass it to Gradio as 'light' or 'dark'
-detect_theme_js = """
 <script>
-    function detectTheme() {
-        if (window.matchMedia && window.matchMedia('(prefers-color-scheme: dark)').matches) {
-            return 'dark';
-        } else {
-            return 'light';
-        }
+function detectTheme() {
+    if (window.matchMedia && window.matchMedia('(prefers-color-scheme: dark)').matches) {
+        return 'dark';
+    } else {
+        return 'light';
     }
-    document.getElementById('theme_output').value = detectTheme();
-    document.getElementById('theme_output').dispatchEvent(new Event('change'));
+}
+
+window.onload = function() {
+    const theme = detectTheme();
+    const themeTextbox = document.getElementById('theme_output');
+    if (themeTextbox) {
+        themeTextbox.value = theme;
+        themeTextbox.dispatchEvent(new Event('change')); // Manually trigger change event
+    }
+};
 </script>
-"""
 
 with gr.Blocks(css="""
     .slider-container {
@@ -552,7 +564,7 @@ with gr.Blocks(css="""
         gr.HTML(detect_theme_js)
     
         # Hidden textbox to store the detected theme
-        theme_output = gr.Textbox(label="Detected Theme", elem_id="theme_output", visible=False)
+        theme_output = gr.Textbox(label="Theme", value="light", visible=False, interactive=False)  # Hidden by default
 
         # Explanation section with creative spacing and minimal design
         gr.Markdown("""