Update app.py

app.py

@@ -31,16 +31,22 @@ def inpaint_image(image, mask_threshold):
         _, mask = cv2.threshold(gray_image, mask_threshold, 255, cv2.THRESH_BINARY)
         mask = mask.astype(bool)
 
-        # 如果遮罩無效，拋出錯誤
+        # 如果遮罩無效，返回遮罩檢查圖片
         if not mask.any():
-            return "遮罩生成失敗，請調整遮罩閾值參數。"
+            error_image = np.zeros_like(image)
+            cv2.putText(error_image, "Invalid Mask! Adjust Threshold", (10, 50),
+                        cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 2)
+            return error_image
 
         # 使用 biharmonic 方法修復圖像
         inpainted = inpaint.inpaint_biharmonic(image, mask, multichannel=True)
         return inpainted
     except Exception as e:
-        # 如果發生錯誤，返回錯誤訊息
-        return f"發生錯誤：{str(e)}"
+        # 如果發生錯誤，返回錯誤圖片
+        error_image = np.zeros((image.shape[0], image.shape[1], 3), dtype=np.uint8)
+        cv2.putText(error_image, f"Error: {str(e)}", (10, 50),
+                    cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 2)
+        return error_image
 
 # 主應用程式
 def app():