Upload utils.py

utils.py

@@ -151,7 +151,7 @@ def download_thing(directory, url, civitai_api_key="", progress=gr.Progress(trac
 
 def get_local_file_list(dir_path):
     file_list = []
-    for file in Path(dir_path).glob("**/*.*"):
+    for file in Path(dir_path).glob("*/*.*"):
         if file.is_file():
             file_path = str(file)
             file_list.append(file_path)
@@ -197,17 +197,13 @@ def download_repo(repo_id: str, dir_path: str, progress=gr.Progress(track_tqdm=T
         return False
 
 
-def upload_repo(repo_id: str, dir_path: str, is_private: bool, progress=gr.Progress(track_tqdm=True)): # for diffusers repo
+def upload_repo(repo_id: str, dir_path: str, is_private: bool, is_pr: bool=False, progress=gr.Progress(track_tqdm=True)): # for diffusers repo
     hf_token = get_token()
     api = HfApi(token=hf_token)
     try:
         progress(0, desc="Start uploading...")
         api.create_repo(repo_id=repo_id, token=hf_token, private=is_private, exist_ok=True)
-        for path in Path(dir_path).glob("*"):
-            if path.is_dir():
-                api.upload_folder(repo_id=repo_id, folder_path=str(path), path_in_repo=path.name, token=hf_token)
-            elif path.is_file():
-                api.upload_file(repo_id=repo_id, path_or_fileobj=str(path), path_in_repo=path.name, token=hf_token)
+        api.upload_folder(repo_id=repo_id, folder_path=dir_path, path_in_repo="", create_pr=is_pr, token=hf_token)
         progress(1, desc="Uploaded.")
         return get_hf_url(repo_id, "model")
     except Exception as e:
@@ -215,6 +211,18 @@ def upload_repo(repo_id: str, dir_path: str, is_private: bool, progress=gr.Progr
         return ""
 
 
+def gate_repo(repo_id: str, gated_str: str, repo_type: str="model"):
+    hf_token = get_token()
+    api = HfApi(token=hf_token)
+    try:
+        if gated_str == "auto": gated = "auto"
+        elif gated_str == "manual": gated = "manual"
+        else: gated = False
+        api.update_repo_settings(repo_id=repo_id, gated=gated, repo_type=repo_type, token=hf_token)
+    except Exception as e:
+        print(f"Error: Failed to update settings {repo_id}. {e}")
+
+
 HF_SUBFOLDER_NAME = ["None", "user_repo"]