Update to handle background images sent as url

app.py

@@ -34,7 +34,7 @@ import modules.user_history
 from modules.version_info import versions_html, commit_hash, get_xformers_version
 from modules.gradio import *
 from modules.file_utils import get_file_parts, get_filename_from_filepath, convert_title_to_filename, get_unique_file_path, delete_file, download_and_save_image
-from modules.constants import IS_SHARED_SPACE, HF_REPO_ID
+from modules.constants import IS_SHARED_SPACE, HF_REPO_ID, TMPDIR, HF_API_TOKEN
 from modules.storage import upload_files_to_repo
 
 MODEL = None
@@ -299,8 +299,8 @@ def predict(model, text, melody_filepath = None, duration=10, dimension=2, topk=
         background = load_background_filepath(video_orientation)
 
     if background.startswith("http://") or background.startswith("https://"):
-        username = profile.value.username if hasattr(profile.value, 'username') else "default_user" if (profile is None) else profile
-        background = download_and_save_image(background, modules.user_history._user_images_path(username))
+        username = profile if isinstance(profile, str) else profile.value.username if hasattr(profile.value, 'username') else "default_user" if (profile is None) else profile
+        background = download_and_save_image(background, Path(TMPDIR) / str(username), HF_API_TOKEN)
     
     if melody_filepath:
         melody_name, melody_extension = get_filename_from_filepath(melody_filepath)

modules/file_utils.py

@@ -120,18 +120,27 @@ def get_unique_file_path(directory, filename, file_ext, counter=0):
 # Example usage:
 # new_file_path = get_unique_file_path(video_dir, title_file_name, video_new_ext)
 
-def download_and_save_image(url: str, dst_folder: Path) -> Path:
+def download_and_save_image(url: str, dst_folder: Path, token: str = None) -> Path:
     """
-    Downloads an image from a URL, verifies it with PIL, and saves it in dst_folder with a unique filename.
-    
+    Downloads an image from a URL with authentication if a token is provided,
+    verifies it with PIL, and saves it in dst_folder with a unique filename.
+
     Args:
         url (str): The image URL.
         dst_folder (Path): The destination folder for the image.
-        
+        token (str, optional): A valid Bearer token. If not provided, the HF_API_TOKEN
+                              environment variable is used if available.
+
     Returns:
         Path: The saved image's file path.
     """
-    response = requests.get(url)
+    headers = {}
+    # Use provided token; otherwise, fall back to environment variable.
+    api_token = token
+    if api_token:
+        headers["Authorization"] = f"Bearer {api_token}"
+    
+    response = requests.get(url, headers=headers)
     response.raise_for_status()
     pil_image = Image.open(BytesIO(response.content))
     
@@ -142,5 +151,6 @@ def download_and_save_image(url: str, dst_folder: Path) -> Path:
     # Use get_unique_file_path from file_utils.py to generate a unique file path.
     unique_filepath_str = get_unique_file_path(str(dst_folder), base, ext)
     dst = Path(unique_filepath_str)
+    dst_folder.mkdir(parents=True, exist_ok=True)
     pil_image.save(dst)
     return dst