From 914e6664369e0ce592865fa61e92b8203ad0a147 Mon Sep 17 00:00:00 2001 From: Wenjie Zhang Date: Tue, 11 Mar 2025 18:10:31 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8DOCR=E8=AF=86=E5=88=AB?= =?UTF-8?q?=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/utils/__init__.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/src/utils/__init__.py b/src/utils/__init__.py index 39882d03..95083746 100644 --- a/src/utils/__init__.py +++ b/src/utils/__init__.py @@ -6,12 +6,21 @@ from src.utils.logging_config import logger def is_text_pdf(pdf_path): import fitz doc = fitz.open(pdf_path) - for page_num in range(len(doc)): + total_pages = len(doc) + if total_pages == 0: + return False + + text_pages = 0 + for page_num in range(total_pages): page = doc.load_page(page_num) text = page.get_text() if text.strip(): # 检查是否有文本内容 - return True - return False + text_pages += 1 + + # 计算有文本内容的页面比例 + text_ratio = text_pages / total_pages + # 如果超过50%的页面有文本内容,则认为是文本PDF + return text_ratio > 0.5 def hashstr(input_string, length=8, with_salt=False): import hashlib