Create app.py

app.py

@@ -0,0 +1,60 @@
+from __future__ import annotations
+
+import os
+import gradio as gr
+from huggingface_hub import HfApi
+
+HF_TOKEN = os.getenv("HF_TOKEN")
+SPACE_ID = os.getenv("SPACE_ID")  # Hugging Face Spaces 内置环境变量，例如 username/space-name
+DISCUSSION_NUM = int(os.getenv("DISCUSSION_NUM", "3"))  # 改成你的 discussion 编号
+
+api = HfApi(token=HF_TOKEN)
+
+
+def send_hf_notification(message: str, profile: gr.OAuthProfile | None) -> str:
+    if profile is None:
+        return "请先点击上面的 Sign in with Hugging Face 登录。"
+
+    if not HF_TOKEN:
+        return "错误：没有配置 HF_TOKEN secret。"
+
+    if not SPACE_ID:
+        return "错误：没有找到 SPACE_ID。请确认代码运行在 Hugging Face Space 上。"
+
+    username = profile.username  # HF username，例如 abidlabs
+
+    comment = f"""@{username}
+
+你的任务已经完成：
+
+{message}
+"""
+
+    api.comment_discussion(
+        repo_id=SPACE_ID,
+        repo_type="space",
+        discussion_num=DISCUSSION_NUM,
+        comment=comment,
+    )
+
+    return f"已发送 Hugging Face notification 给 @{username}。"
+
+
+with gr.Blocks() as demo:
+    gr.Markdown("## Hugging Face Notification Demo")
+    gr.LoginButton()
+
+    msg = gr.Textbox(
+        label="通知内容",
+        value="你的任务已经完成，请回到 Space 查看结果。",
+    )
+    btn = gr.Button("发送 HF Notification")
+    out = gr.Textbox(label="结果")
+
+    btn.click(
+        fn=send_hf_notification,
+        inputs=msg,
+        outputs=out,
+    )
+
+demo.launch()