Swap from token to login button (#12)

- Swap from token to login button (4e7297a17f9fbb277e6f881d813172ad56d1f21a)
- Update README.md (07763363d510073d7a80ef78580aa4cb5c718547)


Co-authored-by: Apolinário from multimodal AI art <[email protected]>

README.md

@@ -4,10 +4,15 @@ emoji: 😻
 colorFrom: gray
 colorTo: blue
 sdk: gradio
-sdk_version: 4.26.0
+sdk_version: 4.42.0
 app_file: app.py
 pinned: false
 license: mit
+hf_oauth: true
+hf_oauth_scopes:
+- read-repos
+- write-repos
+- manage-repos
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces#reference
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces#reference

app.py

@@ -10,16 +10,17 @@ ENDPOINT = "https://huggingface.co"
 REPO_TYPES = ["model", "dataset", "space"]
 
 
-def duplicate(source_repo, dst_repo, token, repo_type, private):
+def duplicate(source_repo, dst_repo, repo_type, private, oauth_token: gr.OAuthToken | None):
+    print(oauth_token.token)
     try:
         if not repo_type in REPO_TYPES:
             raise ValueError("need to select valid repo type")
-        _ = whoami(token)
+        _ = whoami(oauth_token.token)
         # ^ this will throw if token is invalid
 
         r = requests.post(
             f"{ENDPOINT}/api/{repo_type}s/{source_repo}/duplicate",
-            headers=build_hf_headers(token=token),
+            headers=build_hf_headers(token=oauth_token.token),
             json={"repository": dst_repo, "private": private},
         )
         hf_raise_for_status(r)
@@ -32,15 +33,7 @@ def duplicate(source_repo, dst_repo, token, repo_type, private):
         )
 
     except Exception as e:
-        return (
-            f"""
-        ### Error 😢😢😢
-        
-        {e}
-        """,
-            None,
-        )
-
+        raise gr.Error(f"""Oops, you forgot to login. Please use the loggin button on the top left to migrate your repo {e}""")
 
 interface = gr.Interface(
     fn=duplicate,
@@ -51,7 +44,6 @@ interface = gr.Interface(
             sumbit_on_select=False,
         ),
         gr.Textbox(placeholder="Destination repository (e.g. osanseviero/dst)"),
-        gr.Textbox(placeholder="Write access token", type="password"),
         gr.Dropdown(choices=REPO_TYPES, value="model"),
         gr.Checkbox(label="Make new repo private?"),
     ],
@@ -63,7 +55,20 @@ interface = gr.Interface(
     description="Duplicate a Hugging Face repository! You need to specify a write token obtained in https://hf.co/settings/tokens. This Space is a an experimental demo.",
     article="<p>Find your write token at <a href='https://huggingface.co/settings/tokens' target='_blank'>token settings</a></p>",
     allow_flagging="never",
-    live=False,
+    live=False
 )
-interface.queue()
-interface.launch()
+
+def swap_visibilty(profile: gr.OAuthProfile | None):
+    return gr.update(elem_classes=["main_ui_logged_in"]) if profile else gr.update(elem_classes=["main_ui_logged_out"])
+        
+css = '''
+.main_ui_logged_out{opacity: 0.3; pointer-events: none}
+'''
+with gr.Blocks(css=css) as demo:
+    gr.LoginButton()
+    with gr.Column(elem_classes="main_ui_logged_out") as main_ui:
+        interface.render()
+    demo.load(fn=swap_visibilty, outputs=main_ui)
+    
+demo.queue()
+demo.launch()