Update app.py

app.py

@@ -95,17 +95,15 @@ keys_to_organ_kind = {
 
 
 def update_organ_and_kind(selected_key):
-    """根据选择的 Keys 更新 Organ 和 Kind"""
     organs = keys_to_organ_kind[selected_key]["organs"]
-    default_organ = organs[0]  # 默认选择第一个 organ
-    kinds = keys_to_organ_kind[selected_key]["kinds"][default_organ]
-    return gr.update(choices=organs, value=default_organ), gr.update(choices=kinds, value=kinds[:1])  # 只选第一个
+    first_organ = organs[0]  # 默认选第一个 organ
+    kinds = keys_to_organ_kind[selected_key]["kinds"][first_organ]
+    return gr.update(choices=organs, value=first_organ), gr.update(choices=kinds, value=kinds)
 
 
 def update_kind(selected_key, selected_organ):
-    """根据选择的 Organ 更新 Kind"""
     kinds = keys_to_organ_kind[selected_key]["kinds"][selected_organ]
-    return gr.update(choices=kinds, value=kinds[:1])  # 选第一个 kind
+    return gr.update(choices=kinds, value=kinds)
 
 
 def generate_image(organ, kinds, keys):
@@ -134,19 +132,19 @@ with gr.Blocks() as demo:
     output_image = gr.Image(label="Visualization")
     generate_button.click(generate_image, inputs=[organ_dropdown, kind_checkbox, keys_dropdown], outputs=output_image)
 
-    # 使用 Examples 时更新组织和类型
-    gr.Examples(
-        examples=[
-            ["CVC-ClinicDB", "polyp colonoscopy", ["polyp"]],
-            ["BUSI", "breast ultrasound", ["breast tumor"]],
-            ["LiTS2017", "abdomen CT scans", ["liver", "liver tumor"]],
-            ["KiTS2019", "abdomen CT scans", ["kidney", "kidney tumor"]],
-            ["ACDC", "cardiovascular ventricle mri", ["right ventricle", "myocardium", "left ventricle"]],
-            ["AMOS2022", "abdomen CT scans", ["liver", "right kidney", "spleen"]]
-        ],
-        inputs=[keys_dropdown, organ_dropdown, kind_checkbox],
-        outputs=[organ_dropdown, kind_checkbox],
-        fn=update_organ_and_kind,  # 通过 fn 更新 organ 和 kind
-    )
+    # # 使用 Examples 时更新组织和类型
+    # gr.Examples(
+    #     examples=[
+    #         ["CVC-ClinicDB", "polyp colonoscopy", ["polyp"]],
+    #         ["BUSI", "breast ultrasound", ["breast tumor"]],
+    #         ["LiTS2017", "abdomen CT scans", ["liver", "liver tumor"]],
+    #         ["KiTS2019", "abdomen CT scans", ["kidney", "kidney tumor"]],
+    #         ["ACDC", "cardiovascular ventricle mri", ["right ventricle", "myocardium", "left ventricle"]],
+    #         ["AMOS2022", "abdomen CT scans", ["liver", "right kidney", "spleen"]]
+    #     ],
+    #     inputs=[keys_dropdown, organ_dropdown, kind_checkbox],
+    #     outputs=[organ_dropdown, kind_checkbox],
+    #     fn=update_organ_and_kind,  # 通过 fn 更新 organ 和 kind
+    # )
 
 demo.launch()