Update app.py

app.py

@@ -1,55 +1,72 @@
 import gradio as gr
+import logging
+import os
 from paddleocr import PaddleOCR
 from PIL import Image
 import numpy as np
-import logging
 
-# Tắt log debug của Paddle để gọn console
+# 1. Cấu hình tắt log bằng code hệ thống (Thay vì truyền tham số show_log)
+os.environ["CPP_MIN_LOG_LEVEL"] = "3"
 logging.getLogger("ppocr").setLevel(logging.WARNING)
 
-print("Đang tải model PaddleOCR...")
+print("Đang khởi tạo PaddleOCR (Chinese Model)...")
+
+# 2. KHỞI TẠO MODEL (Đã sửa lỗi)
+# - Bỏ show_log=False (Gây lỗi crash)
+# - Bỏ use_gpu=False (Gây lỗi crash ở bản mới, Paddle tự nhận diện CPU)
+# - Đổi use_angle_cls -> use_textline_orientation (Theo chuẩn mới)
 
-# CẤU HÌNH AN TOÀN NHẤT CHO CPU:
-# 1. Bỏ use_gpu=False (Paddle tự detect CPU)
-# 2. Thay use_angle_cls bằng tham số mới nếu cần, nhưng ở bản 2.7.3 thì use_angle_cls vẫn chạy tốt.
-#    Tuy nhiên, để tránh lỗi "Unknown argument", ta khởi tạo đơn giản nhất.
 try:
-    # Thử khởi tạo theo cách chuẩn
-    ocr = PaddleOCR(use_angle_cls=True, lang='ch', show_log=False)
-except Exception as e:
-    print(f"Lỗi khởi tạo lần 1: {e}")
-    # Fallback nếu phiên bản thay đổi
-    ocr = PaddleOCR(use_angle_cls=True, lang='ch', use_gpu=False, show_log=False)
+    # Cách khởi tạo chuẩn cho PaddleOCR v2.9+
+    ocr = PaddleOCR(
+        use_textline_orientation=True, 
+        lang='ch'
+    )
+except TypeError as e:
+    # Trường hợp dự phòng nếu thư viện thay đổi API bất ngờ
+    print(f"Cảnh báo khởi tạo: {e}. Chuyển sang chế độ tối giản.")
+    ocr = PaddleOCR(lang='ch')
+except ValueError as e:
+    # Xử lý lỗi tham số lạ
+    print(f"Lỗi tham số: {e}. Đang thử lại không tham số phụ...")
+    ocr = PaddleOCR(lang='ch')
 
-print("Đã tải model thành công!")
+print("Model đã sẵn sàng hoạt động!")
 
 def predict(image):
     if image is None:
-        return "Vui lòng tải ảnh lên."
+        return "请上传图片 / Vui lòng tải ảnh lên."
     
     try:
+        # Chuyển đổi ảnh PIL sang Numpy array
         if isinstance(image, Image.Image):
             image = np.array(image)
             
+        # Thực hiện nhận dạng
         result = ocr.ocr(image, cls=True)
         
         txts = []
         if result and result[0]:
             for line in result[0]:
                 text = line[1][0]
-                txts.append(text)
+                confidence = line[1][1]
+                # Lọc kết quả có độ tin cậy > 50%
+                if confidence > 0.5:
+                    txts.append(text)
             return "\n".join(txts)
         else:
-            return "Không tìm thấy văn bản nào."
+            return "未发现文字 / Không tìm thấy văn bản."
+            
     except Exception as e:
-        return f"Đã xảy ra lỗi khi nhận dạng: {str(e)}"
+        return f"Error / Lỗi hệ thống: {str(e)}"
 
+# Giao diện Gradio
 iface = gr.Interface(
     fn=predict,
-    inputs=gr.Image(type="pil", label="Tải ảnh lên"),
-    outputs=gr.Textbox(label="Kết quả nhận dạng", lines=10),
-    title="PaddleOCR Tiếng Việt - CPU Mode",
-    description="Chạy trên Docker (PaddleOCR v2.7.3 stable)",
+    inputs=gr.Image(type="pil", label="上传图片 / Upload Image"),
+    outputs=gr.Textbox(label="结果 / Result", lines=15, show_copy_button=True),
+    title="PaddleOCR Chinese - CPU Fixed",
+    description="Phiên bản đã sửa lỗi tham số 'show_log'. Chạy ổn định trên Hugging Face CPU.",
     examples=[]
 )