Update UI

app.py

@@ -1,4 +1,5 @@
 # -*- encoding: utf-8 -*-
+import json
 import time
 from pathlib import Path
 
@@ -12,7 +13,7 @@ font_dict = {
     'ch': 'FZYTK.TTF',
     'japan': 'japan.ttc',
     'korean': 'korean.ttf',
-    'en': 'FZYTK.TTF'
+    'en': 'FZYTK.TTF',
 }
 
 
@@ -64,13 +65,15 @@ def inference(img_path, box_thresh=0.5, unclip_ratio=1.6, text_score=0.5,
     font_path = Path('fonts') / font_dict.get(lan_name)
     img_save_path = visualize(img_path, dt_boxes, rec_res, scores,
                               font_path=str(font_path))
-    output_text = [f'{one_rec} {float(score):.4f}'
-                   for one_rec, score in zip(rec_res, scores)]
-    return img_save_path, output_text, out_log
+    # output_text = [f'{one_rec} {float(score):.4f}'
+    #                for one_rec, score in zip(rec_res, scores)]
+    out_dict = {i: {'rec_txt': rec, 'score': score}
+                for i, (rec, score) in enumerate(zip(rec_res, scores))}
+    out_json = json.dumps(out_dict, ensure_ascii=False)
+    return img_save_path, out_json, out_log
 
 
 if __name__ == '__main__':
-
     examples = [['images/1.jpg'],
                 ['images/ch_en_num.jpg'],
                 ['images/air_ticket.jpg'],
@@ -124,15 +127,15 @@ if __name__ == '__main__':
         with gr.Row():
             input_img = gr.Image(type='filepath', label='Input')
             out_img = gr.Image(type='filepath', label='Output')
+        out_json = gr.JSON(label='Rec Res')
         out_log = gr.outputs.Textbox(type='text', label='Run Log')
-        out_txt = gr.outputs.Textbox(type='text', label='RecText')
         button = gr.Button('Submit')
         button.click(fn=inference,
                      inputs=[input_img, box_thresh, unclip_ratio, text_score,
                              text_det, text_rec],
-                     outputs=[out_img, out_txt, out_log])
+                     outputs=[out_img, out_json, out_log])
         gr.Examples(examples=examples,
                     inputs=[input_img, box_thresh, unclip_ratio, text_score,
                             text_det, text_rec],
-                    outputs=[out_img, out_txt, out_log], fn=inference)
+                    outputs=[out_img, out_json, out_log], fn=inference)
     demo.launch(debug=True, enable_queue=True)