Update app.py

app.py

@@ -8,115 +8,100 @@ from huggingface_hub import HfApi
 
 
 # ============================================================
-# 需要你配置的内容
+# 基础配置
 # ============================================================
 
-# 方式1：推荐。Hugging Face Space 会自动提供 SPACE_ID，例如：
+# Hugging Face Space 通常会自动提供 SPACE_ID，例如：
 # wenjiao/low_bit_open_llm_leaderboard_5
 SPACE_REPO_ID = os.getenv("SPACE_ID")
 
-# 方式2：如果 SPACE_ID 没拿到，就用你手动写的这个。
-# 这里改成你自己的 Space repo id。
+# 如果 SPACE_ID 没拿到，就用你手动填的这个。
+# 这里改成你的 Space 名字。
 if not SPACE_REPO_ID:
-    SPACE_REPO_ID = "wenjiao/timer"
+    SPACE_REPO_ID = "wenjiao/low_bit_open_llm_leaderboard_5"
 
-# 你手动创建的 Space Notifications discussion 编号。
-# 比如 discussion 链接是 /discussions/3，那这里就是 3。
-DISCUSSION_NUM = int(os.getenv("DISCUSSION_NUM", "1"))
+# 你的 Space Notifications discussion 编号。
+# 如果链接是 /discussions/3，这里就是 3。
+DISCUSSION_NUM = int(os.getenv("DISCUSSION_NUM", "3"))
 
-# Hugging Face token，需要在 Space Settings -> Variables and secrets -> New secret 里配置。
-# secret 名字必须叫 HF_TOKEN。
+# 在 Space Settings -> Variables and secrets -> New secret 里添加：
+# HF_TOKEN=hf_xxx
 HF_TOKEN = os.getenv("HF_TOKEN")
 
-api = HfApi()
-
 
 # ============================================================
-# Hugging Face 通知函数
+# 工具函数
 # ============================================================
 
-def notify_hf_user(
-    hf_username: str,
-    task_name: str,
-    status: str = "success",
-    detail: str = "",
-    result_url: str | None = None,
-):
-    """
-    通过在 Space 的固定 Discussion 中评论并 @ 用户，触发 Hugging Face 原生 notification。
-
-    参数：
-    hf_username: 目标 Hugging Face 用户名，不要带 @
-    task_name: 任务名称
-    status: success 或 failed
-    detail: 额外说明
-    result_url: 可选结果链接
-    """
-
+def get_hf_api():
     if not HF_TOKEN:
         raise RuntimeError(
             "没有检测到 HF_TOKEN。请在 Space Settings -> Variables and secrets 里添加 Secret：HF_TOKEN"
         )
 
-    if not SPACE_REPO_ID:
-        raise RuntimeError(
-            "没有检测到 SPACE_REPO_ID。请检查 SPACE_ID 环境变量，或者在 app.py 里手动填写 SPACE_REPO_ID。"
-        )
+    return HfApi(token=HF_TOKEN)
+
+
+def clean_hf_username(hf_username):
+    hf_username = str(hf_username or "").strip()
+    hf_username = hf_username.lstrip("@")
+    return hf_username
+
 
-    hf_username = hf_username.strip().lstrip("@")
+def notify_hf_user(hf_username, task_name, status, detail, result_url=None):
+    hf_username = clean_hf_username(hf_username)
 
     if not hf_username:
         raise ValueError("Hugging Face 用户名不能为空。")
 
+    api = get_hf_api()
+
     now = datetime.now(timezone.utc).strftime("%Y-%m-%d %H:%M:%S UTC")
 
     if status == "success":
-        title_line = "✅ Space 任务已完成"
+        status_text = "✅ Space 任务已完成"
     else:
-        title_line = "❌ Space 任务失败"
-
-    comment = f"""@{hf_username}
-
-{title_line}
-
-**任务名称：** `{task_name}`  
-**Space：** `{SPACE_REPO_ID}`  
-**时间：** {now}
-
-{detail}
-"""
+        status_text = "❌ Space 任务失败"
+
+    comment_lines = [
+        "@" + hf_username,
+        "",
+        status_text,
+        "",
+        "任务名称：" + str(task_name),
+        "Space：" + str(SPACE_REPO_ID),
+        "时间：" + now,
+        "",
+        str(detail),
+    ]
 
     if result_url:
-        comment += f"""
+        comment_lines.extend([
+            "",
+            "结果链接：" + str(result_url),
+        ])
 
-**结果链接：** {result_url}
-"""
+    comment = "\n".join(comment_lines)
 
     api.comment_discussion(
         repo_id=SPACE_REPO_ID,
         repo_type="space",
         discussion_num=DISCUSSION_NUM,
         comment=comment,
-        token=HF_TOKEN,
     )
 
-    discussion_url = f"https://huggingface.co/spaces/{SPACE_REPO_ID}/discussions/{DISCUSSION_NUM}"
+    discussion_url = "https://huggingface.co/spaces/" + SPACE_REPO_ID + "/discussions/" + str(DISCUSSION_NUM)
     return discussion_url
 
 
 # ============================================================
-# 这里替换成你自己的真实任务逻辑
+# 这里替换成你自己的真实任务
 # ============================================================
 
-def run_real_task(task_name: str, progress=gr.Progress()):
+def run_real_task(task_name, progress=gr.Progress()):
     """
     这里现在是模拟任务。
-    你后面把这里替换成：
-    - 模型推理
-    - eval 评测
-    - leaderboard 更新
-    - 文件生成
-    - 任何你想跑完后通知用户的任务
+    你后面可以把 time.sleep 这几行换成你的模型推理、评测、排行榜更新等代码。
     """
 
     progress(0.1, desc="开始任务")
@@ -130,95 +115,125 @@ def run_real_task(task_name: str, progress=gr.Progress()):
 
     progress(1.0, desc="任务完成")
 
-    result_text = f"任务 `{task_name}` 已经成功完成。"
-    return result_text
+    return "任务 " + str(task_name) + " 已经成功完成。"
 
 
 # ============================================================
-# Gradio 按钮回调
+# Gradio 回调函数
 # ============================================================
 
-def run_and_notify(task_name: str, hf_username: str, result_url: str, progress=gr.Progress()):
-    task_name = task_name.strip() or "未命名任务"
-    hf_username = hf_username.strip().lstrip("@")
-    result_url = result_url.strip()
+def run_and_notify(task_name, hf_username, result_url, progress=gr.Progress()):
+    task_name = str(task_name or "").strip()
+    hf_username = clean_hf_username(hf_username)
+    result_url = str(result_url or "").strip()
+
+    if not task_name:
+        task_name = "未命名任务"
 
     if not hf_username:
-        return "❌ 请输入 Hugging Face 用户名，例如：wenjiao，不要写邮箱。"
+        return "❌ 请输入 Hugging Face 用户名，例如 wenjiao，不要写邮箱。"
+
+    if not result_url:
+        result_url = "https://huggingface.co/spaces/" + SPACE_REPO_ID
 
     try:
-        # 1. 先跑真实任务
         result_text = run_real_task(task_name, progress=progress)
 
-        # 2. 任务成功后触发 Hugging Face 原生通知
         discussion_url = notify_hf_user(
             hf_username=hf_username,
             task_name=task_name,
             status="success",
             detail=result_text,
-            result_url=result_url or f"https://huggingface.co/spaces/{SPACE_REPO_ID}",
+            result_url=result_url,
         )
 
-        return f"""✅ 任务完成，并已尝试通过 Hugging Face @ 通知 @{hf_username}
-
-任务结果：
-{result_text}
-
-通知位置：
-{discussion_url}
-"""
+        return (
+            "✅ 任务完成，并已尝试通过 Hugging Face @ 通知 @" + hf_username + "\n\n"
+            + "任务结果：\n"
+            + result_text
+            + "\n\n通知位置：\n"
+            + discussion_url
+        )
 
     except Exception as e:
-        error_detail = traceback.format_exc()
+        error_text = traceback.format_exc()
 
-        # 任务失败时，也尝试通知用户
         try:
             discussion_url = notify_hf_user(
                 hf_username=hf_username,
                 task_name=task_name,
                 status="failed",
-                detail=f"错误信息：\n```text\n{str(e)}\n```",
-                result_url=result_url or f"https://huggingface.co/spaces/{SPACE_REPO_ID}",
+                detail="错误信息：" + str(e),
+                result_url=result_url,
             )
 
-            return f"""❌ 任务失败，但已尝试通过 Hugging Face @ 通知 @{hf_username}
+            return (
+                "❌ 任务失败，但已尝试通过 Hugging Face @ 通知 @" + hf_username + "\n\n"
+                + "错误信息：\n"
+                + str(e)
+                + "\n\n通知位置：\n"
+                + discussion_url
+                + "\n\n完整错误：\n"
+                + error_text
+            )
 
-错误信息：
-{str(e)}
+        except Exception as notify_error:
+            return (
+                "❌ 任务失败，并且发送 Hugging Face 通知也失败了。\n\n"
+                + "任务错误：\n"
+                + str(e)
+                + "\n\n通知错误：\n"
+                + str(notify_error)
+                + "\n\n完整错误：\n"
+                + error_text
+            )
 
-通知位置：
-{discussion_url}
 
-完整错误：
-{error_detail}
-"""
+# ============================================================
+# Gradio 页面
+# ============================================================
 
-        except Exception as notify_error:
-            return f"""❌ 任务失败，并且发送 Hugging Face 通知也失败了。
+description_text = (
+    "# Hugging Face Space 原生通知测试\n\n"
+    + "点击按钮后，Space 会模拟运行一个任务。任务结束后，会自动往固定 Discussion 里评论，并 @ 目标用户。\n\n"
+    + "当前 Space Repo ID：`" + str(SPACE_REPO_ID) + "`\n\n"
+    + "当前 Discussion Num：`" + str(DISCUSSION_NUM) + "`\n\n"
+    + "请确认你已经在 Space Settings 里添加 Secret：`HF_TOKEN=你的 Hugging Face token`。\n"
+)
 
-任务错误：
-{str(e)}
+with gr.Blocks(title="HF Space Notification Demo") as demo:
+    gr.Markdown(description_text)
 
-通知错误：
-{str(notify_error)}
+    with gr.Row():
+        task_name_input = gr.Textbox(
+            label="任务名称",
+            value="test-eval-job",
+            placeholder="例如：llm-eval-run-001",
+        )
 
-完整错误：
-{error_detail}
-"""
+        hf_username_input = gr.Textbox(
+            label="要通知的 Hugging Face 用户名",
+            placeholder="例如：wenjiao，不要带 @，不要写邮箱",
+        )
 
+    result_url_input = gr.Textbox(
+        label="结果链接，可不填",
+        placeholder="https://huggingface.co/spaces/" + str(SPACE_REPO_ID),
+    )
 
-# ============================================================
-# Gradio UI
-# ============================================================
+    run_button = gr.Button("运行任务并触发 Hugging Face 通知", variant="primary")
 
-with gr.Blocks(title="HF Space Notification Demo") as demo:
-    gr.Markdown(
-        f"""
-# Hugging Face Space 原生通知测试
+    output = gr.Textbox(
+        label="运行状态",
+        lines=12,
+    )
 
-这个 Space 会在任务结束后，自动往固定 Discussion 里评论，并 `@目标用户`。
+    run_button.click(
+        fn=run_and_notify,
+        inputs=[task_name_input, hf_username_input, result_url_input],
+        outputs=output,
+    )
 
-当前 Space Repo ID：
 
-```text
-{SPACE_REPO_ID}
+demo.queue()
+demo.launch()