Update app.py

app.py

@@ -6,7 +6,7 @@ import numpy as np
 import onnxruntime as rt
 import pandas as pd
 from PIL import Image
-from huggingface_hub import whoami, get_space_runtime
+from huggingface_hub import whoami
 
 from translator import translate_texts
 
@@ -18,7 +18,6 @@ MODEL_FILENAME = "model.onnx"
 LABEL_FILENAME = "selected_tags.csv"
 
 HF_TOKEN = os.environ.get("HF_TOKEN")
-# 新增：从环境变量读取访问密码
 ACCESS_PASSWORD = os.environ.get("ACCESS_PASSWORD")
 
 # ------------------------------------------------------------------
@@ -146,19 +145,13 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=cus
     gr.Markdown("# 🖼️ AI 图像标签分析器")
     gr.Markdown("上传图片自动识别标签，支持中英文显示和一键复制。[NovelAI在线绘画](https://nai.idlecloud.cc/)")
     
-    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 登录")
+    # 统一的状态和登录/登出控制区域
+    with gr.Row():
+        user_status_html = gr.HTML("<p>ℹ️ 正在检查登录状态...</p>")
+    with gr.Row():
+        login_button = gr.LoginButton(value="🤗 通过 Hugging Face 登录", visible=True)
+        logout_button = gr.LogoutButton(value="退出登录", visible=False)
 
-    user_status_md = gr.Markdown(visible=False)
-    
     state_res = gr.State({})
     state_translations_dict = gr.State({})
 
@@ -206,12 +199,33 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=cus
         if token:
             try:
                 user_info = whoami(token=token)
-                welcome_msg = f"✅ 您好, **{user_info.get('fullname', user_info.get('name'))}**！欢迎使用本应用。"
-                return gr.update(visible=False), gr.update(value=welcome_msg, visible=True), gr.update(visible=True)
+                welcome_msg = f"<p style='color:green;font-weight:bold;'>✅ 您好, {user_info.get('fullname', user_info.get('name'))}！欢迎使用。</p>"
+                # 已登录：显示欢迎信息，隐藏登录按钮，显示登出按钮，显示主界面
+                return (
+                    gr.update(value=welcome_msg),
+                    gr.update(visible=False),
+                    gr.update(visible=True),
+                    gr.update(visible=True)
+                )
             except Exception as e:
                 print(f"Token 无效或已过期: {e}")
-        
-        return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False)
+                error_msg = "<p style='color:red;'>🚫 登录令牌无效或已过期，请重新登录。</p>"
+                # 令牌无效：显示错误，显示登录按钮，隐藏登出按钮，隐藏主界面
+                return (
+                    gr.update(value=error_msg),
+                    gr.update(visible=True),
+                    gr.update(visible=False),
+                    gr.update(visible=False)
+                )
+
+        # 未登录
+        info_msg = "<p style='color:#d46b08;'>🚫 您需要登录才能使用此应用。</p>"
+        return (
+            gr.update(value=info_msg),
+            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>"
@@ -296,7 +310,12 @@ 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=[login_view, user_status_md, main_interface], queue=False)
+    demo.load(
+        fn=check_user_status, 
+        inputs=None, 
+        outputs=[user_status_html, login_button, logout_button, main_interface], 
+        queue=False
+    )
     
     btn.click(
         process_image_and_generate_outputs,