Update app.py

app.py

@@ -1,75 +1,99 @@
 import gradio as gr
 from huggingface_hub import create_repo, upload_file, Repository, whoami
-import subprocess
-import os, shutil
+import os
+import shutil
 
 def duplicate(source_repo, dst_repo, token, repo_type):
-    # Creating repos has inconsistent API (https://github.com/huggingface/huggingface_hub/issues/47)
+    # Get username from token
+    username = whoami(token=token)["name"]
     repo_namespace, dst_id = dst_repo.split("/")
-    username = whoami(token)["name"]
-    org = None
-    if repo_namespace != username:
-      org = repo_namespace
+    org = repo_namespace if repo_namespace != username else None
 
     # 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)
+        url = create_repo(
+            repo_id=dst_id,
+            token=token,
+            organization=org,
+            repo_type=repo_type,
+            space_sdk="gradio" if repo_type == "space" else None,
+            private=False
+        )
     else:
-        url = create_repo(dst_id, token=token, organization=org, private=False)
+        url = create_repo(
+            repo_id=dst_id,
+            token=token,
+            organization=org,
+            private=False
+        )
     
     # Clone source repo
     endpoint = "huggingface.co/"
     if repo_type in ["space", "dataset"]:
-        endpoint += repo_type + "/"
+        endpoint += f"{repo_type}/"
     full_path = f"https://{username}:{token}@{endpoint}{source_repo}"
-    local_dir = "hub/" + source_repo
+    local_dir = f"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)
+    # Clone repository
+    repo = Repository(
+        local_dir=local_dir,
+        clone_from=full_path,
+        repo_type=repo_type if repo_type in ["space", "dataset"] else None,
+        use_auth_token=token
+    )
     
-    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:]) 
+    # Upload files
+    for root, _, files in os.walk(local_dir):
+        if not root.startswith(".") and ".git" not in root:
+            for f in files:
+                if not f.startswith("."):
+                    directory_path_in_repo = "/".join(root.split("/")[2:])
                     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)
+                    upload_file(
+                        path_or_fileobj=local_file_path,
+                        path_in_repo=path_in_repo,
+                        repo_id=dst_repo,
+                        token=token,
+                        repo_type=repo_type if repo_type != "model" else 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)
+    # Clean up
+    shutil.rmtree(local_dir, ignore_errors=True)
     
-    return f"Find your repo <a href='{url}' target=\"_blank\" style=\"text-decoration:underline\">here</a>"
+    return f"Find your repo <a href='{url}' target='_blank' style='text-decoration:underline'>here</a>"
 
+# Updated Gradio interface
 interface = gr.Interface(
     fn=duplicate,
     inputs=[
-        gr.inputs.Textbox(placeholder="Source repository (e.g. osanseviero/src)"),
-        gr.inputs.Textbox(placeholder="Destination repository (e.g. osanseviero/dst)"),
-        gr.inputs.Textbox(placeholder="Write access token"),
-        gr.inputs.Dropdown(choices=["model", "dataset", "space"])
+        gr.Textbox(
+            label="Source repository",
+            placeholder="e.g. osanseviero/src"
+        ),
+        gr.Textbox(
+            label="Destination repository",
+            placeholder="e.g. osanseviero/dst"
+        ),
+        gr.Textbox(
+            label="Write access token",
+            type="password",
+            placeholder="Your HF token"
+        ),
+        gr.Dropdown(
+            choices=["model", "dataset", "space"],
+            label="Repository type",
+            value="model"
+        )
     ],
-    outputs=["html", "image"] ,
+    outputs=gr.HTML(),
     title="Duplicate your repo!",
-    description="Duplicate a Hugging Face repository! You need to specify a write token obtained in https://hf.co/settings/token. This Space is a an experimental demo.",
-    article="<p>Find your write token at <a href='https://huggingface.co/settings/token' target='_blank'>token settings</a></p>",
-    allow_flagging=False,
-    live=False,
+    description="Duplicate a Hugging Face repository! You need to specify a write token obtained from https://hf.co/settings/tokens. This Space is 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"
 )
-interface.launch(enable_queue=True)
+
+interface.launch()