Update app.py

app.py

@@ -131,7 +131,7 @@ async def nano_banana_completion_api(base_image: Image.Image, prompt: str) -> Im
 
 # --- メイン処理ロジック (Main Processing Logic) ---
 
-async def segment_and_inpaint(original_image_np: np.ndarray, sketchpad_image_np: np.ndarray, inpaint_prompt: str):
+async def segment_and_inpaint(original_image_np: np.ndarray, sketchpad_filepath: str, inpaint_prompt: str):
     """
     アップロードされた一枚絵とブラシで描かれたマスクを受け取り、
     マスクされた領域を分割し、欠損部分をAI補完する関数。
@@ -139,13 +139,20 @@ async def segment_and_inpaint(original_image_np: np.ndarray, sketchpad_image_np:
     if original_image_np is None:
         return None, None, None, "エラー: 元の画像がアップロードされていません。", None
     
-    # --- 修正箇所: sketchpad_image_npの型チェックを強化 ---
-    if not isinstance(sketchpad_image_np, np.ndarray):
-        # Sketchpadが空の場合、Noneではなく空のディクショナリなどが返されることがあるため、NumPy配列でなければエラーとする
-        return None, None, None, "エラー: 分割したいパーツをブラシで描画してください。描画キャンバスが空または不正な形式です。", None
+    # --- 修正: Sketchpadの入力がファイルパスの場合の処理 ---
+    if not sketchpad_filepath:
+        # Sketchpadが空（何も描かれていない）場合はファイルパスがNoneまたは空文字列
+        return None, None, None, "エラー: 分割したいパーツをブラシで描画してください。描画キャンバスが空です。", None
+
+    try:
+        # ファイルパスからPIL Imageとして読み込む
+        sketchpad_image = Image.open(sketchpad_filepath).convert("RGBA")
+        sketchpad_image_np = np.array(sketchpad_image)
+    except Exception as e:
+        print(f"Sketchpad画像読み込みエラー: {e}")
+        return None, None, None, "エラー: 描画キャンバスの画像を読み込めませんでした。ファイルパスが不正です。", None
 
     # PIL Image (元の画像) を取得
-    # NOTE: Gradioのデフォルト出力はNumPy配列
     original_image = Image.fromarray(original_image_np).convert("RGBA")
     
     # Sketchpadからの入力は、元の画像の上にブラシで描画されたRGBA画像 (描画部分は不透明)
@@ -159,13 +166,13 @@ async def segment_and_inpaint(original_image_np: np.ndarray, sketchpad_image_np:
         # 配列の次元が4未満（RGBAではない）場合に発生
         return None, None, None, "エラー: 描画データ形式が不正です。ブラシで描画が行われていないか、Gradioの出力形式に問題があります。", None
     except Exception as e:
-        # その他のスライスエラーなど（例えば、sketchpad_image_npがNumPy配列だが次元が異なる場合）
+        # その他のスライスエラーなど
         print(f"Sketchpad処理中の予期せぬエラー: {e}")
         return None, None, None, "エラー: 描画データの処理中に予期せぬエラーが発生しました。ブラシで適切に描画されているか確認してください。", None
 
-    # ユーザーが何も描画しなかった場合のチェック
+    # ユーザーが何も描画しなかった場合のチェック（ファイルパスチェックで大方カバーされるが、念のため）
     if np.all(sketch_mask_alpha == 0):
-        return None, None, None, "エラー: 分割したいパーツをブラシで描画してください。", None
+        return None, None, None, "エラー: 分割したいパーツをブラシで描画してください。（アルファチャンネルが全てゼロです）", None
 
     # NumPy配列からPIL Imageのマスクに変換 (Lモード)
     hair_mask = Image.fromarray(sketch_mask_alpha, mode='L')
@@ -272,8 +279,8 @@ with gr.Blocks(theme=theme, title="Live2D素材自動分割・補完アプリ")
             input_sketchpad = gr.Sketchpad(
                 label="② 分割したいパーツをブラシで塗る (例: 前髪)", 
                 height=400,
-                # Sketchpadにアップロードされた元画像を表示させるためのプレースホルダー
                 brush={"color": "#FF0000", "size": 20}, # ブラシ設定
+                type="filepath" # ★★★ 修正: 出力タイプをファイルパスに変更し、より頑健にする ★★★
             )
 
             # ★★★ 削除: エラーの原因となっているSketchpadの背景更新ロジックを削除 ★★★