Update app.py

app.py

@@ -2,14 +2,16 @@ from fastapi import FastAPI, UploadFile, File, HTTPException
 from fastapi.responses import HTMLResponse
 from fastapi.middleware.cors import CORSMiddleware
 from paddleocr import PaddleOCR
+import paddleocr  # <-- Thêm dòng này để gọi lõi thư viện
 import numpy as np
 from PIL import Image
 import io
+import os         # <-- Thêm dòng này để xử lý đường dẫn
 
 # 1. Khởi tạo ứng dụng FastAPI
 app = FastAPI(title="PaddleOCR Tiếng Việt API")
 
-# Cấu hình CORS (Cho phép giao diện web gọi API mà không bị chặn)
+# Cấu hình CORS
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -18,14 +20,22 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# 2. Tải mô hình PaddleOCR vào RAM (Chỉ chạy 1 lần khi khởi động Server)
-print("Đang tải mô hình PaddleOCR (Tiếng Việt - Bản CPU) vào RAM...")
-# use_angle_cls=True: Tự động xoay ảnh nếu chữ bị nghiêng
-# lang='vi': Nhận diện tiếng Việt
-# use_gpu=False: Chạy hoàn toàn trên CPU
-ocr = PaddleOCR(use_angle_cls=True, lang='vi', ocr_version='PP-OCRv4')
+# 2. Bắt buộc tìm và trỏ đích danh vào file từ điển Tiếng Việt
+vi_dict_path = os.path.join(os.path.dirname(paddleocr.__file__), 'ppocr', 'utils', 'dict', 'vi_dict.txt')
+
+print("Đang tải mô hình PaddleOCR (Tiếng Việt v3) vào RAM...")
+# Ép sử dụng PP-OCRv3 và file từ điển chuẩn
+ocr = PaddleOCR(
+    use_angle_cls=True, 
+    lang='vi', 
+    ocr_version='PP-OCRv3',          # Dùng v3 thay vì v4
+    rec_char_dict_path=vi_dict_path  # Trỏ thẳng vào từ điển tiếng Việt
+)
 print("Tải mô hình hoàn tất, sẵn sàng phục vụ!")
 
+# -----------------------------------------------------
+# -----------------------------------------------------
+
 
 # -----------------------------------------------------
 # ROUTE 1: Trả về giao diện người dùng (Frontend HTML)