Update app.py

app.py

@@ -95,19 +95,22 @@ keys_to_organ_kind = {
 
 
 def update_organ_and_kind(selected_key):
+    """更新 organ 和 kind 的选项，并确保 organ 正确更新"""
     organs = keys_to_organ_kind[selected_key]["organs"]
-    first_organ = organs[0]  # 默认选第一个 organ
-    kinds = keys_to_organ_kind[selected_key]["kinds"][first_organ]
+    first_organ = organs[0] if organs else ""  # 选择第一个 organ
+    kinds = keys_to_organ_kind[selected_key]["kinds"].get(first_organ, [])  # 确保 kinds 不为空
     return gr.update(choices=organs, value=first_organ), gr.update(choices=kinds, value=kinds)
 
 
 def update_kind(selected_key, selected_organ):
-    kinds = keys_to_organ_kind[selected_key]["kinds"][selected_organ]
+    """更新 kind 的选项"""
+    kinds = keys_to_organ_kind[selected_key]["kinds"].get(selected_organ, [])
     return gr.update(choices=kinds, value=kinds)
 
 
 def generate_image(organ, kinds, keys):
     kind = ",".join(kinds)
+    print(f"Debug Info -> Organ: {organ}, Kind: {kind}, Keys: {keys}")
     image, label, image_prompt, mask_prompt = pipeline.generate(organ=organ, kind=kind, keys=keys)
     plt.subplot(1, 2, 1)
     plt.imshow(image)