Update app.py

app.py

@@ -85,6 +85,7 @@ class Tagger:
             tag_categories_for_translation[res_key] = list(res[res_key].keys())
 
         return res, tag_categories_for_translation
+
 try:
     tagger_instance = Tagger()
 except RuntimeError as e:
@@ -156,7 +157,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=cus
                 baidu_json_in = gr.Textbox(label="百度翻译凭证 (JSON 格式)", lines=3, placeholder='[{"app_id": "...", "secret_key": "..."}]')
 
             with gr.Accordion("📊 标签汇总设置", open=True):
-                gr.CheckboxGroup(["通用标签", "角色标签", "评分标签"], value=["通用标签", "角色标签"], label="汇总类别")
+                sum_cats = gr.CheckboxGroup(["通用标签", "角色标签", "评分标签"], value=["通用标签", "角色标签"], label="汇总类别")
                 sum_sep = gr.Dropdown(["逗号", "换行", "空格"], value="逗号", label="标签分隔符")
                 sum_show_zh = gr.Checkbox(False, label="在汇总中显示中文翻译")
 
@@ -204,8 +205,8 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=cus
         parts, sep = [], {"逗号": ", ", "换行": "\n", "空格": " "}.get(sep_type, ", ")
         cat_map = {"通用标签": "general", "角色标签": "characters", "评分标签": "ratings"}
         for cat_name in sum_cats:
-            cat_key = cat_map[cat_name]
-            if current_res.get(cat_key):
+            cat_key = cat_map.get(cat_name)
+            if cat_key and current_res.get(cat_key):
                 tags_en, trans = list(current_res[cat_key].keys()), translations.get(cat_key, [])
                 tags_to_join = [f"{en}({zh})" if show_zh and i < len(trans) and trans[i] else en for i, en in enumerate(tags_en)]
                 if tags_to_join: parts.append(sep.join(tags_to_join))
@@ -278,7 +279,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=cus
         inputs=[
             img_in, gen_slider, char_slider, show_tag_scores,
             tencent_id_in, tencent_key_in, baidu_json_in,
-            sum_general, sum_sep, sum_show_zh
+            sum_cats, sum_sep, sum_show_zh
         ],
         outputs=[
             btn, processing_info,
@@ -288,7 +289,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=cus
         ],
     )
 
-    summary_controls = [sum_general, sum_sep, sum_show_zh]
+    summary_controls = [sum_cats, sum_sep, sum_show_zh]
     for ctrl in summary_controls:
         ctrl.change(
             fn=lambda r, t, c, s, z: generate_summary_text_content(r, t, c, s, z),