add notifications

src/gradio_space_ci/webhook.py

@@ -437,7 +437,7 @@ def delete_ci_space(space_id: str, pr_num: int) -> None:
 def notify_pr(
     space_id: str,
     pr_num: int,
-    action: Literal["created_not_configured", "created_and_configured", "updated", "deleted"],
+    action: Literal["created_not_configured", "created_and_configured", "updated", "deleted","trusted_pr","untrusted_pr"],
 ) -> None:
     ci_space_id = _get_ci_space_id(space_id=space_id, pr_num=pr_num)
     if action == "created_not_configured":
@@ -448,6 +448,10 @@ def notify_pr(
         comment = NOTIFICATION_TEMPLATE_UPDATED.format(ci_space_id=ci_space_id)
     elif action == "deleted":
         comment = NOTIFICATION_TEMPLATE_DELETED
+    elif action == "trusted_pr": 
+        comment = NOTIFICATION_TEMPLATE_TRUSTED_PR
+    elif action == "untrusted_pr":
+        comment = NOTIFICATION_TEMPLATE_UNTRUSTED_PR
     else:
         raise ValueError(f"Status {action} not handled.")
 
@@ -511,13 +515,12 @@ def handle_command(space_id: str, payload: WebhookPayload) -> None:
     event_author = details.events[-1]._event["author"]["name"]  # username of that event
     if event_author in EPHEMERAL_SPACES_CONFIG["trusted_authors"]:
         if payload.comment.content == "/trust_pr":
-            print("trusting pr ...")
             set_config(space_id=space_id, pr_num=pr_num)
-            print("pr has been trusted")
+            notify_pr(space_id=space_id,pr_num=pr_num,action="trusted_pr")
         elif payload.comment.content == "/untrust_pr":
-            print("untrusting pr ...")
             unset_config(space_id=space_id, pr_num=pr_num)
-            print("pr has been untrusted")
+            notify_pr(space_id=space_id,pr_num=pr_num,action="untrusted_pr")
+            
 
 
 NOTIFICATION_TEMPLATE_CREATED_AND_CONFIGURED = """\
@@ -542,6 +545,17 @@ PR is now merged/closed. The ephemeral Space has been deleted.
 _(This is an automated message.)_
 """
 
+NOTIFICATION_TEMPLATE_TRUSTED_PR = """\
+This PR has been granted temporary trust status Thus granting it with the appropriate approriate hardware, storage, and secrets.
+Trust status will be revokeduser either when a trusted author uses `/untrust_pr` command or when new commits are pushed to this PR.
+_(This is an automated message.)_
+"""
+
+NOTIFICATION_TEMPLATE_UNTRUSTED_PR = """\
+This PR has been untrusted. Thus revoking it with the appropriate hardware, storage, and secrets.
+_(This is an automated message.)_
+"""
+
 ### TO MOVE TO ITS OWN MODULE
 # Taken from https://github.com/huggingface/huggingface_hub/issues/1808#issuecomment-1802341663