百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 技术资源 > 正文

三步教你用Elasticsearch+PyMuPDF实现PDF大文件秒搜!

off999 2025-06-30 19:19 31 浏览 0 评论

面对100页以上的大型PDF文件时,阅读和搜索往往效率低下。传统关系型数据库在处理此类数据时容易遇到性能瓶颈,而 Elasticsearch 凭借其强大的全文检索和分布式架构,成为理想解决方案。通过 PyMuPDF 提取PDF文本内容,结合Elasticsearch的倒排索引技术,能够快速定位和检索信息,即使面对海量数据也能保持高效查询。其分布式特性支持多节点扩展,进一步提升性能和可靠性,完美解决传统数据库的性能瓶颈问题。

以下是详细的开发方案:

1. 系统设计

  • 功能需求

(1)支持上传 PDF 文件,并从中提取文本。
(2)支持分词操作以及全文检索。
(3)支持将匹配内容以高亮形式显示。
(4)支持分页展示搜索结果

  • 技术栈
    • PyMuPDF:用于解析PDF文件。
    • jieba:用于中文分词。
    • Elasticsearch:用于存储和检索文本数据。
    • Flask:用于构建Web界面和API。

2. 实现步骤

2.1 安装依赖

Python

pip install pymupdf flask jieba elasticsearch

2.2 启动Elasticsearch

(1)下载并安装Elasticsearch

(2)从 Elasticsearch官网(
https://www.elastic.co/cn/elasticsearch) 下载并解压

(3)运行 bin/elasticsearch 启动服务

(4)确保Elasticsearch运行在 http://localhost:9200

2.3 解析PDF并存入Elasticsearch

使用PyMuPDF提取PDF文本,并将文本数据存入Elasticsearch

Python

import fitz  # PyMuPDF
from elasticsearch import Elasticsearch
import jieba

# 连接Elasticsearch
es = Elasticsearch("http://localhost:9200")

def extract_and_store_pdf(filepath):
    # 提取文件名
    filename = filepath.split('/')[-1]

    # 解析PDF
    doc = fitz.open(filepath)
    for page_num in range(len(doc)):
        page = doc.load_page(page_num)
        text = page.get_text()

        # 对文本进行分词
        tokens = jieba.lcut(text)

        # 存入Elasticsearch
        doc_data = {
            "filename": filename,
            "page": page_num + 1,
            "text": text,
            "tokens": tokens
        }
        es.index(index="pdf_documents", body=doc_data)

# 示例:解析并存储PDF
extract_and_store_pdf('example.pdf')

2.4 实现分词搜索

使用Elasticsearch的全文检索功能实现分词搜索:

Python

def search_keyword(keyword):
    # 对关键词进行分词
    keyword_tokens = jieba.lcut(keyword)

    # 构建Elasticsearch查询
    query = {
        "query": {
            "match": {
                "tokens": " ".join(keyword_tokens)  # 将分词结果组合成查询字符串
            }
        },
        "highlight": {
            "fields": {
                "text": {}  # 高亮显示匹配内容
            }
        }
    }

    # 执行查询
    response = es.search(index="pdf_documents", body=query)
    return response['hits']['hits']

# 示例:搜索关键词
results = search_keyword('Python编程')
for hit in results:
    print(f"文件: {hit['_source']['filename']}, 页码: {hit['_source']['page']}")
    print(f"内容: {hit['highlight']['text'][0]}")

2.5 构建Web界面

使用Flask构建一个简单的Web界面,支持文件上传和搜索:

Python

from flask import Flask, render_template, request, redirect, url_for
import os

app = Flask(__name__)
UPLOAD_FOLDER = 'uploads'
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER

# 首页:上传文件
@app.route('/', methods=['GET', 'POST'])
def index():
    if request.method == 'POST':
        file = request.files['file']
        if file:
            filepath = os.path.join(app.config['UPLOAD_FOLDER'], file.filename)
            file.save(filepath)
            extract_and_store_pdf(filepath)
            return redirect(url_for('index'))
    return render_template('index.html')

# 搜索页面
@app.route('/search', methods=['GET'])
def search():
    keyword = request.args.get('keyword')
    if keyword:
        results = search_keyword(keyword)
        return render_template('search.html', results=results, keyword=keyword)
    return render_template('search.html')

if __name__ == '__main__':
    if not os.path.exists(UPLOAD_FOLDER):
        os.makedirs(UPLOAD_FOLDER)
    app.run(debug=True)

2.6 创建HTML模板

在 templates 目录下创建以下HTML文件:

index.html(文件上传页面):

html



search.html(搜索页面):

html



3. 运行系统

(1) 启动Elasticsearch:

bash

bin/elasticsearch

(2) 启动Flask应用:

bash

python app.py

(3) 访问 http://127.0.0.1:5000,上传PDF文件。

(4) 访问
http://127.0.0.1:5000/search,输入关键词搜索。

4. 功能扩展

  • 分页显示:在Elasticsearch查询中添加 from 和 size 参数,实现分页。
  • 停用词过滤:在分词时过滤常见停用词(如“的”、“是”)。
  • 词性标注:结合词性标注(如名词、动词)提升搜索精度。
  • 性能优化:使用Elasticsearch的分布式特性,支持大规模数据检索。
  • 语义化搜索:通过理解用户意图和上下文,返回相关结果,而非简单关键词匹配。

通过以上步骤,你可以构建一个高性能的PDF文件搜索系统!

创作不易,您的建议,是我们进步的动力!欢迎留言及加关注,帮助我们做得更好!

相关推荐

电脑已联网却无法上网(电脑已经联网了但是不能上网)

电脑连上网后,仍可能存在无法上网的情况,这可能是由多种原因造成的。以下是一些可能的原因和解决方法:1.浏览器问题:有时候,浏览器可能会出现故障,导致无法正常访问网络。您可以尝试清除浏览器的缓存和co...

u盘价格一览表(u盘单价)

不同品牌价格不同,不同内存价格也不同,例如8g、16g、32g、64g等多种容量大小的,根据容量的不同,报价在29元到120元之间不等。闪存盘虽然小,但相对来说却有很大的存储容量。U盘大多能够存储比一...

windows查看ip命令(windows如何查看ip地址)

查看电脑IP:    1)使用Windows+R键打开“运行”窗口,然后输入CMD进入命令提示窗口2)进入命令窗口之后,输入:ipconfig/all回车即可...

内存条的作用(内存条的作用和参数配置)

内存条是存储电脑运行所需的数据和程序,帮助CPU快速读取和运行,提高计算机的运行速度和处理能力。内存条也被称为随机存取存储器(RAM),是电脑中非常必要的一个组件。常见的内存条类型有DDR、DDR2、...

autocad2012安装失败(autocad2012无法安装)

如果您遇到CAD2012安装不了的问题,可能有几个原因导致这种情况。以下是一些常见的解决方法:1.确保系统要求:首先,请确保您的计算机符合CAD2012的系统要求。检查您的操作系统版本、内存、处理器...

win11 16g内存最佳虚拟内存(window10 16个g虚拟内存设置)

内存足够大可以将系统的虚拟内存关掉。1、鼠标右键【此电脑】,在菜单中选择【属性】。2、进入属性后,点击【高级系统设置】。3、进入系统属性后,点击高级下面的【设置】。4、进入性能设置后,点击【高级】。5...

查看windows7激活码(win7激活码哪里看)

windows7激活密钥如下:PPBK3-M92CH-MRR9X-34Y9P-7CH2FQ8JXJ-8HDJR-X4PXM-PW99R-KTJ3H8489X-THF3D-BDJQR-D27PH-P...

win10商业版和消费者版区别(win10商业版与消费者版)

1、用户群体的区别消费者版:通俗来说就是零售版,是一个非常适合个人用户和家庭用户购买的版本。商业版:适合大客户使用的版本,而且还比较适合企业用户使用以及进行批量部署。2、版本区别消费者版Consume...

bilibili加速器(bilibili加速器手机版官网)

需要在电脑上使用bilibili加速器,因为手机上bilibili已经有自带的加速器功能了。可以在bilibili官网或者一些应用商店下载使用,下完后按照安装提示进行安装即可。如果使用的是第三方软件,...

电脑自带的清理垃圾的工具(电脑自带的清理垃圾的工具叫什么)

CCleaner是一款免费的系统优化和隐私保护工具,它的体积小、扫描速度非常快,支持自定义清理规则,增强了应用程序清理范围和效果。CCleaner是Piriform(梨子公司)最著名广受好评的系统清理...

笔记本电脑用什么下载软件(笔记本电脑用什么下载软件比较好)
  • 笔记本电脑用什么下载软件(笔记本电脑用什么下载软件比较好)
  • 笔记本电脑用什么下载软件(笔记本电脑用什么下载软件比较好)
  • 笔记本电脑用什么下载软件(笔记本电脑用什么下载软件比较好)
  • 笔记本电脑用什么下载软件(笔记本电脑用什么下载软件比较好)
如何设置本地连接

在“控制面板”中,选择“网络和Internet”>“网络和共享中心”。在左侧窗格中,选择“更改适配器设置”。在“网络连接”窗口中,右键单击“本地连接”,然后选择“属性”。在“本地连接...

office2007官方免费版安装包
  • office2007官方免费版安装包
  • office2007官方免费版安装包
  • office2007官方免费版安装包
  • office2007官方免费版安装包
戴尔官网官方网站(戴尔产品官网)

查询步骤如下:1.在戴尔电脑的后盖上找到服务编号,并记录下来。2.之后搜索戴尔官网,在打开的官网界面中点击上方的支持选项,并点击产品支持。3.在打开的产品支持界面中,输入电脑后盖上的服务编号。4.如果...

黑鲨u盘重装系统教程(黑鲨u盘重装系统步骤8)

U盘重装WIn10系统:1、用【u深度u盘启动盘制作工具】制作u盘启动盘,插入电脑usb接口,设置好开机启动项进入u深度主菜单界面,选择“【02】u深度win8pe标准版(新机器)”并回车,2、在u深...

取消回复欢迎 发表评论: