Update app.py

app.py

@@ -120,10 +120,12 @@ async def nano_banana_completion_api(base_image: Image.Image, prompt: str) -> Im
                 await asyncio.sleep(delay)
                 delay *= 2  # 指数バックオフ
             else:
+                # ユーザーへの表示用にgr.Errorを再raise
                 raise gr.Error(f"API呼び出しエラー: Client error '{e.response.status_code} {e.response.reason_phrase}' for url '{e.request.url}'")
         except httpx.RequestError as e:
              raise gr.Error(f"APIリクエストエラー: ネットワークまたはタイムアウトエラーが発生しました。詳細: {e}")
         except Exception as e:
+            # その他の予期せぬエラー
             raise gr.Error(f"予期せぬエラーが発生しました: {e}")
     
     raise gr.Error("APIリクエストが最大リトライ回数を超えて失敗しました。")
@@ -135,30 +137,43 @@ async def segment_and_inpaint(original_image_np: np.ndarray, sketchpad_image_np:
     """
     アップロードされた一枚絵とブラシで描かれたマスクを受け取り、
     マスクされた領域を分割し、欠損部分をAI補完する関数。
-    (出力を6つにし、エラーメッセージを分離してorjsonエラーを回避)
     """
     # 失敗時の統一リターン（6つの出力に対応）
     def fail(message):
-        return None, None, None, None, None, f"❌ 処理失敗: {message}"
+        # 画像出力は全てNone, JSONとZIPは空文字列/None, ステータスにエラーメッセージ
+        return None, None, None, "{}", None, f"❌ 処理失敗: {message}"
 
     if original_image_np is None:
         return fail("元の画像がアップロードされていません。")
     
-    # Sketchpad入力の型チェックを強化（TypeError対策）
-    if not isinstance(sketchpad_image_np, np.ndarray) or sketchpad_image_np.ndim != 4:
-        return fail("分割したいパーツをブラシで描画してください。描画キャンバスが空または不正な形式です。")
+    # ★★★ 修正: Sketchpad入力の型チェックと次元調整の強化 ★★★
+    
+    if not isinstance(sketchpad_image_np, np.ndarray):
+        # NumPy配列ではない（NoneやDictなど）場合は、描画がないと判断
+        return fail("分割したいパーツをブラシで描画してください。（描画データが検出されません）")
+
+    if sketchpad_image_np.ndim == 3 and sketchpad_image_np.shape[2] == 3:
+        # 3次元配列（RGB）の場合、透明なアルファチャンネルを追加して4次元（RGBA）に変換
+        print("--- SketchpadデータがRGB (3次元) で渡されました。RGBA (4次元) に変換します。 ---")
+        h, w, _ = sketchpad_image_np.shape
+        # 全て透明(0)のアルファチャンネルを作成
+        alpha_channel = np.zeros((h, w, 1), dtype=np.uint8) 
+        # RGB配列とアルファチャンネルを結合
+        sketchpad_image_np = np.concatenate((sketchpad_image_np, alpha_channel), axis=2)
+        
+    elif sketchpad_image_np.ndim != 4 or sketchpad_image_np.shape[2] != 4:
+        # 4次元配列ではない、または4チャンネルではない場合は不正と判断
+        return fail(f"描画データ形式が不正です。（次元: {sketchpad_image_np.ndim}, チャンネル: {sketchpad_image_np.shape[2] if sketchpad_image_np.ndim >= 3 else 'N/A'}）。")
 
     # PIL Image (元の画像) を取得
     original_image = Image.fromarray(original_image_np).convert("RGBA")
     
     # Sketchpadからの入力は、元の画像の上にブラシで描画されたRGBA画像
-    try:
-        # Sketchpadの画像からアルファチャンネルを抽出（これがユーザーの描いたマスク）
-        sketch_mask_alpha = sketchpad_image_np[:, :, 3] 
-    except IndexError:
-        return fail("描画データ形式が不正です。RGBA (4チャンネル) の画像として処理できませんでした。")
+    # アルファチャンネルからマスクを抽出
+    sketch_mask_alpha = sketchpad_image_np[:, :, 3] 
     
-    # ユーザーが何も描画しなかった場合のチェック
+    # ユーザーが何も描画しなかった場合のチェック (アルファチャンネルが全てゼロ)
+    # ブラシで描画した場合、アルファチャンネルは255の領域を持つはず
     if np.all(sketch_mask_alpha == 0):
         return fail("分割したいパーツをブラシで描画してください。（描画が検出されません）")
 
@@ -182,7 +197,6 @@ async def segment_and_inpaint(original_image_np: np.ndarray, sketchpad_image_np:
     try:
         completed_body_part = await nano_banana_completion_api(body_with_hole, inpaint_prompt or "マスクされた領域のキャラクターの顔と身体の肌、及び下に着ている服を、元のイラストのテイストに合わせて自然に補完してください。")
     except gr.Error as e:
-        # APIエラーはここで捕まえる
         return fail(f"AI補完APIエラー: {e}")
     except Exception as e:
         return fail(f"AI補完処理中に予期せぬエラー: {e}")
@@ -265,12 +279,12 @@ with gr.Blocks(theme=theme, title="Live2D素材自動分割・補完アプリ")
                 type="numpy",
                 height=200,
             )
-            # 描画専用のSketchpadコンポーネント (type="numpy"に戻し、Python側で堅牢にチェック)
+            # 描画専用のSketchpadコンポーネント (type="numpy")
             input_sketchpad = gr.Sketchpad(
                 label="② 分割したいパーツをブラシで塗る (例: 前髪)", 
                 height=400,
                 brush={"color": "#FF0000", "size": 20}, 
-                type="numpy" # ★★★ 修正: numpyに戻す ★★★
+                type="numpy"
             )
             
             inpaint_prompt = gr.Textbox(
@@ -283,8 +297,7 @@ with gr.Blocks(theme=theme, title="Live2D素材自動分割・補完アプリ")
             gr.Markdown(
                 """
                 ### 📌 開発のポイント
-                * **安定性の向上:** 発生していた型の不一致エラー（`slice`や`read`）を回避するため、Gradioの出力をNumPy配列に統一し、入力チェックを強化しました。
-                * **エラー表示の改善:** 処理ステータスを専用のテキストボックスに出力するように変更し、JSONデコードエラーを防いでいます。
+                * **安定性の向上:** Gradioの入力データが予期せず3次元配列で渡された場合にも、アルファチャンネルを追加して処理を継続できるようにロジックを強化しました。
                 """
             )
 
@@ -318,9 +331,9 @@ with gr.Blocks(theme=theme, title="Live2D素材自動分割・補完アプリ")
             completed_body_output, 
             hair_part_output, 
             body_with_hole_output, 
-            output_json_text,   # JSON出力をテキストボックスに変更
+            output_json_text,   
             download_zip,
-            status_message      # ステータスメッセージを追加
+            status_message      
         ]
     )