created main.py

main.py

@@ -0,0 +1,97 @@
+from fastapi import FastAPI, Request, Header, HTTPException
+from pydantic import BaseModel
+from typing import Optional, Any, Dict, List
+import httpx
+import os
+
+app = FastAPI()
+
+# --- Configuration ---
+# Set these in your Space's "Settings" -> "Variables and secrets"
+DISCORD_WEBHOOK_URL = os.environ.get("DISCORD_WEBHOOK_URL")
+# Optional: Secure your webhook so only HF can trigger it
+WEBHOOK_SECRET = os.environ.get("WEBHOOK_SECRET")
+
+# --- Pydantic Models for HF Payload ---
+class HFRepo(BaseModel):
+    type: str
+    name: str
+    id: str
+    url: Dict[str, str]
+
+class HFEvent(BaseModel):
+    action: str
+    scope: str
+
+class HFPayload(BaseModel):
+    repo: HFRepo
+    event: HFEvent
+    discussion: Optional[Dict[str, Any]] = None
+    comment: Optional[Dict[str, Any]] = None
+    webhook: Optional[Dict[str, Any]] = None
+
+@app.get("/")
+def home():
+    return {"status": "Webhook listener is running"}
+
+@app.post("/webhook")
+async def handle_webhook(
+    payload: HFPayload, 
+    x_webhook_secret: Optional[str] = Header(None)
+):
+    # 1. Security Check
+    if WEBHOOK_SECRET and x_webhook_secret != WEBHOOK_SECRET:
+        raise HTTPException(status_code=401, detail="Invalid Secret")
+
+    # 2. Format for Discord
+    embed = build_discord_embed(payload)
+    
+    # 3. Send to Discord
+    if DISCORD_WEBHOOK_URL:
+        async with httpx.AsyncClient() as client:
+            resp = await client.post(
+                DISCORD_WEBHOOK_URL, 
+                json={"embeds": [embed]}
+            )
+            print(f"Sent to Discord: {resp.status_code}")
+    
+    return {"message": "Processed"}
+
+def build_discord_embed(data: HFPayload):
+    repo_name = data.repo.name
+    repo_url = data.repo.url.get('web', '')
+    event_scope = data.event.scope
+    event_action = data.event.action
+    
+    # Default Color (Grey)
+    color = 0x95a5a6
+    title = f"Update on {repo_name}"
+    description = f"**Action:** {event_action}\n**Scope:** {event_scope}"
+
+    # Customize based on event type
+    if event_scope.startswith("repo.content"):
+        # Code/Model update (Blue)
+        color = 0x3498db
+        title = f"New Commit to {repo_name}"
+        description = "Model/Dataset files have been updated."
+    
+    elif event_scope.startswith("discussion"):
+        # Discussion/PR (Green)
+        color = 0x2ecc71
+        disc_title = data.discussion.get('title', 'No Title') if data.discussion else "Unknown"
+        title = f"Discussion Update: {disc_title}"
+        
+        if data.comment:
+            description = f"**New Comment** on {repo_name}:\n> {data.comment.get('content', '')[:200]}..."
+        else:
+            description = f"Discussion status: **{event_action}**"
+
+    return {
+        "title": title,
+        "description": description,
+        "url": repo_url,
+        "color": color,
+        "footer": {
+            "text": "Hugging Face Webhook"
+        }
+    }