Upload 2 files

utils.py

@@ -149,9 +149,10 @@ def download_thing(directory, url, civitai_api_key="", progress=gr.Progress(trac
         print(f"Failed to download: {e}")
 
 
-def get_local_file_list(dir_path):
+def get_local_file_list(dir_path, recursive=False):
     file_list = []
-    for file in Path(dir_path).glob("*/*.*"):
+    pattern = "**/*.*" if recursive else "*/*.*"
+    for file in Path(dir_path).glob(pattern):
         if file.is_file():
             file_path = str(file)
             file_list.append(file_path)
@@ -171,9 +172,10 @@ def get_download_file(temp_dir, url, civitai_key, progress=gr.Progress(track_tqd
             new_file = f"{temp_dir}/{url.split('/')[-1]}"
         else:
             print(f"Start downloading: {url}")
-            before = get_local_file_list(temp_dir)
+            recursive = False if "huggingface.co" in url else True
+            before = get_local_file_list(temp_dir, recursive)
             download_thing(temp_dir, url.strip(), civitai_key)
-            after = get_local_file_list(temp_dir)
+            after = get_local_file_list(temp_dir, recursive)
             new_file = list_sub(after, before)[0] if list_sub(after, before) else ""
         if not new_file:
             print(f"Download failed: {url}")