claud

main.py

@@ -1,13 +1,30 @@
 from fastapi import FastAPI, Request, Response
 import os
+import requests
+import json
 
-KEY = os.environ.get("WEBHOOK_SECRET")
+HF_WEBHOOK_SECRET = os.environ.get("HF_WEBHOOK_SECRET")
+SLACK_WEBHOOK_URL = os.environ.get("SLACK_WEBHOOK_URL")
+
+def send_slack_message(message: str):
+    """Send a message to Slack using webhook URL"""
+    if not SLACK_WEBHOOK_URL:
+        print(f"No Slack webhook URL configured. Message: {message}")
+        return
+    
+    payload = {"text": message}
+    try:
+        response = requests.post(SLACK_WEBHOOK_URL, json=payload)
+        response.raise_for_status()
+        print(f"Slack message sent successfully: {message}")
+    except requests.exceptions.RequestException as e:
+        print(f"Failed to send Slack message: {e}")
 
 app = FastAPI()
 
 @app.post("/webhook")
 async def webhook(request: Request):
-    if request.headers.get("X-Webhook-Secret") != KEY:
+    if request.headers.get("X-Webhook-Secret") != HF_WEBHOOK_SECRET:
         return Response("Invalid secret", status_code=401)
     data = await request.json()
     event = data.get("event", {})
@@ -19,8 +36,10 @@ async def webhook(request: Request):
     if action == "move":
         old_name = data.get("oldRepo", {}).get("name", "unknown") # TODO ernsure good key when name change
         new_name = repo.get("name", "unknown")
-        print(f"Model repo renamed from {old_name} to {new_name}.")
+        message = f"Model renamed: `{old_name}` → `{new_name}`"
+        send_slack_message(message)
     elif action == "delete":
         name = repo.get("name", "unknown")
-        print(f"Model repo {name} was deleted.")
-    return Response("Webhook received!", status_code=200)
+        message = f"Model deleted: `{name}`"
+        send_slack_message(message)
+    return Response("Webhoo notification received and processed!", status_code=200)

requirements.txt

@@ -1,2 +1,3 @@
 fastapi
-uvicorn
+uvicorn
+requests