Update app.py

app.py

@@ -8,15 +8,7 @@ import pandas as pd
 from PIL import Image
 from huggingface_hub import login
 
-# 假设 translator.py 中的 translate_texts 函数已正确定义
-# from translator import translate_texts
-# Mock translator for a standalone example if translator.py is not available
-def translate_texts(texts, src_lang="auto", tgt_lang="zh"):
-    print(f"Mock translating: {texts} from {src_lang} to {tgt_lang}")
-    if not texts:
-        return []
-    # 返回一个简单的模拟翻译结果，实际使用时请确保 translator.py 可用且功能正确
-    return [f"{text}_译" for text in texts]
+from translator import translate_texts
 
 # ------------------------------------------------------------------
 # 模型配置
@@ -64,7 +56,6 @@ class Tagger:
             print("✅ 模型和标签加载成功")
         except Exception as e:
             print(f"❌ 模型或标签加载失败: {e}")
-            # 可以选择抛出异常或设置一个标志，让应用知道模型未就绪
             raise RuntimeError(f"模型初始化失败: {e}")
 
 
@@ -263,9 +254,6 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=cus
             return "<p>暂无标签</p>"
         
         html = '<div class="label-container">'
-        # Ensure translations_list is a list and matches length, or provide empty strings if not.
-        # This assumes translations_list corresponds to the order in tags_dict.keys()
-        # For dictionaries, keys() order is insertion order from Python 3.7+
         
         if not isinstance(translations_list, list): # defensive check
             translations_list = []
@@ -369,11 +357,6 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=cus
             # The predict method now returns res and tag_categories_for_translation
             res, tag_categories_original_order = tagger_instance.predict(img, g_th, c_th)
 
-            # 2. Translate all tags that will be displayed in lists
-            # The `show_zh_in_list_checkbox` now controls if we translate for lists.
-            # For summary, translation is controlled by `s_zh_in_sum`.
-            # We should always translate all potential tags to have them ready.
-            
             all_tags_to_translate = []
             for cat_key in ["general", "characters", "ratings"]:
                 all_tags_to_translate.extend(tag_categories_original_order.get(cat_key, []))
@@ -392,17 +375,11 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=cus
                     offset += num_tags_in_cat
                 else:
                     current_translations_dict[cat_key] = []
-
-
-            # 3. Format HTML outputs (always show English, translations if available and `show_zh_in_list` is true)
-            # Let's assume `show_zh_in_list` is a new checkbox or fixed to true for list display.
-            # For simplicity, let's assume list translations are always prepared if `current_translations_dict` has them.
             
             general_html = format_tags_html(res.get("general", {}), current_translations_dict.get("general", []), "general", s_scores, True)
             char_html = format_tags_html(res.get("characters", {}), current_translations_dict.get("characters", []), "characters", s_scores, True)
             rating_html = format_tags_html(res.get("ratings", {}), current_translations_dict.get("ratings", []), "ratings", s_scores, True)
 
-            # 4. Generate initial summary text (based on current summary settings from UI)
             summary_text = generate_summary_text_content(
                 res, current_translations_dict,
                 s_gen, s_char, s_rat, s_sep, s_zh_in_sum # Use summary specific checkbox for zh
@@ -472,15 +449,6 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=cus
             # show_progress=False # Typically fast, no need for progress indicator
         )
     
-    # If tag score display in lists is changed, re-render HTMLs
-    # This requires storing the raw data or re-processing parts of it.
-    # For simplicity, we can make the list HTML generation also dependent on state if needed,
-    # or re-trigger a lighter version of 'process' that only updates HTML.
-    # Current implementation: score display is set at 'analyze' time.
-    # To make 'show_tag_scores' dynamic for lists *after* analysis without re-analyzing:
-    # We would need a new callback that re-runs `format_tags_html` for each category
-    # using data from `state_res` and `state_translations_dict`.
-
 # ------------------------------------------------------------------
 # 启动
 # ------------------------------------------------------------------