Update app.py

app.py

@@ -6,7 +6,8 @@ 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, get_space_runtime, HfApi
+from huggingface_hub.hf_api import SpaceRuntime
 
 from translator import translate_texts
 
@@ -183,22 +184,33 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=cus
                 with gr.TabItem("⭐ 评分标签"): out_rating = gr.HTML(label="Rating Tags")
             gr.Markdown("### 标签汇总结果")
             out_summary = gr.Textbox(label="标签汇总", lines=5, show_copy_button=True)
-            
+
     def get_token_from_request(request: gr.Request) -> str | None:
-        """Helper function to extract token from request headers."""
         auth_header = request.headers.get("authorization")
         if auth_header and auth_header.startswith("Bearer "):
             return auth_header.split(" ")[1]
         return None
 
+    def check_user_is_owner(user_info: dict | None, space_runtime: SpaceRuntime | None) -> bool:
+        if not user_info or not space_runtime:
+            return False
+        if user_info.get("name") == space_runtime.owner:
+            return True
+        user_orgs = user_info.get("orgs", [])
+        if any(org.get("name") == space_runtime.owner for org in user_orgs):
+            return True
+            
+        return False
+
     def check_user_status(request: gr.Request):
         token = get_token_from_request(request)
         if token:
             try:
                 user_info = whoami(token=token)
                 space_runtime = get_space_runtime()
-                if space_runtime and user_info and user_info["name"] == space_runtime.owner:
-                    return f"✅ 以所有者 **{user_info['fullname']}** 身份登录，将使用空间配置的密钥。", gr.update(visible=False)
+                
+                if check_user_is_owner(user_info, space_runtime):
+                    return f"✅ 以所有者 **{user_info.get('fullname', user_info.get('name'))}** 身份登录，将使用空间配置的密钥。", gr.update(visible=False)
                 else:
                     return f"👋 你好, **{user_info.get('fullname', '用户')}**！请在下方提供你自己的翻译 API 密钥。", gr.update(visible=True, open=True)
             except Exception as e:
@@ -251,7 +263,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=cus
             try:
                 user_info = whoami(token=token)
                 space_runtime = get_space_runtime()
-                if space_runtime and user_info and user_info["name"] == space_runtime.owner:
+                if check_user_is_owner(user_info, space_runtime):
                     is_owner = True
             except Exception: pass
         
@@ -293,7 +305,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=cus
             traceback.print_exc()
             raise gr.Error(f"处理时发生错误: {e}")
 
-    demo.load(fn=check_user_status, outputs=[user_status_md, api_key_accordion], queue=False)
+    demo.load(fn=check_user_status, inputs=None, outputs=[user_status_md, api_key_accordion], queue=False)
     
     btn.click(
         process_image_and_generate_outputs,
@@ -321,4 +333,4 @@ with gr.Blocks(theme=gr.themes.Soft(), title="AI 图像标签分析器", css=cus
 if __name__ == "__main__":
     if tagger_instance is None:
         print("CRITICAL: Tagger 未能初始化，应用功能将受限。")
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860)