Update app.py

app.py

@@ -6,22 +6,30 @@ import numpy as np
 import onnxruntime as rt
 import pandas as pd
 from PIL import Image
-from huggingface_hub import login, HfApi
-from translator import translate_texts, set_user_provided_keys, clear_user_provided_keys
+from huggingface_hub import login
+
+from translator import translate_texts, translate_texts_with_dynamic_keys
 
 MODEL_REPO = "SmilingWolf/wd-eva02-large-tagger-v3"
 MODEL_FILENAME = "model.onnx"
 LABEL_FILENAME = "selected_tags.csv"
+
+OWNER_USERNAME = os.environ.get("OWNER_USERNAME", "").lower()
+
 HF_TOKEN = os.environ.get("HF_TOKEN", "")
 if HF_TOKEN:
     try:
         login(token=HF_TOKEN)
-        print("✅ 应用已使用 HF_TOKEN 登录")
+        print("✅ HF_TOKEN 登录成功")
     except Exception as e:
-        print(f"⚠️ 使用 HF_TOKEN 登录失败: {e}")
+        print(f"⚠️ HF_TOKEN 登录失败: {e}")
 else:
-    print("⚠️ 未检测到应用级别的 HF_TOKEN，私有模型可能下载失败")
+    print("⚠️ 未检测到 HF_TOKEN，私有模型可能下载失败")
+
 
+# ------------------------------------------------------------------
+# Tagger 类 (全局实例化)
+# ------------------------------------------------------------------
 class Tagger:
     def __init__(self):
         self.hf_token = HF_TOKEN
@@ -122,7 +130,6 @@ custom_css = """
 .tag-zh { color: #666; margin-left: 10px; }
 .tag-score { color: #999; font-size: 0.9em; }
 .btn-analyze-container { margin-top: 15px; margin-bottom: 15px; }
-.user-info { text-align: right; color: #666; font-size: 0.9em; padding: 5px; }
 """
 
 _js_functions = """
@@ -136,243 +143,212 @@ function copyToClipboard(text) {
         let displayText = String(text);
         displayText = displayText.substring(0, 30) + (displayText.length > 30 ? '...' : '');
         feedback.textContent = '已复制: ' + displayText;
-        feedback.style.cssText = 'position:fixed; bottom:20px; left:50%; transform:translateX(-50%); background-color:#4CAF50; color:white; padding:10px 20px; border-radius:5px; z-index:10000; transition:opacity 0.5s ease-out;';
+        feedback.style.cssText = 'position: fixed; bottom: 20px; left: 50%; transform: translateX(-50%); background-color: #4CAF50; color: white; padding: 10px 20px; border-radius: 5px; z-index: 10000; transition: opacity 0.5s ease-out;';
         document.body.appendChild(feedback);
         setTimeout(() => {
             feedback.style.opacity = '0';
-            setTimeout(() => { document.body.removeChild(feedback); }, 500);
+            setTimeout(() => { if (document.body.contains(feedback)) document.body.removeChild(feedback); }, 500);
         }, 1500);
     }).catch(err => {
-        console.error('Failed to copy tag.', err, 'Text:', text);
+        console.error('Failed to copy tag. Error:', err, 'Attempted to copy text:', text);
     });
 }
 """
 
-def main_interface(user_info: gr.UserInfo):
-    with gr.Blocks(theme=gr.themes.Soft(), css=custom_css, js=_js_functions) as demo:
-        gr.Markdown(f"<div class='user-info'>已登录: {user_info.name} ({user_info.email})</div>")
-        gr.Markdown("# 🖼️ AI 图像标签分析器")
-        gr.Markdown("上传图片自动识别标签，支持中英文显示和一键复制。[NovelAI在线绘画](https://nai.idlecloud.cc/)")
-
-        state_res = gr.State({})
-        state_translations_dict = gr.State({})
-        state_tag_categories_for_translation = gr.State({})
-
-        with gr.Row():
-            with gr.Column(scale=1):
-                img_in = gr.Image(type="pil", label="上传图片", height=300)
-                btn = gr.Button("🚀 开始分析", variant="primary", elem_classes=["btn-analyze-container"])
-
-                with gr.Accordion("⚙️ 高级设置", open=False):
-                    gen_slider = gr.Slider(0, 1, value=0.35, step=0.01, label="通用标签阈值", info="越高 → 标签更少更准")
-                    char_slider = gr.Slider(0, 1, value=0.85, step=0.01, label="角色标签阈值", info="推荐保持较高阈值")
-                    show_tag_scores = gr.Checkbox(True, label="在列表中显示标签置信度")
-                    
-                    with gr.Accordion("🔑 翻译服务设置", open=False):
-                        gr.Markdown("如果应用配置了全局翻译密钥，可在此输入访问密钥以使用。否则，请在此处填入您自己的翻译API密钥。")
-                        access_key_input = gr.Textbox(label="访问密钥 (Access Key)", type="password", placeholder="如果需要，请输入访问密钥")
-                        
-                        gr.Markdown("---")
-                        gr.Markdown("**或者**，使用你自己的密钥：")
-                        user_tencent_id = gr.Textbox(label="腾讯云 Secret ID", type="password")
-                        user_tencent_key = gr.Textbox(label="腾讯云 Secret Key", type="password")
-                        user_baidu_json = gr.Textbox(label="百度翻译凭证 (JSON格式)", type="password", lines=3, placeholder='[{"app_id":"...", "secret_key":"..."}]')
-
-                with gr.Accordion("📊 标签汇总设置", open=True):
-                    gr.Markdown("选择要包含在下方汇总文本框中的标签类别：")
-                    with gr.Row():
-                        sum_general = gr.Checkbox(True, label="通用标签", min_width=50)
-                        sum_char = gr.Checkbox(True, label="角色标签", min_width=50)
-                        sum_rating = gr.Checkbox(False, label="评分标签", min_width=50)
-                    sum_sep = gr.Dropdown(["逗号", "换行", "空格"], value="逗号", label="标签之间的分隔符")
-                    sum_show_zh = gr.Checkbox(False, label="在汇总中显示中文翻译")
-
-                processing_info = gr.Markdown("", visible=False)
-
-            with gr.Column(scale=2):
-                with gr.Tabs():
-                    with gr.TabItem("🏷️ 通用标签"):
-                        out_general = gr.HTML(label="General Tags")
-                    with gr.TabItem("👤 角色标签"):
-                        gr.Markdown("<p style='color:gray; font-size:small;'>提示：角色标签推测基于截至2024年2月的数据。</p>")
-                        out_char = gr.HTML(label="Character Tags")
-                    with gr.TabItem("⭐ 评分标签"):
-                        out_rating = gr.HTML(label="Rating Tags")
-                
-                gr.Markdown("### 标签汇总结果")
-                out_summary = gr.Textbox(label="标签汇总", placeholder="分析完成后，此处将显示汇总的英文标签...", lines=5, show_copy_button=True)
-
-        def format_tags_html(tags_dict, translations_list, category_name, show_scores=True, show_translation_in_list=True):
-            if not tags_dict: return "<p>暂无标签</p>"
-            html = '<div class="label-container">'
-            if not isinstance(translations_list, list): translations_list = []
-            tag_keys = list(tags_dict.keys())
-            for i, tag in enumerate(tag_keys):
-                score = tags_dict[tag]
-                escaped_tag = tag.replace("'", "\\'")
-                html += '<div class="tag-item">'
-                tag_display_html = f'<span class="tag-en" onclick="copyToClipboard(\'{escaped_tag}\')">{tag}</span>'
-                if show_translation_in_list and i < len(translations_list) and translations_list[i]:
-                    tag_display_html += f'<span class="tag-zh">({translations_list[i]})</span>'
-                html += f'<div>{tag_display_html}</div>'
-                if show_scores: html += f'<span class="tag-score">{score:.3f}</span>'
-                html += '</div>'
-            html += '</div>'
-            return html
-
-        def generate_summary_text_content(current_res, current_translations_dict, s_gen, s_char, s_rat, s_sep_type, s_show_zh):
-            if not current_res: return "请先分析图像或选择要汇总的标签类别。"
-            summary_parts = []
-            separator = {"逗号": ", ", "换行": "\n", "空格": " "}.get(s_sep_type, ", ")
-            categories_to_summarize = []
-            if s_gen: categories_to_summarize.append("general")
-            if s_char: categories_to_summarize.append("characters")
-            if s_rat: categories_to_summarize.append("ratings")
-            if not categories_to_summarize: return "请至少选择一个标签类别进行汇总。"
-            for cat_key in categories_to_summarize:
-                if current_res.get(cat_key):
-                    tags_to_join = []
-                    cat_tags_en = list(current_res[cat_key].keys())
-                    cat_translations = current_translations_dict.get(cat_key, [])
-                    for i, en_tag in enumerate(cat_tags_en):
-                        if s_show_zh and i < len(cat_translations) and cat_translations[i]:
-                            tags_to_join.append(f"{en_tag}({cat_translations[i]})")
-                        else:
-                            tags_to_join.append(en_tag)
-                    if tags_to_join: summary_parts.append(separator.join(tags_to_join))
-            joiner = "\n\n" if separator != "\n" and len(summary_parts) > 1 else separator if separator == "\n" else " "
-            final_summary = joiner.join(summary_parts)
-            return final_summary if final_summary else "选定的类别中没有找到标签。"
-
-        def process_image_and_generate_outputs(
-            img, g_th, c_th, s_scores, 
-            s_gen, s_char, s_rat, s_sep, s_zh_in_sum,
-            access_key, u_tencent_id, u_tencent_key, u_baidu_json,
-            request: gr.Request
-        ):
-            if img is None:
-                yield (gr.update(interactive=True, value="🚀 开始分析"), gr.update(visible=True, value="❌ 请先上传图片。"), "", "", "", "", gr.update(placeholder="请先上传图片并开始分析..."), {}, {}, {})
-                return
-            
-            if tagger_instance is None:
-                yield (gr.update(interactive=True, value="🚀 开始分析"), gr.update(visible=True, value="❌ ���析器未成功初始化，请检查控制台错误。"), "", "", "", "", gr.update(placeholder="分析器初始化失败..."), {}, {}, {})
-                return
+with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=custom_css, js=_js_functions) as demo:
+    with gr.Row():
+        login_button = gr.LoginButton()
+        user_info = gr.Markdown("请先登录...", visible=True)
+
+    gr.Markdown("# 🖼️ AI 图像标签分析器")
+    gr.Markdown("上传图片自动识别标签，支持中英文显示和一键复制。[NovelAI在线绘画](https://nai.idlecloud.cc/)")
+
+    state_res = gr.State({})
+    state_translations_dict = gr.State({})
+    state_tag_categories_for_translation = gr.State({})
+
 
-            yield (gr.update(interactive=False, value="🔄 处理中..."), gr.update(visible=True, value="🔄 正在分析图像，请稍候..."), gr.HTML(value="<p>分析中...</p>"), gr.HTML(value="<p>分析中...</p>"), gr.HTML(value="<p>分析中...</p>"), gr.update(value="分析中，请稍候..."), {}, {}, {})
+    with gr.Row():
+        with gr.Column(scale=1):
+            img_in = gr.Image(type="pil", label="上传图片", height=300)
+            btn = gr.Button("🚀 开始分析", variant="primary", elem_classes=["btn-analyze-container"])
             
-            try:
-                set_user_provided_keys(u_tencent_id, u_tencent_key, u_baidu_json)
-                
-                res, tag_categories_original_order = tagger_instance.predict(img, g_th, c_th)
-                all_tags_to_translate = [tag for cat in tag_categories_original_order.values() for tag in cat]
-                
-                all_translations_flat = []
-                if all_tags_to_translate:
-                    all_translations_flat = translate_texts(all_tags_to_translate, src_lang="auto", tgt_lang="zh", access_key=access_key)
-                
-                current_translations_dict = {}
-                offset = 0
-                for cat_key in ["general", "characters", "ratings"]:
-                    num_tags_in_cat = len(tag_categories_original_order.get(cat_key, []))
-                    current_translations_dict[cat_key] = all_translations_flat[offset : offset + num_tags_in_cat]
-                    offset += num_tags_in_cat
+            with gr.Accordion("⚙️ 高级设置", open=False):
+                gen_slider = gr.Slider(0, 1, value=0.35, step=0.01, label="通用标签阈值", info="越高 → 标签更少更准")
+                char_slider = gr.Slider(0, 1, value=0.85, step=0.01, label="角色标签阈值", info="推荐保持较高阈值")
+                show_tag_scores = gr.Checkbox(True, label="在列表中显示标签置信度")
                 
-                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)
-                summary_text = generate_summary_text_content(res, current_translations_dict, s_gen, s_char, s_rat, s_sep, s_zh_in_sum)
-
-                yield (gr.update(interactive=True, value="🚀 开始分析"), gr.update(visible=True, value="✅ 分析完成!"), general_html, char_html, rating_html, gr.update(value=summary_text), res, current_translations_dict, tag_categories_original_order)
+                with gr.Group(visible=False) as guest_api_group:
+                    gr.Markdown("### 访客翻译API配置\n由于您不是本空间所有者，需要提供自己的翻译API密钥才能使用翻译功能。")
+                    guest_tencent_id = gr.Textbox(label="腾讯云 Secret ID", type="password")
+                    guest_tencent_key = gr.Textbox(label="腾讯云 Secret Key", type="password")
+                    guest_baidu_json = gr.TextArea(
+                        label="百度翻译凭证 JSON",
+                        placeholder='[{"app_id": "...", "secret_key": "..."}, ...]',
+                        lines=3
+                    )
             
-            except Exception as e:
-                import traceback
-                tb_str = traceback.format_exc()
-                print(f"处理时发生错误: {e}\n{tb_str}")
-                yield (gr.update(interactive=True, value="🚀 开始分析"), gr.update(visible=True, value=f"❌ 处理失败: {str(e)}"), "<p>处理出错</p>", "<p>处理出错</p>", "<p>处理出错</p>", gr.update(value=f"错误: {str(e)}", placeholder="分析失败..."), {}, {}, {})
-            finally:
-                clear_user_provided_keys()
-
-        def update_summary_display(s_gen, s_char, s_rat, s_sep, s_zh_in_sum, current_res_from_state, current_translations_from_state):
-            if not current_res_from_state:
-                return gr.update(placeholder="请先完成一次图像分析以生成汇总。", value="")
-            new_summary_text = generate_summary_text_content(current_res_from_state, current_translations_from_state, s_gen, s_char, s_rat, s_sep, s_zh_in_sum)
-            return gr.update(value=new_summary_text)
-
-        btn.click(
-            process_image_and_generate_outputs,
-            inputs=[
-                img_in, gen_slider, char_slider, show_tag_scores,
-                sum_general, sum_char, sum_rating, sum_sep, sum_show_zh,
-                access_key_input, user_tencent_id, user_tencent_key, user_baidu_json
-            ],
-            outputs=[
-                btn, processing_info,
-                out_general, out_char, out_rating,
-                out_summary,
-                state_res, state_translations_dict, state_tag_categories_for_translation
-            ]
-        )
-        
-        summary_controls = [sum_general, sum_char, sum_rating, sum_sep, sum_show_zh]
-        for ctrl in summary_controls:
-            ctrl.change(fn=update_summary_display, inputs=summary_controls + [state_res, state_translations_dict], outputs=[out_summary])
-
-        return demo
-
-with gr.Blocks(title="登录到图像标签分析器") as demo:
-    CLIENT_ID = os.environ.get("HUGGING_FACE_CLIENT_ID")
-    if not CLIENT_ID:
-        gr.Markdown("# 错误：应用未配置 OIDC 客户端ID\n请在 Space secrets 中设置 `HUGGING_FACE_CLIENT_ID`")
-    else:
-        gr.Markdown("# 欢迎使用 AI 图像标签分析器\n请通过 Hugging Face 登录以继续")
-        login_button = gr.LoginButton(
-            value="🤗 通过 Hugging Face 登录",
-            oauth_client_id=CLIENT_ID,
-            oauth_scopes=["openid", "profile", "email"],
-            oauth_redirect_uri=f"https://huggingface.co/spaces/{os.environ.get('SPACE_ID')}"
-        )
-        user_info_state = gr.State()
-        login_button.login(lambda: None, None, None, js="""
-            (btn) => {
-                const url = new URL(window.location);
-                if (url.searchParams.has('code')) {
-                    btn.style.display = 'none';
-                }
-                return btn;
-            }
-        """)
-        demo.load(
-            fn=lambda request: request.auth,
-            inputs=gr.Request(inputs=[]),
-            outputs=user_info_state,
-            queue=False,
-            js="""
-            (request) => {
-                const url = new URL(window.location);
-                if (!url.searchParams.has('code') && !request.auth) {
-                } else {
-                    document.getElementById('login-interface').style.display = 'none';
-                    document.getElementById('main-app-interface').style.display = 'block';
-                }
-                return request;
-            }
-            """
-        )
+            with gr.Accordion("📊 标签汇总设置", open=True):
+                gr.Markdown("选择要包含在下方汇总文本框中的标签类别：")
+                with gr.Row():
+                    sum_general = gr.Checkbox(True, label="通用标签", min_width=50)
+                    sum_char = gr.Checkbox(True, label="角色标签", min_width=50)
+                    sum_rating = gr.Checkbox(False, label="评分标签", min_width=50)
+                sum_sep = gr.Dropdown(["逗号", "换行", "空格"], value="逗号", label="标签之间的分隔符")
+                sum_show_zh = gr.Checkbox(False, label="在汇总中显示中文翻译")
+
+            processing_info = gr.Markdown("", visible=False)
+
+        with gr.Column(scale=2):
+            with gr.Tabs():
+                with gr.TabItem("🏷️ 通用标签"):
+                    out_general = gr.HTML(label="General Tags")
+                with gr.TabItem("👤 角色标签"):
+                    gr.Markdown("<p style='color:gray; font-size:small;'>提示：角色标签推测基于截至2024年2月的数据。</p>")
+                    out_char = gr.HTML(label="Character Tags")
+                with gr.TabItem("⭐ 评分标签"):
+                    out_rating = gr.HTML(label="Rating Tags")
+            
+            gr.Markdown("### 标签汇总结果")
+            out_summary = gr.Textbox(
+                label="标签汇总",
+                placeholder="分析完成后，此处将显示汇总的英文标签...",
+                lines=5,
+                show_copy_button=True
+            )
 
-        with gr.Column(elem_id="login-interface", visible=True):
-             pass
+    def format_tags_html(tags_dict, translations_list, category_name, show_scores=True, show_translation_in_list=True):
+        if not tags_dict: return "<p>暂无标签</p>"
+        html = '<div class="label-container">'
+        tag_keys = list(tags_dict.keys())
+        for i, tag in enumerate(tag_keys):
+            score = tags_dict[tag]
+            escaped_tag = tag.replace("'", "\\'")
+            html += '<div class="tag-item">'
+            tag_display_html = f'<span class="tag-en" onclick="copyToClipboard(\'{escaped_tag}\')">{tag}</span>'
+            if show_translation_in_list and i < len(translations_list) and translations_list[i]:
+                tag_display_html += f'<span class="tag-zh">({translations_list[i]})</span>'
+            html += f'<div>{tag_display_html}</div>'
+            if show_scores: html += f'<span class="tag-score">{score:.3f}</span>'
+            html += '</div>'
+        html += '</div>'
+        return html
+
+    def generate_summary_text_content(current_res, current_translations_dict, s_gen, s_char, s_rat, s_sep_type, s_show_zh):
+        if not current_res: return "请先分析图像或选择要汇总的标签类别。"
+        summary_parts, separator = [], {"逗号": ", ", "换行": "\n", "空格": " "}.get(s_sep_type, ", ")
+        categories_to_summarize = []
+        if s_gen: categories_to_summarize.append("general")
+        if s_char: categories_to_summarize.append("characters")
+        if s_rat: categories_to_summarize.append("ratings")
+        if not categories_to_summarize: return "请至少选择一个标签类别进行汇总。"
+
+        for cat_key in categories_to_summarize:
+            if current_res.get(cat_key):
+                tags_to_join = []
+                cat_tags_en = list(current_res[cat_key].keys())
+                cat_translations = current_translations_dict.get(cat_key, [])
+                for i, en_tag in enumerate(cat_tags_en):
+                    if s_show_zh and i < len(cat_translations) and cat_translations[i]:
+                        tags_to_join.append(f"{en_tag}({cat_translations[i]})")
+                    else:
+                        tags_to_join.append(en_tag)
+                if tags_to_join: summary_parts.append(separator.join(tags_to_join))
+        joiner = "\n\n" if separator != "\n" and len(summary_parts) > 1 else separator
+        final_summary = joiner.join(summary_parts)
+        return final_summary if final_summary else "选定的类别中没有找到标签。"
+
+    def process_image_and_generate_outputs(img, g_th, c_th, s_scores, s_gen, s_char, s_rat, s_sep, s_zh_in_sum, guest_tc_id, guest_tc_key, guest_bd_json, profile: gr.OAuthProfile | None):
+        if profile is None:
+            gr.Warning("请先登录后再进行分析！")
+            # 返回当前状态，不做任何改变
+            yield (gr.update(), gr.update(), gr.HTML(), gr.HTML(), gr.HTML(), gr.update(), gr.State(), gr.State(), gr.State())
+            return
 
-        with gr.Column(elem_id="main-app-interface", visible=False):
-            main_app = main_interface(gr.UserInfo())
+        if img is None:
+            yield (gr.update(interactive=True, value="🚀 开始分析"), gr.update(visible=True, value="❌ 请先上传图片。"), "", "", "", "", gr.update(placeholder="请先上传图片并开始分析..."), {}, {}, {})
+            return
+        if tagger_instance is None:
+            yield (gr.update(interactive=True, value="🚀 开始分析"), gr.update(visible=True, value="❌ 分析器未成功初始化，请检查控制台错误。"), "", "", "", "", gr.update(placeholder="分析器初始化失败..."), {}, {}, {})
+            return
 
+        yield (gr.update(interactive=False, value="🔄 处理中..."), gr.update(visible=True, value="🔄 正在分析图像..."), gr.HTML(value="<p>分析中...</p>"), gr.HTML(value="<p>分析中...</p>"), gr.HTML(value="<p>分析中...</p>"), gr.update(value="分析中..."), {}, {}, {})
+        
+        try:
+            res, tag_categories_original_order = tagger_instance.predict(img, g_th, c_th)
+            all_tags_to_translate = [tag for cat_key in ["general", "characters", "ratings"] for tag in tag_categories_original_order.get(cat_key, [])]
+            
+            all_translations_flat = []
+            if all_tags_to_translate:
+                is_owner = profile.username.lower() == OWNER_USERNAME
+                if is_owner:
+                    print("- [Auth] 所有者身份，使用预置密钥进行翻译。")
+                    all_translations_flat = translate_texts(all_tags_to_translate)
+                else:
+                    print("- [Auth] 访客身份，使用用户提供的密钥进行翻译。")
+                    if not guest_tc_id and not guest_bd_json:
+                        print("  - [Warning] 访客未提供任何API密钥，将跳过翻译。")
+                        all_translations_flat = all_tags_to_translate
+                    else:
+                        all_translations_flat = translate_texts_with_dynamic_keys(all_tags_to_translate, guest_tc_id, guest_tc_key, guest_bd_json)
+
+            current_translations_dict = {}
+            offset = 0
+            for cat_key in ["general", "characters", "ratings"]:
+                num_tags = len(tag_categories_original_order.get(cat_key, []))
+                current_translations_dict[cat_key] = all_translations_flat[offset : offset + num_tags]
+                offset += num_tags
+            
+            general_html = format_tags_html(res.get("general", {}), current_translations_dict.get("general", []), "general", s_scores)
+            char_html = format_tags_html(res.get("characters", {}), current_translations_dict.get("characters", []), "characters", s_scores)
+            rating_html = format_tags_html(res.get("ratings", {}), current_translations_dict.get("ratings", []), "ratings", s_scores)
+            summary_text = generate_summary_text_content(res, current_translations_dict, s_gen, s_char, s_rat, s_sep, s_zh_in_sum)
 
+            yield (gr.update(interactive=True, value="🚀 开始分析"), gr.update(visible=True, value="✅ 分析完成!"), general_html, char_html, rating_html, gr.update(value=summary_text), res, current_translations_dict, tag_categories_original_order)
+            
+        except Exception as e:
+            import traceback
+            tb_str = traceback.format_exc()
+            print(f"处理时发生错误: {e}\n{tb_str}")
+            yield (gr.update(interactive=True, value="🚀 开始分析"), gr.update(visible=True, value=f"❌ 处理失败: {str(e)}"), "<p>处理出错</p>", "<p>处理出错</p>", "<p>处理出错</p>", gr.update(value=f"错误: {str(e)}", placeholder="分析失败..."), {}, {}, {})
+
+    def update_summary_display(s_gen, s_char, s_rat, s_sep, s_zh_in_sum, current_res, current_translations):
+        if not current_res: return gr.update(placeholder="请先完成一次图像分析以生成汇总。", value="")
+        new_summary_text = generate_summary_text_content(current_res, current_translations, s_gen, s_char, s_rat, s_sep, s_zh_in_sum)
+        return gr.update(value=new_summary_text)
+
+    # --- 修改后的认证检查函数 ---
+    def check_user_auth(profile: gr.OAuthProfile | None):
+        if not OWNER_USERNAME: print("⚠️ 警告: 未设置 OWNER_USERNAME 环境变量。所有用户都将被视为访客。")
+        
+        if profile is None:
+            print("- [Auth] 用户未登录。")
+            return gr.update(visible=True), "请先登录..."
+
+        username = profile.username.lower()
+        if username == OWNER_USERNAME:
+            print(f"- [Auth] 所有者 '{profile.username}' 已连接。")
+            user_info_md = f"✅ **所有者模式**: 欢迎, {profile.name}! 将使用预置翻译服务。"
+            return gr.update(visible=False), user_info_md
+        else:
+            print(f"- [Auth] 访客 '{profile.username}' 已连接，显示 API Key 输入框。")
+            user_info_md = f"👋 **访客模式**: 欢迎, {profile.name}! 请在高级设置中提供您自己的翻译密钥。"
+            return gr.update(visible=True), user_info_md
+
+    # --- 修改事件绑定 ---
+    login_button.load(fn=check_user_auth, inputs=login_button, outputs=[guest_api_group, user_info])
+
+    btn.click(
+        process_image_and_generate_outputs,
+        inputs=[img_in, gen_slider, char_slider, show_tag_scores, sum_general, sum_char, sum_rating, sum_sep, sum_show_zh, guest_tencent_id, guest_tencent_key, guest_baidu_json, login_button],
+        outputs=[btn, processing_info, out_general, out_char, out_rating, out_summary, state_res, state_translations_dict, state_tag_categories_for_translation]
+    )
+
+    summary_controls = [sum_general, sum_char, sum_rating, sum_sep, sum_show_zh]
+    for ctrl in summary_controls:
+        ctrl.change(fn=update_summary_display, inputs=summary_controls + [state_res, state_translations_dict], outputs=[out_summary])
+    
 if __name__ == "__main__":
     if tagger_instance is None:
         print("CRITICAL: Tagger 未能初始化，应用功能将受限。请检查之前的错误信息。")
-    if "SPACE_ID" in os.environ:
-        demo.launch()
-    else:
-        with gr.Blocks() as local_demo:
-             fake_user_info = gr.UserInfo(name="local_user", email="local@test.com")
-             main_interface(fake_user_info)
-        local_demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860)