Python調(diào)用ollama本地大模型進行批量識別PDF
更新時間：2025年03月10日 09:15:07 作者：月野難潯丶
現(xiàn)在市場上有很多PDF文件的識別,然而隨著AI的興起,本地大模型的部署,這些成為一種很方便的方法,本文我們就來看看Python如何調(diào)用ollama本地大模型進行PDF相關操作吧
現(xiàn)在市場上有很多PDF文件的識別，轉(zhuǎn)化，等等。有些業(yè)務可能需要總結摘要和關鍵詞等等一系列的操作。然而隨著AI的興起，本地大模型的部署，這些成為一種很方便的方法，接下來我將為各位介紹我所使用的方法。
本篇文章旨在自動化處理 PDF 文檔，提取并清理文本數(shù)據(jù)，然后使用一種大型模型生成摘要和關鍵詞。最后，處理結果會被整理并輸出到 Excel 文件中，便于后續(xù)分析和查看。
人工智能（AI）是一種模擬人類智能的科技，它已經(jīng)在現(xiàn)代科技中得到廣泛應用，并且是未來發(fā)展的重點領域之一。人工智能應用領域多樣，包括機器學習和數(shù)據(jù)分析、自然語言處理、機器視覺、自動化和機器人等。未來發(fā)展趨勢包括深度學習和神經(jīng)網(wǎng)絡、增強學習、多模態(tài)融合和泛用人工智能?？傮w而言，人工智能的應用將繼續(xù)擴大，并在不同領域帶來更多的創(chuàng)新和進步。（廢話~~~）
首先我們需要下載兩個庫PyPDF2以及ollama庫。（通過ollama部署好本地大模型：qwen2：14b或者其他大模型，這里部署步驟不再贅述，已經(jīng)有很成熟的步驟）方便調(diào)用~~終端輸入如下指令。
pip install PyPDF2 pip install ollama
PyPDF2是一個用于合并、分割、提取文本和元數(shù)據(jù)等PDF文件操作的Python庫。它建立在PDFMiner庫的基礎上，提供了更高級別的功能和易用性。ollama庫是一個用于機器學習和深度學習的Python庫。它提供了一系列功能強大的工具和函數(shù)，用于數(shù)據(jù)處理、模型構建、特征工程、模型選擇和評估等任務。兩者的結合則成為了如今的成果。話不多說，直接上代碼。
首先，在我們進行批量處理PDF文件時，先要了解如何處理單個PDF，然后再進行實現(xiàn)批量PDF的處理實現(xiàn)，如下是如何處理單個PDF，并設有異常處理，在處理PDF時存在部分亂碼，可能是包含有圖片格式的問題，故此設置了清洗文本，只保留了可以打印的字符，在提交給大模型進行回答時不受影響，個人沒有進行未清洗測試。
def clean_text(text):
text = re.sub(r'[^\x20-\x7E]+', '', text) # 只保留可打印的 ASCII 字符
return re.sub(r'\s+', ' ', text).strip()
def process_pdf(pdf_path, output_path):
try:
with open(pdf_path, "rb") as file:
reader = PyPDF2.PdfReader(file)
with open(output_path, "w", encoding='utf-8') as output_file:
for page in reader.pages:
text = page.extract_text()
if text: # 檢查是否成功提取文本
clean_text_result = clean_text(text) # 清理文本
output_file.write(clean_text_result + "\n") # 寫入文件
else:
output_file.write("未提取到有效文本\n")
except FileNotFoundError:
print(f"文件未找到: {pdf_path}")
return False
except PyPDF2.errors.PdfReadError:
print(f"無法讀取PDF文件: {pdf_path}")
return False
except Exception as e:
print(f"處理PDF文件時發(fā)生錯誤: {pdf_path}, 錯誤信息: {e}")
return False
return True接下來是定義超時處理異常類，在后面進行測試時發(fā)現(xiàn)，部分PDF通過這里無法執(zhí)行，就會一直卡著，增加超時處理，更方便后續(xù)進程的實現(xiàn)。
# 定義超時處理異常類
class TimeoutException(Exception):
pass
# 定義帶超時功能的線程類
class TimeoutThread(threading.Thread):
"""
允許超時處理的線程類。
"""
def __init__(self, target, args=(), kwargs={}):
threading.Thread.__init__(self)
self.target = target
self.args = args
self.kwargs = kwargs
self.result = None
self.exception = None
def run(self):
try:
self.result = self.target(*self.args, **self.kwargs)
except Exception as e:
self.exception = e
def join(self, timeout=None):
super(TimeoutThread, self).join(timeout)
if self.is_alive():
raise TimeoutException("處理超時")
if self.exception:
raise self.exception
return self.result這段是處理指定文件夾中的所有PDF文件，并讀取PDF識別后的txt文件中的文章信息，提交給本地大模型，我這里使用的qwen2.5：14b，總體上來說，qwen2.5還是好用的，并將結果保存到EXCEL中。至于替換信息是因為，qwen2.5給到的返回信息也是需要清理的。
def process_folder(folder_path, output_folder, excel_path):
"""
處理指定文件夾中的所有PDF文件，并將結果保存到Excel文件中。
"""
if not os.path.exists(output_folder):
os.makedirs(output_folder)
pdf_files = glob.glob(os.path.join(folder_path, "*.pdf"))
results = []
total_files = len(pdf_files)
processed_files = 0
errors = []
unprocessed_files = []
for pdf_file in pdf_files:
base_name = os.path.basename(pdf_file).replace(".pdf", ".txt")
output_path = os.path.join(output_folder, base_name)
success = process_pdf(pdf_file, output_path)
if not success:
errors.append(pdf_file)
continue
with open(output_path, "r", encoding='utf-8') as file:
content = file.read()
try:
# 使用線程實現(xiàn)超時處理
def process_model():
title = base_name.split(".txt")[0]
res = ollama.chat(model='qwen2.5:14b', stream=False, messages=[{"role": "user", "content": f"{content}總結成摘要和關鍵詞"}], options={"temperature": 0})
summary = res['message']['content'].split('### 摘要\n\n')[1].split('\n\n### 關鍵詞')[0]
keywords = res['message']['content'].split('### 關鍵詞\n\n')[1].split('\n- ')[1:]
keywords = '、'.join(keywords)
results.append({"文件名": title, "摘要": summary, "關鍵詞": keywords})
print(res)
timeout_thread = TimeoutThread(target=process_model)
timeout_thread.start()
timeout_thread.join(timeout=30)
except TimeoutException:
print(f"處理大模型時超時: {pdf_file}")
errors.append(pdf_file)
except Exception as e:
print(f"處理大模型時發(fā)生錯誤: {pdf_file}, 錯誤信息: {e}")
errors.append(pdf_file)
processed_files += 1
print(f"進度: {processed_files}/{total_files} 文件已處理")
# 每次處理完一個文件后保存Excel文件
write_to_excel(results, excel_path)
# 記錄未處理的文件
unprocessed_files = pdf_files[processed_files:]
return results, errors, unprocessed_files返回的信息如圖所示，所以我們需要進一步處理。
最后我們將總結出來的關鍵詞，文章摘要，以及對應的PDF標題寫入EXCEL中。
def write_to_excel(results, excel_path):
df = pd.DataFrame(results)
df.to_excel(excel_path, index=False)
最后加上我們的主函數(shù)，完整代碼如下：
import PyPDF2
import re
import ollama
import os
import glob
import pandas as pd
import threading
import time
# 定義函數(shù)來去除特殊空格和非法字符
def clean_text(text):
# 移除特定的非法字符
text = re.sub(r'[^\x20-\x7E]+', '', text) # 只保留可打印的 ASCII 字符
# 替換多個空格
return re.sub(r'\s+', ' ', text).strip()
# 定義函數(shù)來處理單個PDF文件
def process_pdf(pdf_path, output_path):
"""
處理單個PDF文件，提取文本并輸出到指定路徑。
"""
try:
with open(pdf_path, "rb") as file:
reader = PyPDF2.PdfReader(file)
with open(output_path, "w", encoding='utf-8') as output_file:
for page in reader.pages:
text = page.extract_text()
if text: # 檢查是否成功提取文本
clean_text_result = clean_text(text) # 清理文本
output_file.write(clean_text_result + "\n") # 寫入文件
else:
output_file.write("未提取到有效文本\n")
except FileNotFoundError:
print(f"文件未找到: {pdf_path}")
return False
except PyPDF2.errors.PdfReadError:
print(f"無法讀取PDF文件: {pdf_path}")
return False
except Exception as e:
print(f"處理PDF文件時發(fā)生錯誤: {pdf_path}, 錯誤信息: {e}")
return False
return True
# 定義超時處理異常類
class TimeoutException(Exception):
pass
# 定義帶超時功能的線程類
class TimeoutThread(threading.Thread):
"""
允許超時處理的線程類。
"""
def __init__(self, target, args=(), kwargs={}):
threading.Thread.__init__(self)
self.target = target
self.args = args
self.kwargs = kwargs
self.result = None
self.exception = None
def run(self):
try:
self.result = self.target(*self.args, **self.kwargs)
except Exception as e:
self.exception = e
def join(self, timeout=None):
super(TimeoutThread, self).join(timeout)
if self.is_alive():
raise TimeoutException("處理超時")
if self.exception:
raise self.exception
return self.result
# 定義函數(shù)來處理文件夾中的所有PDF文件
def process_folder(folder_path, output_folder, excel_path):
"""
處理指定文件夾中的所有PDF文件，并將結果保存到Excel文件中。
"""
if not os.path.exists(output_folder):
os.makedirs(output_folder)
pdf_files = glob.glob(os.path.join(folder_path, "*.pdf"))
results = []
total_files = len(pdf_files)
processed_files = 0
errors = []
unprocessed_files = []
for pdf_file in pdf_files:
base_name = os.path.basename(pdf_file).replace(".pdf", ".txt")
output_path = os.path.join(output_folder, base_name)
success = process_pdf(pdf_file, output_path)
if not success:
errors.append(pdf_file)
continue
with open(output_path, "r", encoding='utf-8') as file:
content = file.read()
try:
# 使用線程實現(xiàn)超時處理
def process_model():
title = base_name.split(".txt")[0]
res = ollama.chat(model='qwen2.5:14b', stream=False, messages=[{"role": "user", "content": f"{content}總結成摘要和關鍵詞"}], options={"temperature": 0})
summary = res['message']['content'].split('### 摘要\n\n')[1].split('\n\n### 關鍵詞')[0]
keywords = res['message']['content'].split('### 關鍵詞\n\n')[1].split('\n- ')[1:]
keywords = '、'.join(keywords)
results.append({"文件名": title, "摘要": summary, "關鍵詞": keywords})
print(res)
timeout_thread = TimeoutThread(target=process_model)
timeout_thread.start()
timeout_thread.join(timeout=30)
except TimeoutException:
print(f"處理大模型時超時: {pdf_file}")
errors.append(pdf_file)
except Exception as e:
print(f"處理大模型時發(fā)生錯誤: {pdf_file}, 錯誤信息: {e}")
errors.append(pdf_file)
processed_files += 1
print(f"進度: {processed_files}/{total_files} 文件已處理")
# 每次處理完一個文件后保存Excel文件
write_to_excel(results, excel_path)
# 記錄未處理的文件
unprocessed_files = pdf_files[processed_files:]
return results, errors, unprocessed_files
# 定義函數(shù)來將結果寫入Excel文件
def write_to_excel(results, excel_path):
"""
將處理結果寫入指定的Excel文件。
"""
df = pd.DataFrame(results)
df.to_excel(excel_path, index=False)
# 主程序
if __name__ == "__main__":
a = input("PDF文件夾路徑:")
b = input("TXT文件輸出路徑：")
c = input("EXCEl文件輸出路徑:")
folder_path = fr"{a}" # 文件夾路徑
output_folder = fr"" # TXT文件輸出路徑
excel_path = fr"{c}\results.xlsx" # Excel文件輸出路徑
results, errors, unprocessed_files = process_folder(folder_path, output_folder, excel_path)
print(f"所有PDF文件已處理完畢，結果已保存到 {excel_path}")
if errors:
print("以下PDF文件處理失敗:")
for error in errors:
print(error)
if unprocessed_files:
print("以下PDF文件未處理:")
for unprocessed in unprocessed_files:
print(unprocessed)附輸出結果以及EXCEL表。
實測56，成功45，失敗9，總體來說70-80的成功率，但也大大降低的工作量。
以上就是Python調(diào)用ollama本地大模型進行批量識別PDF的詳細內(nèi)容，更多關于Python ollama識別PDF的資料請關注腳本之家其它相關文章！
相關文章
Python中存在一些特殊的方法，有些方法以雙下劃線 “__” 開頭和結尾，它們是Python的魔法函數(shù)，比如__init__()和__str__等等，不用要這種方式命名自己的變量或者函數(shù)，這篇文章主要介紹了python里面單雙下劃線的區(qū)別,需要的朋友可以參考下2023-04-04
在本篇文章中小編給大家分享了關于python占位符怎么輸入的相關知識點內(nèi)容，有興趣的朋友們可以學習下。2019-05-05
這篇文章主要介紹了Pytest mark使用實例及原理解析,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-02-02
這篇文章主要介紹了使用Flask集成bootstrap的方法，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-07-07
這篇文章主要介紹了在Django中進行用戶注冊和郵箱驗證的方法的相關資料,需要的朋友可以參考下2016-05-05
Python編程實現(xiàn)數(shù)學運算求一元二次方程的實根算法示例
這篇文章主要介紹了Python編程實現(xiàn)數(shù)學運算求一元二次方程的實根算法,涉及Python數(shù)學運算求解方程的相關實現(xiàn)技巧,需要的朋友可以參考下2017-04-04
pycharm默認的【新建】文件，格式很不友好，那么就需要改一下文件模板。下面這篇文章主要給大家介紹了關于PyChar學習教程之自定義文件與代碼模板的相關資料，文中通過示例代碼介紹的非常詳細，需要的朋友們下面跟著小編來一起看看吧。2017-07-07
python內(nèi)置函數(shù)anext的具體使用
本文主要介紹了python內(nèi)置函數(shù)anext的具體使用，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2023-01-01
通過python爬蟲mechanize庫爬取本機ip地址的方法
python中的mechanize算是一個比較古老的庫了，在python2的時代中，使用的多一些，在python3以后就很少使用了，現(xiàn)在已經(jīng)是2202年了，可能很多人都沒聽說過mechanize，這不要緊，我們先來簡單的講解一下，如何使用mechanize，感興趣的朋友一起看看吧2022-08-08
Django ORM多對多查詢方法(自定義第三張表&ManyToManyField)
今天小編就為大家分享一篇Django ORM多對多查詢方法(自定義第三張表&ManyToManyField)，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-08-08