# ==========================================
# OCR DETAIL COLUMN TEST
# ==========================================

from paddleocr import PaddleOCR
import cv2

IMAGE_FILE = r"G:\00. PROJECTS\API\Inventory\output\NamaBarang.jpg"

# ==========================================
# RESIZE 3X
# ==========================================

img = cv2.imread(IMAGE_FILE)

img = cv2.resize(
    img,
    None,
    fx=3,
    fy=3,
    interpolation=cv2.INTER_CUBIC
)

temp_file = "temp_namabarang.jpg"

cv2.imwrite(
    temp_file,
    img
)

# ==========================================
# INIT OCR
# ==========================================

print("Loading PaddleOCR...")

ocr = PaddleOCR(
    use_angle_cls=True,
    lang="en"
)

print("Ready")
print("")

# ==========================================
# OCR
# ==========================================

result = ocr.ocr(
    temp_file,
    cls=True
)

print("")
print("===== OCR RESULT =====")
print("")

if not result or not result[0]:

    print("Tidak ada text ditemukan")

else:

    for i, line in enumerate(result[0], start=1):

        box = line[0]
        text = line[1][0]
        conf = line[1][1]

        y_top = int(box[0][1])

        print(
            f"{i:02d}. "
            f"Y={y_top:4d}  "
            f"CONF={conf:.2f}  "
            f"{text}"
        )

print("")
print("======================")