Chainlit结合百度飞浆的ocr识别和nlp自然语言处理做图片文字信息提取

PP飞桨简介

PaddlePaddle(PArallel Distributed Deep LEarning),是由百度公司开发的一款开源深度学习平台,支持动态和静态图模式,提供了从模型构建到训练、预测等一系列的功能。PaddlePaddle 的设计目标是让开发者能够更容易地实现、训练和部署自己的深度学习模型。它支持多种操作系统,并提供了多种编程接口,包括 Python 和 C++。

  • PaddleOCR:PaddleOCR 是基于 PaddlePaddle 的 OCR(Optical Character Recognition,光学字符识别)工具包,主要用于文字检测和识别任务。它可以识别多种语言的文字,并且支持多种预训练模型。PaddleOCR 提供了一个简单易用的接口,让用户能够快速地进行文字识别任务,同时也可以根据具体需求进行模型的微调和扩展。

  • Paddlenlp:Paddlenlp 是一个自然语言处理(NLP, Natural Language Processing)工具包,基于 PaddlePaddle 构建。它为文本分类、命名实体识别、情感分析等多种 NLP 任务提供了预训练模型和相应的 API。Paddlenlp 旨在帮助研究人员和开发者快速构建和测试 NLP 模型,并提供了丰富的中文预训练模型资源。

官网地址 :https://www.paddlepaddle.org.cn/

快速上手

创建一个文件,例如"chainlit_chat"

bash 复制代码
mkdir chainlit_chat

进入 chainlit_chat文件夹下,执行命令创建python 虚拟环境空间(需要提前安装好python sdkChainlit 需要python>=3.8。,具体操作,由于文章长度问题就不在叙述,自行百度),命令如下:

bash 复制代码
python -m venv .venv
  • 这一步是避免python第三方库冲突,省事版可以跳过
  • .venv是创建的虚拟空间文件夹可以自定义

接下来激活你创建虚拟空间,命令如下:

bash 复制代码
#linux or mac
source .venv/bin/activate
#windows
.venv\Scripts\activate

在项目根目录下创建requirements.txt,内容如下:

bash 复制代码
chainlit~=1.1.306
paddlepaddle
paddleocr
paddlenlp
colorama
openai

在项目根目录下创建.env环境变量,配置如下:

bash 复制代码
OPENAI_BASE_URL="https://open.bigmodel.cn/api/paas/v4"
OPENAI_API_KEY="your-api-key"
  • OPENAI_BASE_URL 是 openai的api 的base地址,如果你能直接使用openai,这可以不设置,这里我用的是智谱GLM-4,因为他和openai的接口兼容
  • OPENAI_API_KEY 替换成你自己的API 密匙

执行以下命令安装依赖:

bash 复制代码
pip install -r .\requirements.txt
  • 安装后,项目根目录下会多出.chainlit.files文件夹和chainlit.md文件

在项目根目录下创建app.py文件,代码如下:

python 复制代码
import base64
import re

import chainlit as cl
from chainlit.input_widget import Select, Slider, Switch
from openai import AsyncOpenAI
from paddleocr import PaddleOCR

from fp import ocr_nlp_processor

# 初始化 PaddleOCR 模型
# 需运行一次以下载并加载模型到内存中
ocr = PaddleOCR(use_angle_cls=True, lang="ch")

client = AsyncOpenAI()


def encode_image(image_path):
    with open(image_path, "rb") as image_file:
        return base64.b64encode(image_file.read()).decode("utf-8")


@cl.on_settings_update
async def on_settings_update(settings: cl.chat_settings):
    cl.user_session.set("settings", settings)


# 提取数字
def get_number(value):
    # 提取文本中的数字部分
    number_pattern = re.compile(r'\d+')
    result = number_pattern.search(value)

    if result:
        extracted_number = result.group()
        return f"'{extracted_number}"  # 添加'
    else:
        return value  # 如果没有找到数字,直接返回原始值


@cl.step(type="tool", name="图片OCR识别")
async def tool(images, content):
    text_result = ""
    if images:
        for image in images:
            print(image)
            # 使用 PaddleOCR 模型进行文字识别
            result = ocr.ocr(image.path, cls=True)
            # 打印识别结果
            print(result)
            for idx in range(len(result)):
                res = result[idx]
                for line in res:
                    text_result += line[1][0]+'\n'
    ocr_nlp_processor(text_result)
    return text_result


@cl.on_chat_start
async def start_chat():
    settings = await cl.ChatSettings(
        [
            Select(
                id="Model",
                label="Model",
                values=["llava:7b", "qwen-vl-plus", "qwen-vl-max"],
                initial_index=0,
            ),
            Slider(
                id="Temperature",
                label="Temperature",
                initial=0.5,
                min=0,
                max=2,
                step=0.1,
            ),
            Slider(
                id="MaxTokens",
                label="MaxTokens",
                initial=1000,
                min=1000,
                max=3000,
                step=100,
            ),
            Switch(id="Streaming", label="Stream Tokens", initial=True),
        ]
    ).send()
    cl.user_session.set("settings", settings)
    content = "你好,我是泰山AI智能客服,有什么可以帮助您吗?"
    msg = cl.Message(content=content)
    await msg.send()


@cl.on_message
async def on_message(message: cl.Message):
    images = [file for file in message.elements if "image" in file.mime]
    msg = cl.Message(content="")
    await msg.send()
    tool_res = await tool(images, message.content)
    await msg.stream_token(tool_res)
    await msg.update()


async def on_message1(message: cl.Message):
    settings = cl.user_session.get("settings")
    streaming = settings['Streaming']
    images = [file for file in message.elements if "image" in file.mime]
    msg = cl.Message(content="")
    await msg.send()
    tool_res = await tool(images)
    elements = []
    if tool_res:
        messages = [
            {
                "role": "user",
                "content": [{"type": "text", "text": f"{tool_res},将内容格式化为xx年纪xx科目输出"}],
            }
        ]
        response = await client.chat.completions.create(
            model=settings['Model'],
            messages=messages,
            temperature=settings['Temperature'],
            max_tokens=int(settings['MaxTokens']),
            stream=streaming
        )
        if streaming:
            async for part in response:
                if token := part.choices[0].delta.content or "":
                    await msg.stream_token(token)
                    msg.elements = elements
        else:
            if token := response.choices[0].message.content or "":
                await msg.stream_token(token)
    else:
        await msg.stream_token("不能识别")
    await msg.update()

在项目根目录下创建fp.py文件,代码如下:

python 复制代码
import os
import time

from colorama import Fore
from paddlenlp import Taskflow
from paddleocr import PaddleOCR

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

# 获取当前脚本文件的目录
script_directory = os.path.dirname(os.path.abspath(__file__))

# 构建 blank.jpg 的路径
blank_image_path = os.path.join(script_directory, "blank.jpg")


def ocr_nlp_processor(text):
    """
    将OCR和NLP结合处理的函数。

    返回:
    无,直接将识别结果写入到指定的输出文件中。
    """
    schema = ['教育部审定', '年级', '出版社']
    # 初始化NLP任务
    docPromptTask = Taskflow("information_extraction", schema=schema)

    # 记录程序开始时间
    start_time = time.time()
    # 执行NLP任务
    nlp_result = docPromptTask(text)
    print('nlp_result', nlp_result)
    # 计算程序执行时间
    end_time = time.time()
    total_time_seconds = end_time - start_time

    # 打印文件处理完成的消息和运行总耗时
    print(Fore.GREEN + f"==> 本次NLP运行总耗时:{total_time_seconds:.2f} 秒")

运行应用程序

要启动 Chainlit 应用程序,请打开终端并导航到包含的目录app.py。然后运行以下命令:

bash 复制代码
 chainlit run app.py -w   
  • -w标志告知 Chainlit 启用自动重新加载,因此您无需在每次更改应用程序时重新启动服务器。您的聊天机器人 UI 现在应该可以通过http://localhost:8000访问。
  • 自定义端口可以追加--port 80

启动后界面如下:

  • 上传图片通过paddleocr识别到了图片中的文字信息,当然对于字体特别小的,可能会识别错误
  • 再使用paddlenlp框架提取中文本中你需要的信息即可

相关文章推荐

《Chainlit快速实现AI对话应用的界面定制化教程》
《Chainlit接入FastGpt接口快速实现自定义用户聊天界面》
《使用 Xinference 部署本地模型》
《Fastgpt接入Whisper本地模型实现语音输入》
《Fastgpt部署和接入使用重排模型bge-reranker》
《Fastgpt部署接入 M3E和chatglm2-m3e文本向量模型》
《Fastgpt 无法启动或启动后无法正常使用的讨论(启动失败、用户未注册等问题这里)》
《vllm推理服务兼容openai服务API》
《vLLM模型推理引擎参数大全》
《解决vllm推理框架内在开启多显卡时报错问题》
《Ollama 在本地快速部署大型语言模型,可进行定制并创建属于您自己的模型》

相关推荐
光芒再现dev6 小时前
已解决,部署GPTSoVITS报错‘AsyncRequest‘ object has no attribute ‘_json_response_data‘
运维·python·gpt·语言模型·自然语言处理
人工智能培训咨询叶梓7 小时前
探索开放资源上指令微调语言模型的现状
人工智能·语言模型·自然语言处理·性能优化·调优·大模型微调·指令微调
zzZ_CMing7 小时前
大语言模型训练的全过程:预训练、微调、RLHF
人工智能·自然语言处理·aigc
软工菜鸡7 小时前
预训练语言模型BERT——PaddleNLP中的预训练模型
大数据·人工智能·深度学习·算法·语言模型·自然语言处理·bert
放飞自我的Coder7 小时前
【python ROUGE BLEU jiaba.cut NLP常用的指标计算】
python·自然语言处理·bleu·rouge·jieba分词
vivid_blog7 小时前
大语言模型(LLM)入门级选手初学教程 III
人工智能·语言模型·自然语言处理
qzhqbb9 小时前
语言模型的采样方法
人工智能·语言模型·自然语言处理
qzhqbb9 小时前
基于 Transformer 的语言模型
人工智能·语言模型·自然语言处理·transformer
OCR_wintone42110 小时前
易泊车牌识别相机,助力智慧工地建设
人工智能·数码相机·ocr
诗句藏于尽头10 小时前
基于百度飞桨paddle的paddlepaddle2.4.2等系列项目的运行
百度·paddlepaddle·paddle