Update pictureDeal.py

pictureDeal.py

@@ -59,13 +59,15 @@ with gr.Blocks() as interface:
         closed = cv2.morphologyEx(thresh, cv2.MORPH_CLOSE, kernel)
         image = Image.fromarray(img_output)
         closed = closed.astype(img.dtype)
+        result = cv2.bitwise_and(img_output, img_output, mask=closed)
+        result[closed==0] = (255,255,255)
         # 颜色空间转换  
         enhancer = ImageEnhance.Color(image=image)
         # 增强颜色  
         img1 = enhancer.enhance(enhance).convert('RGB')
-        img2 = Image.fromarray(closed).convert('RGB')
+        img2 = Image.fromarray(result).convert('RGB')
         union_img = np.asarray(Image.blend(img2, img1, blend))
-        return closed,union_img
+        return result,union_img
     
     section_btn2.click(turn_arguments,inputs=[img_input, img_output,gaussian_blur,
                                               structuring_element,canny_start,canny_end,thresh_val,maxval,enhance,blend ],