Update app.py

app.py

@@ -535,15 +535,28 @@ def process_image(image_mask_dict):
     
     # 新しいGradioバージョンに対応（background, layers, compositeのキー）
     background = image_mask_dict.get("background")
-    layers = image_mask_dict.get("layers")
     
-    if background is None or layers is None or not layers:
+    if background is None:
         return None
     
+    # 背景画像がPIL形式でない場合はPILに変換
+    if not isinstance(background, Image.Image):
+        if isinstance(background, np.ndarray):
+            background = Image.fromarray(background.astype(np.uint8))
+        else:
+            return None
+    
     # 背景画像を取得
     img_array = np.array(background)
     
-    # レイヤー（マスク）を取得 - 最初のレイヤーを使用
+    # レイヤー（マスク）を取得
+    layers = image_mask_dict.get("layers")
+    
+    # マスクがない場合、元の画像をそのままPIL形式で返す
+    if layers is None or not layers:
+        return background
+    
+    # マスク処理を行う（マスクが存在する場合）
     mask_array = np.array(layers[0])
     
     # マスクを2値化（閾値は0より大きい値を全て255に）