Update app.py

app.py

@@ -572,21 +572,16 @@ with gr.Blocks(css="""
             with gr.Column():
                 data_percentage_slider = gr.Slider(label="Data Percentage for Training", minimum=10, maximum=100, step=10, value=10)
                 task_complexity_dropdown = gr.Dropdown(label="Task Complexity (Number of Beams)", choices=[16, 32, 64, 128, 256], value=16)
+                theme_dropdown = gr.Dropdown(label="Select Theme", choices=['Light', 'Dark'], value='Light')
 
         with gr.Row():
             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)
-
-        # Use the theme detection from JavaScript
-        theme_output = gr.Textbox(label="Theme", visible=False)
-        # Update the theme on button click (or any event)
-        theme_button = gr.Button("Detect Theme")
-        # Trigger JS to detect theme and set value to theme_output
-        theme_button.click(fn=None, inputs=[], outputs=[theme_output], _js=js_code)
     
         # Update the confusion matrices whenever sliders change
-        data_percentage_slider.change(fn=beam_prediction_task, inputs=[data_percentage_slider, task_complexity_dropdown, theme_output], outputs=[raw_img_bp, embeddings_img_bp])
-        task_complexity_dropdown.change(fn=beam_prediction_task, inputs=[data_percentage_slider, task_complexity_dropdown, theme_output], outputs=[raw_img_bp, embeddings_img_bp])
+        data_percentage_slider.change(fn=beam_prediction_task, inputs=[data_percentage_slider, task_complexity_dropdown, theme_dropdown], outputs=[raw_img_bp, embeddings_img_bp])
+        task_complexity_dropdown.change(fn=beam_prediction_task, inputs=[data_percentage_slider, task_complexity_dropdown, theme_dropdown], outputs=[raw_img_bp, embeddings_img_bp])
+        theme_dropdown.change(fn=beam_prediction_task, inputs=[data_percentage_slider, task_complexity_dropdown, theme_dropdown], outputs=[raw_img_bp, embeddings_img_bp])
         
         # Add a conclusion section at the bottom
         gr.Markdown("""