Update app.py

app.py

@@ -0,0 +1,224 @@
+import os
+import time
+import traceback
+from datetime import datetime, timezone
+
+import gradio as gr
+from huggingface_hub import HfApi
+
+
+# ============================================================
+# 需要你配置的内容
+# ============================================================
+
+# 方式1：推荐。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。
+if not SPACE_REPO_ID:
+    SPACE_REPO_ID = "wenjiao/timer"
+
+# 你手动创建的 Space Notifications discussion 编号。
+# 比如 discussion 链接是 /discussions/3，那这里就是 3。
+DISCUSSION_NUM = int(os.getenv("DISCUSSION_NUM", "1"))
+
+# Hugging Face token，需要在 Space Settings -> Variables and secrets -> New secret 里配置。
+# secret 名字必须叫 HF_TOKEN。
+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: 可选结果链接
+    """
+
+    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。"
+        )
+
+    hf_username = hf_username.strip().lstrip("@")
+
+    if not hf_username:
+        raise ValueError("Hugging Face 用户名不能为空。")
+
+    now = datetime.now(timezone.utc).strftime("%Y-%m-%d %H:%M:%S UTC")
+
+    if status == "success":
+        title_line = "✅ Space 任务已完成"
+    else:
+        title_line = "❌ Space 任务失败"
+
+    comment = f"""@{hf_username}
+
+{title_line}
+
+**任务名称：** `{task_name}`  
+**Space：** `{SPACE_REPO_ID}`  
+**时间：** {now}
+
+{detail}
+"""
+
+    if result_url:
+        comment += f"""
+
+**结果链接：** {result_url}
+"""
+
+    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}"
+    return discussion_url
+
+
+# ============================================================
+# 这里替换成你自己的真实任务逻辑
+# ============================================================
+
+def run_real_task(task_name: str, progress=gr.Progress()):
+    """
+    这里现在是模拟任务。
+    你后面把这里替换成：
+    - 模型推理
+    - eval 评测
+    - leaderboard 更新
+    - 文件生成
+    - 任何你想跑完后通知用户的任务
+    """
+
+    progress(0.1, desc="开始任务")
+    time.sleep(1)
+
+    progress(0.4, desc="正在运行任务")
+    time.sleep(1)
+
+    progress(0.7, desc="正在生成结果")
+    time.sleep(1)
+
+    progress(1.0, desc="任务完成")
+
+    result_text = f"任务 `{task_name}` 已经成功完成。"
+    return result_text
+
+
+# ============================================================
+# 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()
+
+    if not hf_username:
+        return "❌ 请输入 Hugging Face 用户名，例如：wenjiao，不要写邮箱。"
+
+    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}",
+        )
+
+        return f"""✅ 任务完成，并已尝试通过 Hugging Face @ 通知 @{hf_username}
+
+任务结果：
+{result_text}
+
+通知位置：
+{discussion_url}
+"""
+
+    except Exception as e:
+        error_detail = 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}",
+            )
+
+            return f"""❌ 任务失败，但已尝试通过 Hugging Face @ 通知 @{hf_username}
+
+错误信息：
+{str(e)}
+
+通知位置：
+{discussion_url}
+
+完整错误：
+{error_detail}
+"""
+
+        except Exception as notify_error:
+            return f"""❌ 任务失败，并且发送 Hugging Face 通知也失败了。
+
+任务错误：
+{str(e)}
+
+通知错误：
+{str(notify_error)}
+
+完整错误：
+{error_detail}
+"""
+
+
+# ============================================================
+# Gradio UI
+# ============================================================
+
+with gr.Blocks(title="HF Space Notification Demo") as demo:
+    gr.Markdown(
+        f"""
+# Hugging Face Space 原生通知测试
+
+这个 Space 会在任务结束后，自动往固定 Discussion 里评论，并 `@目标用户`。
+
+当前 Space Repo ID：
+
+```text
+{SPACE_REPO_ID}