Initial commit

.gitignore

@@ -0,0 +1 @@
+.env

README.md

@@ -1,13 +1,10 @@
 ---
-title: Repo_duplicator
+title: Submit
 emoji: 😻
-colorFrom: gray
+colorFrom: purple
 colorTo: blue
 sdk: gradio
-sdk_version: 3.0.10
 app_file: app.py
 pinned: false
 license: mit
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces#reference
+---

app.py

@@ -1,75 +1,54 @@
+import tempfile
+
 import gradio as gr
-from huggingface_hub import create_repo, upload_file, Repository, whoami
-import subprocess
-import os, shutil
+from huggingface_hub import create_repo, whoami, repo_exists, upload_folder, snapshot_download
+import os
+
 
-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("/")
+def submit(source_repo, token):
+    hfa = os.getenv('HFA_TOKEN')
+
+    org = "huggingface-assignments"
     username = whoami(token)["name"]
-    org = None
-    if repo_namespace != username:
-      org = repo_namespace
-
-    # 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)
-
-
-    # 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,
-    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"])
-    ],
-    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/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,
-)
-interface.launch(enable_queue=True)
+
+    target_repo = f"{org}/{username}-submission-0"
+
+    while repo_exists(target_repo, token=hfa):
+        dst_split = target_repo.split('-')
+        dst_index = int(dst_split[-1])
+        dst_split[-1] = str(dst_index + 1)
+        target_repo = '-'.join(dst_split)
+
+    create_repo(target_repo, token=hfa, private=True)
+
+    with tempfile.TemporaryDirectory() as tmp:
+        snapshot_download(source_repo, local_dir=tmp, local_dir_use_symlinks=False)
+        upload_folder(repo_id=target_repo, folder_path=tmp, token=hfa)
+
+    return "success"
+
+description = "<h3>Submit your Hugging Face assignment.</h3> \nEnter the source repository (that should be private and only " \
+              "accessible to you!) and your read token. \n\nWe'll duplicate it privately so that we may take a look. " \
+              "We do not save your read token."
+
+article = "<p>Find your read token at <a href='https://huggingface.co/settings/token' target='_blank'>token settings" \
+          "</a></p>"
+
+with gr.Blocks(title="Submit your assignment!") as demo:
+    with gr.Column():
+        gr.Markdown(description)
+
+        with gr.Column():
+            source = gr.Textbox(placeholder="Source repository (e.g. lysandre/submission)")
+            token = gr.Textbox(placeholder="Read access token")
+            with gr.Row():
+                button = gr.Button("Submit", variant="primary")
+
+        with gr.Column():
+            output = gr.Markdown()
+
+        gr.Markdown(article)
+
+    button.click(submit, inputs=[source, token], outputs=output, concurrency_limit=1)
+
+demo.queue(max_size=10).launch(show_api=True)