🔧 fix need of pil dep

shroominic · May 17, 2024 · 2b5fec2 · 2b5fec2
1 parent 7c860c9
commit 2b5fec2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/funcchain/utils/image.py b/src/funcchain/utils/image.py
@@ -8,8 +8,10 @@
 
 if TYPE_CHECKING:
     from PIL.Image import Image as PImage
+    from PIL.Image import open as pil_open
 else:
     PImage = type("PImage")
+    pil_open = lambda x: x  # noqa
 
 
 def image_to_base64_url(image: Image) -> str:
@@ -28,9 +30,7 @@ def pillow_image_to_base64_url(image: PImage) -> str:
 
 
 def base64_url_to_pillow_image(base64_url: str) -> PImage:
-    from PIL.Image import Image as PImage
-
     base64_image = base64_url.split(",")[1]
     image_bytes = b64decode(base64_image)
-    image = PImage.open(BytesIO(image_bytes))
+    image = pil_open(BytesIO(image_bytes))
     return image