目录

前言

设计方案

代码具体优化

前言

补充前篇的一些优化。
场景是识别pdf文档，但还需要把pdf文档中的图片也保存下来，在知识库增强检索的时候，直接可以显示图片。

设计方案

1、保存知识库中的图片
2、存入我们的文件服务器中，比如minio
3、名称规则为：image_files/年/月/日/文件
4、在知识库中用[Image 图片路径]
5、在前端知识库增强检索中，根据这个默认替换成图片可以访问的方法

代码具体优化

主要在上一篇的基础上，添加这一段代码

images = page.images
                    if images:
                        image_text = ""
                        for i, img in enumerate(images):
                            # Generate a unique filename for the image
                            image_filename = f"image_files/{datetime.now().strftime('%Y/%m/%d')}/{uuid.uuid4()}_{page_number}_{i}.png"
                            # Save image to storage
                            image_bytes = io.BytesIO(img['stream'].get_data())
                            storage.save(image_filename, image_bytes.getvalue())
                            # Create a File object to get the signed URL
                            image_text += f"[Image {image_filename}] "
                        content += image_text