Update app.py

app.py

@@ -1,14 +1,73 @@
 import gradio as gr
+from huggingface_hub import create_repo, whoami
+import subprocess
+import os, shutil
 
-def fork(source_repo, token, type):
-  return token
+def fork(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)
+    org = None
+    if repo_namespace != username:
+      org = repo_namespace
+
+    # Create the destination repo
+    if repo_type in ["spaces", "dataset"]:
+        # For some reason create_repo does not allow repo_type="model"..., even if documentation says
+        # that's the default.
+        create_repo(dst_id, token=token, organization=org, repo_type=repo_type, space_sdk="static")
+    else:
+        create_repo(dst_id, token=token, organization=org)
+    
+    # Clone source repo
+    endpoint = "https://huggingface.co/"
+    if repo_type in ["spaces", "dataset"]:
+        endpoint += repo_type
+    full_path = endpoint + "/" + source_repo
+    local_dir = "hub/" + source_repo
+    repo = Repository(local_dir=local_dir, clone_from=full_path)
+    
+    # Change remote origin
+    command = f"git remote set-url origin https://user:{token}@huggingface.co/"
+    if repo_type in ["spaces", "dataset"]:
+      command += repo_type
+    command += dst_repo
+    subprocess.run(
+        command.split(),
+        stderr=subprocess.PIPE,
+        stdout=subprocess.PIPE,
+        encoding="utf-8",
+        check=True,
+        cwd=local_dir,
+    )
+    
+    # Push!
+    subprocess.run(
+        "git push --force origin main".split(),
+        stderr=subprocess.PIPE,
+        stdout=subprocess.PIPE,
+        encoding="utf-8",
+        check=True,
+        cwd=local_dir,
+    )
+
+    # Clean up
+    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 endpoint + "/" + dst_repo
 
 interface = gr.Interface(
     fn=fork,
     inputs=[
-      gr.inputs.Textbox(placeholder="Source repository"),
-      gr.inputs.Textbox(placeholder="Write access token"),
-      gr.inputs.Dropdown(choices=["model", "dataset", "space"])
+        gr.inputs.Textbox(placeholder="Source repository"),
+        gr.inputs.Textbox(placeholder="Destination repository name"),
+        gr.inputs.Textbox(placeholder="Write access token"),
+        gr.inputs.Dropdown(choices=["model", "dataset", "space"])
     ],
     outputs=["textbox"],
     allow_flagging=False,