Update app.py

app.py

@@ -3,9 +3,8 @@ import logging
 import time
 import gradio as gr
 import datasets
-from huggingface_hub import snapshot_download, WebhooksServer, WebhookPayload
+from huggingface_hub import snapshot_download, WebhooksServer, WebhookPayload, RepoCard
 from gradio_leaderboard import Leaderboard, ColumnFilter, SelectColumns
-from gradio_space_ci import enable_space_ci
 
 from src.display.about import (
     CITATION_BUTTON_LABEL,
@@ -44,8 +43,7 @@ from src.tools.plots import create_metric_plot_obj, create_plot_df, create_score
 # Configure logging
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
 
-# Start ephemeral Spaces on PRs (see config in README.md)
-enable_space_ci()
+
 
 
 def restart_space():
@@ -292,10 +290,41 @@ with demo:
 
 demo.queue(default_concurrency_limit=40)
 
-# Add webhooks
-app = WebhooksServer(ui=demo)
+# Start ephemeral Spaces on PRs (see config in README.md)
+from gradio_space_ci.webhook import IS_EPHEMERAL_SPACE, SPACE_ID, configure_space_ci
+
+
+
+def enable_space_ci_and_return_server(ui: gr.Blocks) -> WebhooksServer:
+    # Taken from https://huggingface.co/spaces/Wauplin/gradio-space-ci/blob/075119aee75ab5e7150bf0814eec91c83482e790/src/gradio_space_ci/webhook.py#L61
+    # Compared to original, this one do not monkeypatch Gradio which allows us to define more webhooks.
+    if SPACE_ID is None:
+        print("Not in a Space: Space CI disabled.")
+        return WebhooksServer(ui=demo)
 
-@app.add_webhook
+    if IS_EPHEMERAL_SPACE:
+        print("In an ephemeral Space: Space CI disabled.")
+        return WebhooksServer(ui=demo)
+
+    card = RepoCard.load(repo_id_or_path=SPACE_ID, repo_type="space")
+    config = card.data.get("space_ci", {})
+    print(f"Enabling Space CI with config from README: {config}")
+
+    return configure_space_ci(
+        blocks=self,
+        trusted_authors=config.get("trusted_authors"),
+        private=config.get("private", "auto"),
+        variables=config.get("variables", "auto"),
+        secrets=config.get("secrets"),
+        hardware=config.get("hardware"),
+        storage=config.get("storage"),
+    )
+
+# Create webhooks server (with CI url if in Space and not ephemeral)
+webhooks_server = enable_space_ci_and_return_server(ui=demo)
+
+# Add webhooks
+@webhooks_server.add_webhook
 async def update_leaderboard(payload: WebhookPayload) -> None:
     if payload.repo.type == "dataset" and payload.event.action == "update":
         leaderboard_dataset = datasets.load_dataset(AGGREGATED_REPO, "default", split="train", cache_dir=HF_HOME)
@@ -306,7 +335,7 @@ async def update_leaderboard(payload: WebhookPayload) -> None:
         )
         leaderboard.value = leaderboard_df
 
-@app.add_webhook    
+@webhooks_server.add_webhook    
 async def update_queue(payload: WebhookPayload) -> None:
     if payload.repo.type == "dataset" and payload.event.action == "update":
         download_dataset(QUEUE_REPO, EVAL_REQUESTS_PATH)
@@ -317,5 +346,4 @@ async def update_queue(payload: WebhookPayload) -> None:
         running_eval_table.value = running_eval_queue_df
         pending_eval_table.value = pending_eval_queue_df
 
-
 app.launch()