From 40df25866106116365ad9aa548377ef75e003efd Mon Sep 17 00:00:00 2001 From: Yizhan Date: Tue, 2 Sep 2025 23:00:36 -0700 Subject: [PATCH] fix: Ensure default background is created when no images are found - Added a warning log when no background images are loaded. - Implemented creation of a default white background if the background images list is empty. --- text_renderer/bg_manager.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/text_renderer/bg_manager.py b/text_renderer/bg_manager.py index 025224a3..ef48bf58 100644 --- a/text_renderer/bg_manager.py +++ b/text_renderer/bg_manager.py @@ -30,7 +30,12 @@ def __init__(self, bg_dir: Path = None, pre_load: bool = True): if pre_load: self.bg_imgs.append(self._get_bg(str(p))) - assert len(self.bg_imgs) != 0, "background image is empty" + if len(self.bg_imgs) == 0: + logger.warning("No background images found. Creating a default white background.") + # Create a default white background + default_bg = Image.new('RGB', (800, 600), (255, 255, 255)) + self.bg_imgs = [default_bg] + self.bg_paths = ['default_white'] def _is_transparent_image(self, p: Path): pil_img: PILImage = Image.open(p)