import os
from pdf2image import convert_from_path

# PDF文件路徑
pdf_path = '/Users/xxx/2022.pdf'
# 輸出圖片的文件夾
output_folder = './output_images2022'
# 輸出圖片的命名格式
output_name = 'page'

# 如果輸出文件夾不存在，創(chuàng)建它
if not os.path.exists(output_folder):
    os.makedirs(output_folder)

# 將PDF轉換為圖像列表，設置分辨率為300 DPI
images = convert_from_path(pdf_path, dpi=300)

# 保存每一頁為PNG圖片
for i, image in enumerate(images):
    image.save(f'{output_folder}/{output_name}_{i+1}.png', 'PNG')

from PIL import ImageEnhance
import pytesseract
from PIL import Image
from openpyxl import Workbook

# 配置 Tesseract 的路徑（如果需要）
# pytesseract.pytesseract.tesseract_cmd = r'/usr/local/bin/tesseract'  # Mac 的路徑
# pytesseract.pytesseract.tesseract_cmd = r'C:\Program Files\Tesseract-OCR\tesseract.exe'  # Windows 的路徑

# 打開圖片
# image_path = "/Users/xxx/page_3.png"  # 替換為你的圖片路徑


def enhance_image(img):
    img = img.convert('L')  # 轉灰度
    img = ImageEnhance.Contrast(img).enhance(2.0)
    return img


def allimngs(image_path):

    image = Image.open(image_path)

    image = enhance_image(image)

    # 使用 pytesseract 進行 OCR
    text = pytesseract.image_to_string(image, lang="chi_sim")  # 中文

    # # 打印提取的文本
    # print("提取的文本：")
    # print(text.replace(' ', ''))

    return text.replace(' ', '')


# 統(tǒng)計子字符串出現(xiàn)次數(shù)

class TrieNode:
    def __init__(self):
        self.children = {}
        self.keywords = []


class Trie:
    def __init__(self):
        self.root = TrieNode()

    def insert(self, keyword):
        node = self.root
        for char in keyword:
            if char not in node.children:
                node.children[char] = TrieNode()
            node = node.children[char]
        node.keywords.append(keyword)


def count_keywords(text, keywords):
    # 去重關鍵詞以確保唯一性
    keywords = list(set(keywords))

    # 構建Trie樹
    trie = Trie()
    for kw in keywords:
        trie.insert(kw)

    # 初始化計數(shù)器
    counters = {kw: 0 for kw in keywords}
    i = 0
    n = len(text)

    while i < n:
        current_node = trie.root
        max_len = 0
        current_len = 0
        end_pos = i

        # 尋找從當前位置開始的最長匹配關鍵詞
        for j in range(i, n):
            char = text[j]
            if char in current_node.children:
                current_node = current_node.children[char]
                current_len += 1
                if current_node.keywords:  # 當前節(jié)點是某個關鍵詞的結尾
                    max_len = current_len
                    end_pos = j + 1  # 更新結束位置為當前字符的下一個位置
            else:
                break  # 無后續(xù)匹配，退出循環(huán)

        if max_len > 0:
            # 更新所有匹配的關鍵詞計數(shù)器
            for kw in current_node.keywords:
                counters[kw] += 1
            i = end_pos  # 跳躍到已匹配部分的末尾
        else:
            i += 1  # 無匹配，移動到下一個字符

    return counters


if __name__ == "__main__":
    keywords = ['矮小',
                '安于現(xiàn)狀',
                '暗藏',
                '暗淡',
                '暗黑']
    all_text = ''
    workbook = Workbook()
    sheet = workbook.active

    for i in range(108):
        i = i+1
        image_path = f"/Users/xxx/output_images2022/page_{i}.png"
        all_text = all_text + allimngs(image_path)

    all_text = all_text.replace(' ', '').replace('\n', '')

    result = count_keywords(all_text, keywords)

    num = 1

    for k, v in result.items():
        sheet[f'A{num}'] = k
        sheet[f'B{num}'] = v
        print(k, v, num)
        num = num + 1

    workbook.save(filename='2022.xlsx')

pip install paddleocr

from paddleocr import PaddleOCR, draw_ocr
from PIL import Image
 
# 初始化 PaddleOCR
# 參數(shù)解釋：
# `lang`：指定語言模型，如 'en'（英文）、'ch'（中文）等。
# `use_angle_cls`：是否啟用文字方向分類器。
ocr = PaddleOCR(use_angle_cls=True, lang='en')  # 也可以設置為 'ch' 用于中文[^28^]
 
# 指定圖片路徑
img_path = 'example.jpg'  # 替換為你的圖片路徑
 
# 執(zhí)行 OCR 識別
result = ocr.ocr(img_path, cls=True)  # `cls=True` 表示啟用方向分類器
 
# 打印識別結果
for line in result:
    print(line)
 
# 可選：繪制識別結果并保存
if result:
    image = Image.open(img_path).convert('RGB')
    boxes = [line[0] for line in result]  # 提取文字框
    txts = [line[1][0] for line in result]  # 提取文字內(nèi)容
    scores = [line[1][1] for line in result]  # 提取置信度
 
    # 繪制結果
    im_show = draw_ocr(image, boxes, txts, scores, font_path='path/to/PaddleOCR/doc/fonts/simfang.ttf')
    im_show = Image.fromarray(im_show)
    im_show.save('result.jpg')  # 保存繪制后的圖片[^28^]

pip install rapidocr_onnxruntime

from rapidocr_onnxruntime import RapidOCR
 
# 初始化 OCR 引擎
ocr = RapidOCR()
 
# 指定圖片路徑
img_path = 'example.jpg'  # 替換為你的圖片路徑
 
# 執(zhí)行識別
result, _ = ocr(img_path)
 
# 提取并打印識別結果（僅數(shù)字和字母）
if result:
    for line in result:
        text = line[1]  # 提取文字內(nèi)容
        # 篩選只包含數(shù)字和字母的文本
        if text.isalnum():
            print(text)
else:
    print("未識別到文字")

pip install easyocr

import easyocr
 
reader = easyocr.Reader(['en', 'ch_sim'])  # 支持多語言
img_path = 'example.jpg'
result = reader.readtext(img_path)
for line in result:
    print(line[1])  # 打印識別結果

pip install pytesseract

from PIL import Image
import pytesseract
 
img_path = 'example.jpg'
text = pytesseract.image_to_string(Image.open(img_path), lang='eng')
print(text)  # 打印識別結果

pip install python-doctr

from doctr.models import ocr_predictor
from doctr.io import DocumentFile
 
img_path = 'example.jpg'
doc = DocumentFile.from_images(img_path)
model = ocr_predictor(pretrained=True)
result = model(doc)
for block in result.pages[0].blocks:
    for line in block.lines:
        for word in line.words:
            print(word.value)  # 打印識別結果

pip install pyocr

import pyocr
from PIL import Image
 
tools = pyocr.get_available_tools()
ocr_tool = tools[0]
img_path = 'example.jpg'
text = ocr_tool.image_to_string(Image.open(img_path), lang='eng')
print(text)  # 打印識別結果