Update roop/utilities.py

roop/utilities.py

@@ -132,5 +132,31 @@ def is_video(video_path: str) -> bool:
         return bool(mimetype and mimetype.startswith('video/'))
     return False
 
+
+
+def conditional_download(download_directory_path: str, urls: List[str]) -> None:
+    if not os.path.exists(download_directory_path):
+        os.makedirs(download_directory_path)
+    for url in urls:
+        download_file_path = os.path.join(download_directory_path, os.path.basename(url))
+        if not os.path.exists(download_file_path):
+            response = requests.get(url, stream=True)
+            total = int(response.headers.get('Content-Length', 0))
+            with open(download_file_path, 'wb') as file, tqdm(
+                    total=total,
+                    unit='B',
+                    unit_scale=True,
+                    unit_divisor=1024,
+                    desc='Downloading'
+            ) as progress:
+                for data in response.iter_content(chunk_size=1024):
+                    file.write(data)
+                    progress.update(len(data))
+
+
+
+
+
+
 def resolve_relative_path(path: str) -> str:
     return os.path.abspath(os.path.join(os.path.dirname(__file__), path))