Update app.py

app.py

@@ -18,25 +18,29 @@ 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
+    try:
+        # 確保圖片格式為 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():
+            return "遮罩生成失敗，請調整遮罩閾值參數。"
+
+        # 使用 biharmonic 方法修復圖像
+        inpainted = inpaint.inpaint_biharmonic(image, mask, multichannel=True)
+        return inpainted
+    except Exception as e:
+        # 如果發生錯誤，返回錯誤訊息
+        return f"發生錯誤：{str(e)}"
 
 # 主應用程式
 def app():
@@ -74,19 +78,18 @@ def app():
         with gr.Tab("圖像修復"):
             with gr.Row():
                 input_image = gr.Image(label="輸入圖片")
-                inpaint_result = gr.Image(label="修復結果")
+                inpaint_result = gr.Image(label="修復結果或錯誤訊息")
             with gr.Row():
                 mask_threshold = gr.Slider(0, 255, value=128, step=1, label="遮罩閾值")
             inpaint_button = gr.Button("執行圖像修復")
             inpaint_button.click(
                 fn=inpaint_image, 
                 inputs=[input_image, mask_threshold], 
-                outputs=inpaint_result,
-                show_error=True  # 在 UI 上顯示錯誤訊息
+                outputs=inpaint_result  # 錯誤訊息或結果會顯示在此
             )
     
     return demo
 
 # 啟動應用程式
 if __name__ == "__main__":
-    app().launch()
+    app().launch()