Revamp app.py to use exposed API

app.py

@@ -1,61 +1,63 @@
 import gradio as gr
-from huggingface_hub import create_repo, upload_file, Repository, whoami
-import subprocess
-import os, shutil
+from huggingface_hub import whoami
+import requests
+from requests import HTTPError, JSONDecodeError
+
+
+ENDPOINT = "https://huggingface.co"
+# ENDPOINT = "http://localhost:5564"
+
+REPO_TYPES = ["model", "dataset", "space"]
+
 
 def duplicate(source_repo, dst_repo, token, repo_type):
-    # Creating repos has inconsistent API (https://github.com/huggingface/huggingface_hub/issues/47)
-    repo_namespace, dst_id = dst_repo.split("/")
-    username = whoami(token)["name"]
-    org = None
-    if repo_namespace != username:
-      org = repo_namespace
+    try:
+        if not repo_type in REPO_TYPES:
+            raise ValueError("need to select valid repo type")
+        _ = whoami(token)
+        # ^ this will throw if token is invalid
 
-    # Create the destination repo
-    if repo_type in ["space", "dataset"]:
-        # For some reason create_repo does not allow repo_type="model"..., even if documentation says
-        # that's the default.
-        url = create_repo(dst_id, token=token, organization=org, repo_type=repo_type, space_sdk="gradio", private=False)
-    else:
-        url = create_repo(dst_id, token=token, organization=org, private=False)
-    
-    # Clone source repo
-    endpoint = "huggingface.co/"
-    if repo_type in ["space", "dataset"]:
-        endpoint += repo_type + "/"
-    full_path = f"https://{username}:{token}@{endpoint}{source_repo}"
-    local_dir = "hub/" + source_repo
-    
-    if repo_type in ["space", "dataset"]:
-        # Same as above
-        repo = Repository(local_dir=local_dir, clone_from=full_path, repo_type=repo_type)
-    else:
-        repo = Repository(local_dir=local_dir, clone_from=full_path)
-    
-    for root, dirs, files in  os.walk(local_dir):
-      if not root.startswith("."):
-        if repo_type == "model":
-          repo_type = None
-        for f in files:
-            if not f.startswith("."):
-                if ".git" not in root:
-                    # remove hub/namespace/reponame
-                    directory_path_in_repo = "/".join(root.split("/")[3:]) 
-                    path_in_repo = os.path.join(directory_path_in_repo, f)
-                    local_file_path = os.path.join(local_dir, path_in_repo)
-                    print("From: ", local_file_path, " to: ", path_in_repo)
-                    upload_file(path_or_fileobj=local_file_path, path_in_repo=path_in_repo, repo_id=dst_repo, token=token, repo_type=repo_type)
+        r = requests.post(
+            f"{ENDPOINT}/api/{repo_type}s/{source_repo}/duplicate",
+            headers={"authorization": f"Bearer {token}"},
+            json={
+                "repository": dst_repo,
+            },
+        )
 
+        ## Following lines:
+        ## TODO(wauplin)
+        ## Expose `hf_raise_for_status` from huggingface_hub
+        ## or even provide a helper to do any API request to hf.co passing just the path and json payload?
+        try:
+            r.raise_for_status()
+        except HTTPError as e:
+            try:
+                server_data = r.json()
+            except JSONDecodeError:
+                server_data = {}
+            server_message_from_body = server_data.get("error")
+            if server_message_from_body:
+                raise ValueError(server_message_from_body)
+            raise e
+
+        repo_url = r.json().get("url")
+
+        return (
+            f'Find your repo <a href=\'{repo_url}\' target="_blank" style="text-decoration:underline">here</a>',
+            "sp.jpg",
+        )
+
+    except Exception as e:
+        return (
+            f"""
+        ### Error 😢😢😢
+        
+        {e}
+        """,
+            None,
+        )
 
-    # Clean up to be nice with the environment
-    for filename in os.listdir(local_dir):
-      file_path = os.path.join(local_dir, filename)
-      if os.path.isfile(file_path) or os.path.islink(file_path):
-          os.unlink(file_path)
-      elif os.path.isdir(file_path):
-          shutil.rmtree(file_path)
-    
-    return f"Find your repo <a href='{url}' target=\"_blank\" style=\"text-decoration:underline\">here</a>", "sp.jpg"
 
 interface = gr.Interface(
     fn=duplicate,
@@ -63,13 +65,16 @@ interface = gr.Interface(
         gr.Textbox(placeholder="Source repository (e.g. osanseviero/src)"),
         gr.Textbox(placeholder="Destination repository (e.g. osanseviero/dst)"),
         gr.Textbox(placeholder="Write access token"),
-        gr.Dropdown(choices=["model", "dataset", "space"])
+        gr.Dropdown(choices=REPO_TYPES, value="model"),
+    ],
+    outputs=[
+        gr.Markdown(label="output"),
+        gr.Image(show_label=False),
     ],
-    outputs=["html", "image"] ,
     title="Duplicate your repo!",
     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'>tokens settings</a></p>",
     allow_flagging="never",
     live=False,
 )
-interface.launch(enable_queue=True)
+interface.launch(enable_queue=True)