ui.py

import gradio as gr
from enum import Enum
from database import is_space_existing, is_space_registered, update_status


TITLE = "⚙️ Spaces CI Bot ⚙️"
DESCRIPTION = """
Welcome! This app lets you register a Space on the CI Bot's watchlist.

Once your Space is on the list, any PR opened will be deployed as an ephemeral Space. Any changes pushed to the PRs will trigger a re-deployment. Once the PR is merged, the ephemeral Space is
deleted.

If your Space requires configuration (secrets or upgraded hardware), you must duplicate the ephemeral Space to your account and configure the settings by yourself. You are responsible of making sure that the changes introduced in the PR are not harmful (leak secrets, run malicious scripts,...).
"""


class Action(Enum):
    REGISTER = "Enable CI Bot"
    UNREGISTER = "Disable CI Bot"
    CHECK_STATUS = "Check status"


def gradio_fn(space_id: str, action: str) -> str:
    if not is_space_existing(space_id):
        return f"""## Error
        Could not find Space '**{space_id}**' on the Hub.
        Please make sure you are trying to register a public repository.
        """

    registered = is_space_registered(space_id)
    if action == Action.REGISTER.value:
        if registered:
            return f"""## Did nothing
            The Space '**{space_id}**' is already in the watchlist. Any PR opened on
            this repository will trigger an ephemeral Space.
            """
        else:
            update_status(space_id, should_watch=True)
            return f"""## Success
            The Space '**{space_id}**' has been added to the watchlist. Any PR opened on
            this repository will trigger an ephemeral Space.
            """
    elif action == Action.UNREGISTER.value:
        if not registered:
            return f"""## Did nothing
            The Space '**{space_id}**' is currently not in the watchlist.
            """
        else:
            update_status(space_id, should_watch=False)
            return f"""## Success
            The Space '**{space_id}**' has been removed from the watchlist.
            """
    elif action == Action.CHECK_STATUS.value:
        if registered:
            return f"""## Watched
            The Space '**{space_id}**' is already in the watchlist. Any PR opened on
            this repository will trigger an ephemeral Space.
            """
        else:
            return f"""## Not watched
            The Space '**{space_id}**' is currently not in the watchlist.
            """
    else:
        return f"**Error:** action {action} not implemented."


def generate_ui() -> gr.Blocks:
    return gr.Interface(
        fn=gradio_fn,
        inputs=[
            gr.Textbox(lines=1, placeholder="username/my_cool_space", label="Space ID"),
            gr.Radio(
                [action.value for action in Action],
                value=Action.REGISTER.value,
                label="What should I do?",
            ),
        ],
        outputs=[gr.Markdown()],
        title=TITLE,
        description=DESCRIPTION,
        allow_flagging="never",
    )