Update app.py

app.py

@@ -19,19 +19,11 @@ def draw_boxes(img, bounds, color='yellow', width=2):
     draw = ImageDraw.Draw(img, "RGBA")
     for bound in bounds:
         p0, p1, p2, p3 = bound[0]
-        
-        #for masking the image
-        """
-        x_mid0, y_mid0 = midpoint(x1, y1, x2, y2)
-        x_mid1, y_mi1 = midpoint(x0, y0, x3, y3)
-        thickness = int(math.sqrt( (x2 - x1)**2 + (y2 - y1)**2))
-        
-        mask = np.zeros(img.shape[:2], dtype="uint8")
-        cv2.line(mask, (x_mid0, y_mid0), (x_mid1, y_mi1), 255, thickness)
-        masked = cv2.bitwise_and(img, img, mask=mask)
-        """
-            
-        draw.polygon((*p0, *p1, *p2, *p3), fill=(200,0,0,100))
+        draw.polygon((*p0, *p1, *p2, *p3), fill=(200,0,0,100))   
+    return img
+    
+def remove_text(img,imgwmask):
+    masked = cv2.bitwise_and(img, img, mask=imgwmask)
     return img
         
 def inference(img, lang):
@@ -46,12 +38,13 @@ def inference(img, lang):
     reader = easyocr.Reader(lang)
     bounds = reader.readtext(img.name)
     im = PIL.Image.open(img.name)
-    im2 = PIL.Image.open(img.name)
-    draw_boxes(im, bounds)
-    #remove_text(im, bounds)
+    
+    mask = draw_boxes(im, bounds)
+    #remove_text(im, mask, bounds)
     lang = ""
     im.save('result.png')
-    return ['result.png', pd.DataFrame(bounds). iloc[: , 1:2]]
+    mask.save('mask.png')
+    return ['result.png','mask.png', pd.DataFrame(bounds). iloc[: , 1:2]]
 
 title = 'EasyOCR'
 description = 'Gradio demo for EasyOCR. EasyOCR demo supports 80+ languages.To use it, simply upload your image and choose a language from the dropdown menu, or click one of the examples to load them. Read more at the links below.'
@@ -67,7 +60,7 @@ choices = [
 gr.Interface(
     inference,
     [gr.inputs.Image(type='file', label='Input'),gr.inputs.Dropdown(choices, type="value", default="korean", label='language')],
-    [gr.outputs.Image(type='file', label='Output'), gr.outputs.Dataframe()],
+    [gr.outputs.Image(type='file', label='Output'), gr.outputs.Image(type='file', label='Output'), gr.outputs.Dataframe()],
     title=title,
     description=description,
     article=article,