Update utils/image_utils.py

utils/image_utils.py

@@ -5,18 +5,30 @@ from rembg import remove
 
 def background_removal(input_image_path):
     """
-    指定された画像から背景を除去し、透明部分を白背景にブレンドして返す関数
+    指定された画像から背景を除去し、白以外の領域でトリミングした画像を返す関数
     """
     try:
-        input_image = Image.open(input_image_path)
+        input_image = Image.open(input_image_path).convert("RGB")
     except IOError:
         print(f"Error: Cannot open {input_image_path}")
         return None
 
-    # 背景除去処理
-    result = remove(input_image)
+    # 背景除去処理 (白背景のRGB画像として返される)
+    result_image = remove(input_image).convert("RGB")
+
+    # トリミング処理
+    bg = Image.new("RGB", result_image.size, (255, 255, 255))  # 白背景
+    diff = ImageChops.difference(result_image, bg)
+    bbox = diff.getbbox()  # 白以外の部分を含む境界ボックスを取得
+
+    if bbox:
+        cropped_image = result_image.crop(bbox)
+    else:
+        cropped_image = result_image  # 全部が白の場合はそのまま返す
+
+    # 結果を保存
     result_path = "tmp.png"
-    result.save(result_path)
+    cropped_image.save(result_path)
     
     return result_path