Update app.py

app.py

@@ -1,69 +1,77 @@
 import gradio as gr
-import cv2
+from paddleocr import PaddleOCR, draw_ocr
+from PIL import Image
 import numpy as np
-import easyocr
-from paddleocr import PaddleOCR
 
-# Khởi tạo EasyOCR cho English, Japanese
-readers = {
-    "English": easyocr.Reader(['en'], gpu=False),
-    "Japanese": easyocr.Reader(['ja', 'en'], gpu=False),
+# ===== Khởi tạo model OCR cho từng ngôn ngữ =====
+LANG_CONFIG = {
+    "ch": "Chinese (Simplified)",
+    "en": "English",
+    "japan": "Japanese",
+    "korean": "Korean",
+    "fr": "French",
+    "german": "German",
 }
 
-# Khởi tạo PaddleOCR cho Chinese (Simplified)
-paddle_reader = PaddleOCR(
-    use_doc_orientation_classify=False,
-    use_doc_unwarping=False,
-    use_textline_orientation=False
-)
+# Load toàn bộ model sẵn (startup lâu nhưng inference nhanh)
+ocr_models = {
+    lang: PaddleOCR(lang=lang, use_angle_cls=True, use_gpu=False)
+    for lang in LANG_CONFIG.keys()
+}
 
-def preprocess_image(image, max_width=1024):
+def inference(img, lang):
     """
-    Resize ảnh nếu quá lớn, convert sang RGB.
+    Nhận diện text bằng PaddleOCR và vẽ kết quả ra ảnh.
     """
-    if image.shape[2] == 4:
-        image = cv2.cvtColor(image, cv2.COLOR_RGBA2RGB)
-    h, w = image.shape[:2]
-    if w > max_width:
-        ratio = max_width / w
-        image = cv2.resize(image, (max_width, int(h * ratio)))
-    return image
+    # OCR
+    ocr = ocr_models[lang]
+    result = ocr.ocr(img, cls=True)[0]
 
-def ocr_image(image, lang_choice):
-    """
-    OCR với EasyOCR (English/Japanese) hoặc PaddleOCR (Chinese).
-    """
-    # image = preprocess_image(image)
+    # Load ảnh gốc bằng PIL
+    image = Image.open(img).convert("RGB")
+
+    # Lấy bbox, text, confidence
+    boxes = [line[0] for line in result]
+    txts = [line[1][0] for line in result]
+    scores = [line[1][1] for line in result]
 
-    if lang_choice == "Chinese (Simplified)":
-        # PaddleOCR: input có thể là numpy array
-        results = paddle_reader.ocr(image, cls=False)
-        texts = []
-        for res in results:
-            for line in res:
-                texts.append(line[1][0])  # line[1] = (text, confidence)
-        return "\n".join(texts) if texts else "Không nhận diện được văn bản"
-    else:
-        # EasyOCR
-        reader = readers[lang_choice]
-        results = reader.readtext(image)
-        texts = [res[1] for res in results]
-        return "\n".join(texts) if texts else "Không nhận diện được văn bản"
+    # Vẽ kết quả OCR
+    im_show = draw_ocr(image, boxes, txts, scores, font_path="./simfang.ttf")
+    im_show = Image.fromarray(im_show.astype(np.uint8))  # từ numpy -> PIL
+
+    # Trả về text + ảnh
+    text_out = "\n".join([f"{t} (conf={s:.2f})" for t, s in zip(txts, scores)])
+    return im_show, text_out
+
+
+# ===== Giao diện Gradio =====
+title = "PaddleOCR Multi-language"
+description = """
+PaddleOCR hỗ trợ Chinese, English, Japanese, Korean, French, German.
+- Upload ảnh + chọn ngôn ngữ để nhận diện.
+- Model load sẵn toàn bộ, inference nhanh.
+"""
+
+examples = [
+    ["en_example.jpg", "en"],
+    ["cn_example.jpg", "ch"],
+    ["jp_example.jpg", "japan"],
+]
 
-# Giao diện Gradio
 demo = gr.Interface(
-    fn=ocr_image,
+    fn=inference,
     inputs=[
-        gr.Image(type="numpy", label="Upload ảnh"),
-        gr.Dropdown(
-            choices=["English", "Chinese (Simplified)", "Japanese"],
-            value="English",
-            label="Chọn ngôn ngữ OCR"
-        )
+        gr.Image(type="filepath", label="Upload ảnh"),
+        gr.Dropdown(choices=list(LANG_CONFIG.keys()), value="ch", label="Ngôn ngữ OCR")
+    ],
+    outputs=[
+        gr.Image(type="pil", label="Ảnh kết quả"),
+        gr.Textbox(label="Kết quả nhận diện")
     ],
-    outputs="text",
-    title="OCR Đa Ngôn Ngữ (EasyOCR + PaddleOCR)",
-    description="English & Japanese dùng EasyOCR, Chinese Simplified dùng PaddleOCR"
+    title=title,
+    description=description,
+    examples=examples,
+    cache_examples=False,
 )
 
 if __name__ == "__main__":