From c4c4e60acf7eecf1b46053bb914779c84d286aff Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Lukas=20B=C3=BCrgi?= <buergi.lukas@gmail.com>
Date: Mon, 27 Jan 2025 12:13:07 +0100
Subject: [PATCH] fixed b_command (and raise error if -x is given but invalid)

---
 rembg/commands/b_command.py | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/rembg/commands/b_command.py b/rembg/commands/b_command.py
index 21fc64a..6fb1fa6 100644
--- a/rembg/commands/b_command.py
+++ b/rembg/commands/b_command.py
@@ -6,7 +6,7 @@ import sys
 from typing import IO
 
 import click
-from PIL.Image import Image as PILImage
+import PIL
 
 from ..bg import remove
 from ..session_factory import new_session
@@ -118,10 +118,11 @@ def b_command(
     Returns:
         None
     """
-    try:
-        kwargs.update(json.loads(extras))
-    except Exception:
-        pass
+    if extras:
+        try:
+            kwargs.update(json.loads(extras))
+        except Exception:
+            raise click.BadParameter("extras must be a valid JSON string")
 
     session = new_session(model, **kwargs)
     bytes_per_img = image_width * image_height * 3
@@ -134,7 +135,7 @@ def b_command(
         if not os.path.isdir(output_dir):
             os.makedirs(output_dir, exist_ok=True)
 
-    def img_to_byte_array(img: PILImage) -> bytes:
+    def img_to_byte_array(img: PIL.Image) -> bytes:
         buff = io.BytesIO()
         img.save(buff, format="PNG")
         return buff.getvalue()
@@ -162,7 +163,7 @@ def b_command(
                 if not img_bytes:
                     break
 
-                img = PILImage.frombytes("RGB", (image_width, image_height), img_bytes)
+                img = PIL.Image.frombytes("RGB", (image_width, image_height), img_bytes)
                 output = remove(img, session=session, **kwargs)
 
                 if output_specifier: