fix(app): Remove the single visualize function

app.py

@@ -9,10 +9,10 @@ import numpy as np
 import pandas as pd
 import streamlit as st
 from PIL import Image
-from rapidocr_onnxruntime import RapidOCR
+from rapidocr_onnxruntime import RapidOCR, VisRes
 from streamlit_image_select import image_select
 
-from utils import visualize
+vis = VisRes()
 
 font_dict = {
     "ch": "chinese_cht.ttf",
@@ -115,9 +115,8 @@ def inference(
     dt_boxes, rec_res, scores = list(zip(*ocr_result))
     font_path = Path("fonts") / font_dict.get(lan_name)
 
-    vis_img = visualize(
-        Image.fromarray(img), dt_boxes, rec_res, scores, font_path=str(font_path)
-    )
+    vis_img = vis(img, dt_boxes, rec_res, scores, font_path=str(font_path))
+    vis_img = vis_img[..., ::-1]
     out_df = pd.DataFrame(
         [[rec, score] for rec, score in zip(rec_res, scores)],
         columns=("Rec", "Score"),
@@ -137,8 +136,8 @@ if __name__ == "__main__":
     )
     st.markdown(
         """
-    <p align="left">
-        <a href=""><img src="https://img.shields.io/badge/Python->=3.6,<3.12-aff.svg"></a>
+    <p align="center">
+        <a href=""><img src="https://img.shields.io/badge/Python->=3.6,<3.13-aff.svg"></a>
         <a href=""><img src="https://img.shields.io/badge/OS-Linux%2C%20Win%2C%20Mac-pink.svg"></a>
         <a href="https://pepy.tech/project/rapidocr_onnxruntime"><img src="https://static.pepy.tech/personalized-badge/rapidocr_onnxruntime?period=total&units=abbreviation&left_color=grey&right_color=blue&left_text=Downloads%20Ort"></a>
         <a href="https://pypi.org/project/rapidocr-onnxruntime/"><img alt="PyPI" src="https://img.shields.io/pypi/v/rapidocr-onnxruntime"></a>

utils.py

@@ -1,78 +0,0 @@
-# -*- encoding: utf-8 -*-
-# @Author: SWHL
-# @Contact: [email protected]
-import math
-import random
-from pathlib import Path
-
-import numpy as np
-from PIL import Image, ImageDraw, ImageFont
-
-
-def draw_ocr_box_txt(image, boxes, txts, font_path, scores=None, text_score=0.5):
-    h, w = image.height, image.width
-    img_left = image.copy()
-    img_right = Image.new("RGB", (w, h), (255, 255, 255))
-
-    random.seed(0)
-    draw_left = ImageDraw.Draw(img_left)
-    draw_right = ImageDraw.Draw(img_right)
-    for idx, (box, txt) in enumerate(zip(boxes, txts)):
-        if scores is not None and float(scores[idx]) < text_score:
-            continue
-
-        color = (random.randint(0, 255), random.randint(0, 255), random.randint(0, 255))
-
-        box = [tuple(v) for v in box]
-        draw_left.polygon(box, fill=color)
-        draw_right.text([box[3][0], box[3][1]], str(idx), fill=color)
-
-        draw_right.polygon(
-            [
-                box[0][0],
-                box[0][1],
-                box[1][0],
-                box[1][1],
-                box[2][0],
-                box[2][1],
-                box[3][0],
-                box[3][1],
-            ],
-            outline=color,
-        )
-
-        box_height = math.sqrt(
-            (box[0][0] - box[3][0]) ** 2 + (box[0][1] - box[3][1]) ** 2
-        )
-
-        box_width = math.sqrt(
-            (box[0][0] - box[1][0]) ** 2 + (box[0][1] - box[1][1]) ** 2
-        )
-
-        if box_height > 2 * box_width:
-            font_size = max(int(box_width * 0.9), 10)
-            font = ImageFont.truetype(font_path, font_size, encoding="utf-8")
-            cur_y = box[0][1]
-            for c in txt:
-                char_size = font.getsize(c)
-                draw_right.text((box[0][0] + 3, cur_y), c, fill=(0, 0, 0), font=font)
-                cur_y += char_size[1]
-        else:
-            font_size = max(int(box_height * 0.8), 10)
-            font = ImageFont.truetype(font_path, font_size, encoding="utf-8")
-            draw_right.text([box[0][0], box[0][1]], txt, fill=(0, 0, 0), font=font)
-
-    img_left = Image.blend(image, img_left, 0.5)
-    img_show = Image.new("RGB", (w * 2, h), (255, 255, 255))
-    img_show.paste(img_left, (0, 0, w, h))
-    img_show.paste(img_right, (w, 0, w * 2, h))
-    return np.array(img_show)
-
-
-def visualize(image, boxes, txts, scores, font_path="./fonts/FZYTK.TTF"):
-    draw_img = draw_ocr_box_txt(image, boxes, txts, font_path, scores, text_score=0.5)
-
-    draw_img_save = Path("./inference_results/")
-    if not draw_img_save.exists():
-        draw_img_save.mkdir(parents=True, exist_ok=True)
-    return draw_img[:, :, ::-1]