Update app.py

app.py

@@ -1,37 +1,25 @@
 import gradio as gr
 import os
 import cv2
-import numpy as np
-from paddleocr import PPStructure,save_structure_res
-from paddleocr.ppstructure.recovery.recovery_to_doc import sorted_layout_boxes, convert_info_docx
-
+from paddleocr import PPStructure,draw_structure_result,save_structure_res
 
 # Chinese image
-table_engine = PPStructure(recovery=True)
-
-# English image
-# table_engine = PPStructure(recovery=True, lang='en')
+table_engine = PPStructure(show_log=True)
 
 examples = ['0.png']
 
 def find_layout(image):
-    # save_folder = './output'
-    # img = cv2.imread(image)
-    img = np.array(image)
+    img = cv2.imread(image)
     result = table_engine(img)
-    # save_structure_res(result, save_folder, os.path.basename("result").split('.')[0])
+    # save_structure_res(result, save_folder,os.path.basename(img_path).split('.')[0])
 
-    final_text = ""
+    output = []
     for line in result:
-        if "res" in line:        
-            print(line)
-            final_text += line["res"][0]["text"]
-
-    # h, w, _ = img.shape
-    # res = sorted_layout_boxes(result, w)
-    # convert_info_docx(img, res, save_folder, os.path.basename("result").split('.')[0])
-
-    return final_text
+        line.pop('img')
+        print(line)
+        output.append(line)
 
+    return output
+    
 iface = gr.Interface(fn=find_layout, inputs=[gr.Image(type="pil")], outputs="text", examples=examples)
 iface.launch()