Spaces:
Running
Running
| #!/usr/bin/env python | |
| import json | |
| import gradio as gr | |
| from gradio_space_ci import enable_space_ci | |
| from huggingface_hub import CommitScheduler, hf_hub_download | |
| from open_pr import open_pr | |
| enable_space_ci() | |
| TITLE = "Listen to Pull Requests and start ephemeral Spaces on new PRs! 🚀" | |
| DESCRIPTION = """ | |
| ## Gradio Space CI is a tool to create ephemeral Spaces for each PR opened on your Space repo. | |
| The goal is to improve developer experience by making the review process as lean as possible. | |
| The app below let you open a PR to enable Space CI on a Space. | |
| The steps are the following: | |
| 1. Sign-in with Hugging Face. | |
| 2. Input a Space id from the Hub | |
| 3. Click "Submit" | |
| 4. That's it! You'll get feedback if it works or not, and if it worked, you'll get the URL of the opened PR 🔥 | |
| #### ⚠️ Disclaimer: **Gradio Space CI** works only on public Spaces. | |
| For more details about **Gradio Space CI**, checkout [this page](https://huggingface.co/spaces/Wauplin/gradio-space-ci/blob/main/README.md). | |
| If you find any issues, please report here: https://huggingface.co/spaces/Wauplin/gradio-space-ci/discussions | |
| """ | |
| SUCCESS_MESSAGE = """ | |
| ### Success 🔥 | |
| Yay! A PR has been open to enable Space CI on {space_id}. Check it out here: [{pr_url}]({pr_url}). | |
| You can contact the Space owner to let them know about this PR. | |
| """ | |
| ERROR_MESSAGE = """ | |
| ### Error ❌ | |
| An error happened while trying to open a PR to enable Space CI on {space_id}. | |
| {error}""" | |
| DATASET_ID = "Wauplin/gradio-space-ci-report" | |
| csv_path = hf_hub_download(repo_id=DATASET_ID, repo_type="dataset", filename="report.jsonl", local_dir="./report") | |
| scheduler = CommitScheduler(repo_id=DATASET_ID, repo_type="dataset", folder_path="./report") | |
| def append(**kwargs): | |
| with scheduler.lock: | |
| with open(csv_path, "a") as f: | |
| f.write(json.dumps(kwargs) + "\n") | |
| def fn(space_id_or_url: str, oauth_profile: gr.OAuthProfile | None, oauth_token: gr.OAuthToken | None) -> str: | |
| user = oauth_profile.username if oauth_profile is not None else "???" | |
| try: | |
| pr_url = open_pr(space_id_or_url, oauth_token) | |
| append(space_id=space_id_or_url, status="success", pr_url=pr_url, user=user) | |
| return SUCCESS_MESSAGE.format(space_id=space_id_or_url, pr_url=pr_url) | |
| except Exception as e: | |
| append(space_id=space_id_or_url, status="error", error=str(e), user=user) | |
| return ERROR_MESSAGE.format(space_id=space_id_or_url, error=str(e)) | |
| with gr.Blocks() as blocks: | |
| gr.Markdown(f"<h1 style='text-align: center; margin-bottom: 1rem'>{TITLE}</h1>") | |
| gr.Markdown(DESCRIPTION) | |
| with gr.Row(equal_height=False): | |
| with gr.Column(): | |
| space_id = gr.Textbox(label="Space ID or URL", lines=1) | |
| gr.LoginButton() | |
| submit_btn = gr.Button("Submit", variant="primary") | |
| with gr.Column(): | |
| output = gr.Markdown() | |
| submit_btn.click(fn, space_id, output) | |
| blocks.launch() | |