Update app.py

app.py

@@ -21,21 +21,24 @@ def image_segmentation(image, compactness):
 
 # 圖像修復函式
 def inpaint_image(image, mask_threshold):
-    # 轉為灰階圖片
+    # 確保圖片格式為 RGB
+    if image.shape[-1] == 4:  # 若圖片有 alpha 通道
+        image = cv2.cvtColor(image, cv2.COLOR_RGBA2RGB)
+    elif image.shape[-1] == 3:
+        image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+    
     gray_image = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY)
-    # 根據遮罩閾值生成遮罩
     _, mask = cv2.threshold(gray_image, mask_threshold, 255, cv2.THRESH_BINARY)
     mask = mask.astype(bool)
     
-    # 檢查遮罩是否正確
     if not mask.any():
         raise ValueError("遮罩生成失敗，請調整遮罩閾值參數。")
     
-    # 使用 biharmonic 方法修復圖像
     inpainted = inpaint.inpaint_biharmonic(image, mask, multichannel=True)
     return inpainted
 
 
+
 # 主應用程式
 def app():
     with gr.Blocks() as demo: