Update app.py

app.py

@@ -158,7 +158,7 @@ function copyToClipboard(text) {
 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)
+        user_info = gr.Markdown("正在检查登录状态...", visible=True)
 
     gr.Markdown("# 🖼️ AI 图像标签分析器")
     gr.Markdown("上传图片自动识别标签，支持中英文显示和一键复制。[NovelAI在线绘画](https://nai.idlecloud.cc/)")
@@ -229,7 +229,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=cus
             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>'
+            if show_scores: html += f'<span class.tag-score">{score:.3f}</span>'
             html += '</div>'
         html += '</div>'
         return html
@@ -257,21 +257,37 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=cus
         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 update_auth_ui(profile: gr.OAuthProfile | None):
+        if profile is None:
+            return gr.update(visible=True), "请先登录..."
+        
+        username = profile.username.lower()
+        is_owner = username == OWNER_USERNAME
+        
+        if is_owner:
+            user_info_md = f"✅ **所有者模式**: 欢迎, {profile.name}! 将使用预置翻译服务。"
+        else:
+            user_info_md = f"👋 **访客模式**: 欢迎, {profile.name}! 请在高级设置中提供您自己的翻译密钥。"
+            
+        return gr.update(visible=not is_owner), user_info_md
 
     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):
+        guest_api_update, user_info_update = update_auth_ui(profile)
+
         if profile is None:
             gr.Warning("请先登录后再进行分析！")
-            yield (gr.update(), gr.update(visible=False), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update())
+            yield (gr.update(), gr.update(visible=False), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), guest_api_update, user_info_update)
             return
 
         if img is None:
-            yield (gr.update(interactive=True, value="🚀 开始分析"), gr.update(visible=True, value="❌ 请先上传图片。"), "", "", "", "", gr.update(placeholder="请先上传图片并开始分析..."), {}, {}, {})
+            yield (gr.update(interactive=True, value="🚀 开始分析"), gr.update(visible=True, value="❌ 请先上传图片。"), "", "", "", gr.update(placeholder="请先上传图片并开始分析..."), {}, {}, {}, guest_api_update, user_info_update)
             return
         if tagger_instance is None:
-            yield (gr.update(interactive=True, value="🚀 开始分析"), gr.update(visible=True, value="❌ 分析器未成功初始化，请检查控制台错误。"), "", "", "", "", gr.update(placeholder="分析器初始化失败..."), {}, {}, {})
+            yield (gr.update(interactive=True, value="🚀 开始分析"), gr.update(visible=True, value="❌ 分析器未成功初始化，请检查控制台错误。"), "", "", "", gr.update(placeholder="分析器初始化失败..."), {}, {}, {}, guest_api_update, user_info_update)
             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="分析中..."), {}, {}, {})
+        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="分析中..."), {}, {}, {}, guest_api_update, user_info_update)
         
         try:
             res, tag_categories_original_order = tagger_instance.predict(img, g_th, c_th)
@@ -303,42 +319,39 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=cus
             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)
+            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, guest_api_update, user_info_update)
             
         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="分析失败..."), {}, {}, {})
+            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="分析失败..."), {}, {}, {}, guest_api_update, user_info_update)
 
     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):
+    def check_user_auth(profile: gr.OAuthProfile | None, request: gr.Request = 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
-
-    demo.load(fn=check_user_auth, inputs=login_button, outputs=[guest_api_group, user_info])
+            if profile.username.lower() == OWNER_USERNAME:
+                print(f"- [Auth] 所有者 '{profile.username}' 已连接。")
+            else:
+                print(f"- [Auth] 访客 '{profile.username}' 已连接，显示 API Key 输入框。")
+        return update_auth_ui(profile)
+    
+    demo.load(fn=check_user_auth, inputs=[login_button], outputs=[guest_api_group, user_info])
+    
+    img_in.upload(fn=update_auth_ui, 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]
+        outputs=[btn, processing_info, out_general, out_char, out_rating, out_summary, state_res, state_translations_dict, state_tag_categories_for_translation, guest_api_group, user_info]
     )
 
     summary_controls = [sum_general, sum_char, sum_rating, sum_sep, sum_show_zh]