Update app.py

app.py

@@ -59,7 +59,7 @@ def compute_f1_score(cm):
     f1 = np.nan_to_num(f1)  # Replace NaN with 0
     return np.mean(f1)  # Return the mean F1-score across all classes
 
-def plot_confusion_matrix_beamPred(cm, classes, title, save_path, user_mode="light"):
+def plot_confusion_matrix_beamPred(cm, classes, title, save_path, user_mode="dark"):
     # Compute the average F1-score
     avg_f1 = compute_f1_score(cm)
 
@@ -559,21 +559,12 @@ with gr.Blocks(css="""
                 task_complexity_dropdown = gr.Dropdown(label="Task Complexity (Number of Beams)", choices=[16, 32, 64, 128, 256], value=16)
 
         with gr.Row():
-            
-            mode_input = gr.Textbox(visible=False)  # Hidden input to capture user mode
-            gr.Markdown("""
-            <script>
-            const userPrefersDark = window.matchMedia && window.matchMedia('(prefers-color-scheme: dark)').matches;
-            document.querySelector('input[name="mode_input"]').value = userPrefersDark ? 'dark' : 'light';
-            </script>
-            """)
-            
             raw_img_bp = gr.Image(label="Raw Channels", type="pil", width=300, height=500)
             embeddings_img_bp = gr.Image(label="Embeddings", type="pil", width=300, height=500)
 
         # Update the confusion matrices whenever sliders change
-        data_percentage_slider.change(fn=beam_prediction_task, inputs=[data_percentage_slider, task_complexity_dropdown, mode_input], outputs=[raw_img_bp, embeddings_img_bp])
-        task_complexity_dropdown.change(fn=beam_prediction_task, inputs=[data_percentage_slider, task_complexity_dropdown, mode_input], outputs=[raw_img_bp, embeddings_img_bp])
+        data_percentage_slider.change(fn=beam_prediction_task, inputs=[data_percentage_slider, task_complexity_dropdown], outputs=[raw_img_bp, embeddings_img_bp])
+        task_complexity_dropdown.change(fn=beam_prediction_task, inputs=[data_percentage_slider, task_complexity_dropdown], outputs=[raw_img_bp, embeddings_img_bp])
 
         # Add a conclusion section at the bottom
         gr.Markdown("""