Upload app.py

app.py

@@ -3,6 +3,8 @@ import requests
 from pathlib import Path
 import re
 import os
+import tempfile
+import shutil
 from huggingface_hub import whoami, HfApi, hf_hub_download
 from huggingface_hub.utils import build_hf_headers, hf_raise_for_status
 from gradio_huggingfacehub_search import HuggingfaceHubSearch
@@ -23,7 +25,10 @@ def duplicate(source_repo, dst_repo, repo_type, private, overwrite, auto_dir, oa
             raise ValueError("need to select valid repo type")
         _ = whoami(oauth_token.token)
         # ^ this will throw if token is invalid
-
+    except Exception as e:
+        raise gr.Error(f"""Oops, you forgot to login. Please use the loggin button on the top left to migrate your repo {e}""")
+    
+    try:
         if re.fullmatch(r'^[\w_\-\.]+/[\w_\-\.]+$', dst_repo): subfolder = ""
         else:
             dst_repo, subfolder = re.findall(r'^([\w_\-\.]+/[\w_\-\.]+)/?(.+)?$', dst_repo)[0]
@@ -31,9 +36,10 @@ def duplicate(source_repo, dst_repo, repo_type, private, overwrite, auto_dir, oa
         if auto_dir: subfolder = source_repo
 
         if overwrite and api.repo_exists(repo_id=dst_repo, repo_type=repo_type, token=hf_token) or subfolder:
+            temp_dir = tempfile.mkdtemp()
             api.create_repo(repo_id=dst_repo, repo_type=repo_type, private=private, exist_ok=True, token=hf_token)
             for path in api.list_repo_files(repo_id=source_repo, repo_type=repo_type, token=hf_token):
-                file = hf_hub_download(repo_id=source_repo, filename=path, repo_type=repo_type, token=hf_token)
+                file = hf_hub_download(repo_id=source_repo, filename=path, repo_type=repo_type, local_dir=temp_dir, token=hf_token)
                 if not Path(file).exists(): continue
                 if Path(file).is_dir(): # unused for now
                     api.upload_folder(repo_id=dst_repo, folder_path=file, path_in_repo=f"{subfolder}/{path}" if subfolder else path, repo_type=repo_type, token=hf_token)
@@ -43,6 +49,7 @@ def duplicate(source_repo, dst_repo, repo_type, private, overwrite, auto_dir, oa
             if repo_type == "dataset": repo_url = f"https://huggingface.co/datasets/{dst_repo}"
             elif repo_type == "space": repo_url = f"https://huggingface.co/spaces/{dst_repo}"
             else: repo_url = f"https://huggingface.co/{dst_repo}"
+            shutil.rmtree(temp_dir)
         else:
             r = requests.post(
                 f"{ENDPOINT}/api/{repo_type}s/{source_repo}/duplicate",
@@ -59,7 +66,7 @@ def duplicate(source_repo, dst_repo, repo_type, private, overwrite, auto_dir, oa
         )
 
     except Exception as e:
-        raise gr.Error(f"""Oops, you forgot to login. Please use the loggin button on the top left to migrate your repo {e}""")
+        print(e)
 
 interface = gr.Interface(
     fn=duplicate,