Update app.py

app.py

@@ -146,13 +146,23 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=cus
     gr.Markdown("# 🖼️ AI 图像标签分析器")
     gr.Markdown("上传图片自动识别标签，支持中英文显示和一键复制。[NovelAI在线绘画](https://nai.idlecloud.cc/)")
     
-    # 登录状态显示
-    user_status_md = gr.Markdown("ℹ️ 正在检查登录状态...")
+    with gr.Group(visible=True) as login_view:
+        gr.Markdown(
+            """
+            <div style="padding: 20px; text-align: center; background-color: #fffbe6; border: 1px solid #ffe58f; border-radius: 8px;">
+                <p style="font-size: 1.1em; color: #d46b08;">🚫 <strong>您需要登录才能使用此应用</strong></p>
+                <p style="color: #666;">请点击下方的按钮通过 Hugging Face 登录。登录后，请刷新此页面。</p>
+            </div>
+            """
+        )
+        gr.LoginButton(value="🤗 通过 Hugging Face 登录")
+
+    user_status_md = gr.Markdown(visible=False)
     
     state_res = gr.State({})
     state_translations_dict = gr.State({})
 
-    with gr.Row(visible=False) as main_interface: # 默认隐藏主界面
+    with gr.Row(visible=False) as main_interface:
         with gr.Column(scale=1):
             img_in = gr.Image(type="pil", label="上传图片", height=300)
             btn = gr.Button("🚀 开始分析", variant="primary", elem_classes=["btn-analyze-container"])
@@ -197,14 +207,11 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=cus
             try:
                 user_info = whoami(token=token)
                 welcome_msg = f"✅ 您好, **{user_info.get('fullname', user_info.get('name'))}**！欢迎使用本应用。"
-                return welcome_msg, gr.update(visible=True) # 显示主界面
+                return gr.update(visible=False), gr.update(value=welcome_msg, visible=True), gr.update(visible=True)
             except Exception as e:
                 print(f"Token 无效或已过期: {e}")
         
-        # 未登录或Token无效
-        login_url = "/login?redirect=/"
-        msg = f"**<p style='color:red; text-align:center;'>🚫 检测到您尚未登录，请先 <a href='{login_url}' target='_top'>[点击此处登录]</a> 后刷新页面。</p>**"
-        return msg, gr.update(visible=False) # 隐藏主界面
+        return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False)
         
     def format_tags_html(tags_dict, translations_list, show_scores):
         if not tags_dict: return "<p>暂无标签</p>"
@@ -248,7 +255,6 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=cus
 
         yield gr.update(interactive=False, value="🔄 处理中..."), gr.update(visible=True, value="🔄 正在分析..."), *["<p>分析中...</p>"]*3, "分析中...", {}, {}
 
-        # 密钥决策逻辑
         use_space_keys = bool(ACCESS_PASSWORD and access_pwd == ACCESS_PASSWORD)
         
         final_tencent_id, final_tencent_key, baidu_json_str = (
@@ -290,7 +296,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=cus
             raise gr.Error(f"处理时发生错误: {e}")
 
     # ----------------- 绑定事件 -----------------
-    demo.load(fn=check_user_status, inputs=None, outputs=[user_status_md, main_interface], queue=False)
+    demo.load(fn=check_user_status, inputs=None, outputs=[login_view, user_status_md, main_interface], queue=False)
     
     btn.click(
         process_image_and_generate_outputs,