Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import pandas as pd
 import easyocr
 import torch
 import PIL
@@ -6,7 +7,7 @@ from PIL import Image
 from PIL import ImageDraw
 
 
-def draw_boxes(image, bounds, color='blue', width=2):
+def draw_boxes(image, bounds, color='blue', width=1):
     draw = ImageDraw.Draw(image)
     for bound in bounds:
         p0, p1, p2, p3 = bound[0]
@@ -18,12 +19,14 @@ def detect(img, lang=['en']):
     bounds = reader.readtext(img)
     im = PIL.Image.open(img)
     im_out=draw_boxes(im, bounds)
-    return im_out
+    return im_out,pd.DataFrame(bounds).iloc[: , 1:],pd.DataFrame(bounds).iloc[: , 1:]
 
 with gr.Blocks() as robot:
     im=gr.Image(type="filepath")
     go_btn=gr.Button()
     out_im=gr.Image()
-    out_txt=gr.Textbox(lines=8)
-    go_btn.click(detect,im,out_im)
+    with gr.Row():
+        out_txt=gr.Textbox(lines=8)
+        data_f=gr.outputs.Dataframe(headers=['text', 'confidence'])]
+    go_btn.click(detect,im,[out_im,out_txt,data_f])
 robot.queue(concurrency_count=10).launch()