Update app.py

app.py

@@ -646,7 +646,11 @@ with gr.Blocks(css="""
 
         # File uploader for dataset (only visible if user chooses to upload a dataset)
         file_input = gr.File(label="Upload HDF5 Dataset", file_types=[".h5"], visible=False)
-
+        # Dropdown for embedding type, also only visible when "Upload Dataset" is selected
+        emb_type = gr.Dropdown(choices=["Channel Embedding", "CLS Embedding"], 
+                               value="CLS Embedding", 
+                               label="Embedding Type", 
+                               visible=False)
         # Confusion matrices display
         with gr.Row():
             raw_img_los = gr.Image(label="Raw Channels", type="pil", width=300, height=300)
@@ -658,10 +662,20 @@ with gr.Blocks(css="""
                            label="Embedding Type", interactive=True)
             
         # Update the file uploader visibility based on user choice
-        def toggle_file_input(choice):
-            return gr.update(visible=(choice == "Upload Dataset"))
+        #def toggle_file_input(choice):
+        #    return gr.update(visible=(choice == "Upload Dataset"))
+        
+        #choice_radio.change(fn=toggle_file_input, inputs=[choice_radio], outputs=file_input)
+
+        # Update the visibility of file_input and emb_type based on user choice
+        def toggle_file_input_and_emb_type(choice):
+            visible = (choice == "Upload Dataset")
+            return gr.update(visible=visible), gr.update(visible=visible)
         
-        choice_radio.change(fn=toggle_file_input, inputs=[choice_radio], outputs=file_input)
+        # Change visibility of file input and embedding type dropdown based on choice
+        choice_radio.change(fn=toggle_file_input_and_emb_type, 
+                            inputs=[choice_radio], 
+                            outputs=[file_input, emb_type])
 
         # When user makes a choice, update the display
         choice_radio.change(fn=handle_user_choice, inputs=[choice_radio, percentage_slider_los, file_input, emb_type], 
@@ -672,8 +686,8 @@ with gr.Blocks(css="""
                                        outputs=[raw_img_los, embeddings_img_los, output_textbox])
 
         # When embedding type changes
-        emb_type.change(fn=handle_user_choice, inputs=[choice_radio, percentage_slider_los, file_input, emb_type], 
-                        outputs=[raw_img_los, embeddings_img_los, output_textbox])
+        #emb_type.change(fn=handle_user_choice, inputs=[choice_radio, percentage_slider_los, file_input, emb_type], 
+        #                outputs=[raw_img_los, embeddings_img_los, output_textbox])
 
         # Add a conclusion section at the bottom
         gr.Markdown("""