Update app.py

app.py

@@ -43,8 +43,35 @@ def blurr_object(image, object, blur_strength):
     blur_image = np.asarray(image.filter(ImageFilter.GaussianBlur(radius=blur_strength)))
     mask_array_three_channel_invert = 1-mask_array_three_channel
     blur_image_reverse_mask = blur_image*mask_array_three_channel_invert
+    
+    #seg_out=image_array.astype(np.uint8)
+
+    
+    blurred_output_image = Image.fromarray((blur_image_reverse_mask).astype(np.uint8)+segmented_image.astype(np.uint8))
+    for _ in range(int(blur_strength//2.5)):
+        blurred_output_image = blurred_output_image.filter(ImageFilter.SMOOTH_MORE)
+    return blurred_output_image
+
+    
+def get_seg(image, object):
+    image = img_resize(image)
+
+    object_number = int(object.split('_')[0])
+    mask_array = np.asarray(pred[object_number]['mask'])/255
+    image_array = np.asarray(image)
+
+    mask_array_three_channel = np.zeros_like(image_array)
+    mask_array_three_channel[:,:,0] = mask_array
+    mask_array_three_channel[:,:,1] = mask_array
+    mask_array_three_channel[:,:,2] = mask_array
+
+    segmented_image = image_array*mask_array_three_channel
+
+    #blur_image = np.asarray(image.filter(ImageFilter.GaussianBlur(radius=blur_strength)))
+    mask_array_three_channel_invert = 1-mask_array_three_channel
+    #blur_image_reverse_mask = blur_image*mask_array_three_channel_invert
     seg_box=[]
-    for i in range(int(object)):
+    for i in range(len(object)):
         object_number = int(object.split('i')[0])
         mask_array = np.asarray(pred[object_number]['mask'])/255
         image_array = np.asarray(image)
@@ -56,14 +83,11 @@ def blurr_object(image, object, blur_strength):
     
         segmented_image = image_array*mask_array_three_channel
         seg_out=segmented_image.astype(np.uint8)
-        seg_box.append(seg_out)
-    #seg_out=image_array.astype(np.uint8)
+        seg_box.append(seg_out)  
+
+    return(seg_box)
+
 
-    
-    blurred_output_image = Image.fromarray((blur_image_reverse_mask).astype(np.uint8)+segmented_image.astype(np.uint8))
-    for _ in range(int(blur_strength//2.5)):
-        blurred_output_image = blurred_output_image.filter(ImageFilter.SMOOTH_MORE)
-    return blurred_output_image, seg_box
 
 app = gr.Blocks()
 
@@ -104,8 +128,12 @@ with app:
                 """)
             image_output = gr.Image()
         with gr.Column():
+            seg_btn.click(label="Run")
             gal1=gr.Gallery(type="filepath").style(grid=10)
 
+    seg_btn.click(get_seg, inputs=[image_input,object_output], outputs=gal1)
+
+    
     image_input.change(fn=image_objects, 
         inputs=image_input, 
         outputs=object_output
@@ -113,11 +141,11 @@ with app:
 
     object_output.change(fn=blurr_object, 
         inputs=[image_input, object_output, blur_slider],
-        outputs=[image_output, gal1])
+        outputs=[image_output])
 
     blur_slider.change(fn=blurr_object, 
         inputs=[image_input, object_output, blur_slider],
-        outputs=[image_output, gal1])
+        outputs=[image_output])
     
 
 app.launch()