Update app.py

app.py

@@ -16,11 +16,11 @@ def midpoint(x1, y1, x2, y2):
     y_mid = int((y1 + y2)/2)
     return (x_mid, y_mid)
 
-def draw_mask(img, bounds, color='yellow', width=2):
+def draw_mask(img, bounds):
     draw = ImageDraw.Draw(img)
     for bound in bounds:
         p0, p1, p2, p3 = bound[0]
-        draw.line([*p0, *p1, *p2, *p3, *p0], fill=color, width=width)
+        draw.polygon((*p0, *p1, *p2, *p3, *p0), fill=255)
     return img
        
 def inference(img, lang):
@@ -36,14 +36,14 @@ def inference(img, lang):
     bounds = reader.readtext(img.name)
     im = PIL.Image.open(img.name)
     #mask = PIL.Image.open(img.name)
-    #mask = Image.new("L", im.size, 0)
+    mask = Image.new("L", im.size, 0)
     
-    draw_mask(im, bounds)
+    draw_mask(mask, bounds)
     #remove_text(im, mask, bounds)
     lang = ""
     im.save('result.jpg')
-    #mask.save('mask.png')
-    return ['result.jpg', pd.DataFrame(bounds). iloc[: , 1:2]]
+    mask.save('mask.png')
+    return ['result.jpg', 'mask.jpg', 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.'
@@ -59,7 +59,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,