UI

app.py

@@ -7,6 +7,8 @@ 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.
 
@@ -26,17 +28,20 @@ For more details about **Gradio Space CI**, checkout [this page]](https://huggin
 If you find any issues, please report here: https://huggingface.co/spaces/Wauplin/gradio-space-ci/discussions
 """
 
-interface = gr.Interface(
-    fn=open_pr,
-    inputs=["text", gr.LoginButton(), gr.LogoutButton()],
-    outputs="markdown",
-    title="Listen to Pull Requests and start ephemeral Spaces on new PRs! 🚀",
-    description=DESCRIPTION,
-    article="Provide a space ID and a token and click 'Submit'. A PR will be open to Space CI on the provided Space. You don't have to be the Space owner to open the PR.",
-    allow_flagging=False,
-    allow_duplication=False,
-)
-
-with gr.Blocks() as demo:
-    interface.render()
-demo.launch()
+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()
+            gr.LogoutButton()
+            submit_btn = gr.Button("Submit", variant="primary")
+
+        with gr.Column():
+            output = gr.Markdown()
+
+        submit_btn.click(open_pr, space_id, output)
+
+blocks.launch()

open_pr.py

@@ -42,7 +42,7 @@ You can contact the Space owner to let them know about this PR.
 """
 
 
-def open_pr(space_id_or_url: str, _1, _2, oauth_token: gr.OAuthToken | None) -> str:
+def open_pr(space_id_or_url: str, oauth_token: gr.OAuthToken | None) -> str:
     if oauth_token is None:
         raise gr.Error("You must be logged in to open a PR. Click on 'Sign in with Huggingface' first.")
     token = oauth_token.token