Update pictureDeal2.py

pictureDeal2.py

@@ -7,13 +7,16 @@ with gr.Blocks() as interface:
 
     with gr.Accordion("请选择一张图片"):
         # gr.Markdown("Look at me...")
-        img_input = gr.Image(label='请选择一张待加工图片')
+        with gr.Row():
+            img_input = gr.Image(label='请选择一张待加工图片',interactive=True)
+            img_param_output1 = gr.Image(label='勾边合成结果',interactive=False,tool='canvas')
 
     with gr.Accordion("每次调整参数后，点击【加工图片】按钮，得到图片的勾边"):
         with gr.Row():
             enhance = gr.Slider(0, 1, 0.8, step=0.1, label="图片彩色度")
-            blend = gr.Slider(0, 1, 0.4, step=0.1, label="颜色填充度")
+            # blend = gr.Slider(0, 1, 0.4, step=0.1, label="颜色填充度")
             color = gr.ColorPicker(label="勾边颜色")
+
         
         section_btn = gr.Button("加工图片")
     
@@ -36,7 +39,8 @@ with gr.Blocks() as interface:
         
 
     # 调整模型结果参数
-    def turn_arguments(img,enhance,blend,color):
+    def turn_arguments(img,enhance,color):
+    # def turn_arguments(img,enhance,blend,color):
         imageX = Image.fromarray(img)
         contrast = ImageEnhance.Contrast(imageX)
         imageX = contrast.enhance(1.5)
@@ -53,6 +57,7 @@ with gr.Blocks() as interface:
         canny_end_0 = 100
         thresh_val_0 = 205
         maxval_0 = 330
+        blend = 0.4
         gray0 = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
         # 对灰度图像进行高斯滤波，以去除噪声  
         gray0 = cv2.GaussianBlur(gray0, (gaussian_blur_0,gaussian_blur_0), 0)
@@ -75,7 +80,7 @@ with gr.Blocks() as interface:
         enhancer0 = ImageEnhance.Color(image=image0)
         # 增强颜色  
         img0 = enhancer0.enhance(enhance).convert('RGB')
-        union_img0 = np.asarray(Image.blend(close00, img0, blend))
+        # union_img0 = np.asarray(Image.blend(close00, img0, blend))
 
 
         #####################################
@@ -142,7 +147,7 @@ with gr.Blocks() as interface:
         enhancer2 = ImageEnhance.Color(image=image2)
         # 增强颜色  
         img2 = enhancer2.enhance(enhance).convert('RGB')
-        union_img2 = np.asarray(Image.blend(close02, img2, blend))
+        # union_img2 = np.asarray(Image.blend(close02, img2, blend))
 
         #####################################
         #   简单勾边-彩色勾边                #
@@ -157,13 +162,13 @@ with gr.Blocks() as interface:
         enhancer3 = ImageEnhance.Color(image=image3)
         # 增强颜色  
         img3 = enhancer3.enhance(enhance).convert('RGB')
-        union_img3 = np.asarray(Image.blend(close03, img3, blend))
+        # union_img3 = np.asarray(Image.blend(close03, img3, blend))
 
         # return result0,union_img0,result1,union_img1,result2,union_img2,result3,union_img3
-        return result0,result1,result2,result3
+        return result0,result1,result2,result3,union_img1
     
-    section_btn.click(turn_arguments,inputs=[img_input,enhance,blend,color],
-                      outputs = [closed_output0,closed_output1,closed_output2,closed_output3])
+    section_btn.click(turn_arguments,inputs=[img_input,enhance,color],
+                      outputs = [closed_output0,closed_output1,closed_output2,closed_output3,img_param_output1])
                                     # outputs = [closed_output0,img_param_output0,closed_output1,img_param_output1,closed_output2,img_param_output2,closed_output3,img_param_output3])
 
 interface.launch(show_api=False)