add oauth functionality

app.py

@@ -86,6 +86,7 @@ with gr.Blocks() as app:
                         dataset_name_input = gr.Textbox(
                             label="Dataset Repo ID", value=dataset.load_dataset_name()
                         )
+                        gr.LoginButton()
                     with gr.Column():
                         split_input = gr.Dropdown(
                             label="Dataset Split",

src/spaces.py

@@ -1,12 +1,16 @@
+import gradio as gr
 from huggingface_hub import duplicate_space
 
 
-def create_argilla_space(target_argilla_space):
+def create_argilla_space(target_argilla_space, oauth_token: gr.OAuthToken | None):
+    if oauth_token is None:
+        return "Please log in to create an Argilla space."
     url = duplicate_space(
         from_id="argilla/argilla-template-space",
         to_id=target_argilla_space,
         private=False,
         exist_ok=True,
+        token=oauth_token.token,
     )
     return repo_url_to_embed_url(url)