Gradio入门学习笔记

一些比较好的参考文献

Gradio官网
如何使用 Gradio 创建聊天机器人

文件上传、分页下载功能

设想一个简单的功能,用户在界面上上传一个PDF文件,应用经过一系列的处理,把这个PDF的每一页都转成了png图片,并且打包压缩成一个压缩包,提供给用户下载。

这个功能可以很方便的使用gradio来实现。

python 复制代码
import gradio as gr
import fitz  # PyMuPDF
import os
import zipfile

def pdf_to_images(pdf_file):
    # 打开PDF文件
    doc = fitz.open(pdf_file)
    images = []

    for page_id in range(doc.page_count):
        page = doc[page_id]
        # 将页面转换为PNG图片
        pix = page.get_pixmap()
        img_bytes = pix.tobytes("png")
        images.append((img_bytes, f"{page_id+1}.png"))

    # 关闭PDF文档
    doc.close()

    # 创建一个临时目录来存储图片
    temp_dir = "temp_images"
    os.makedirs(temp_dir, exist_ok=True)

    # 保存图片到临时目录
    for img_bytes, img_name in images:
        with open(os.path.join(temp_dir, img_name), "wb") as f:
            f.write(img_bytes)

    # 创建压缩包
    zip_path = "images.zip"
    with zipfile.ZipFile(zip_path, "w") as zipf:
        for img_name in os.listdir(temp_dir):
            zipf.write(os.path.join(temp_dir, img_name), img_name)

    # 删除临时目录
    for img_name in os.listdir(temp_dir):
        os.remove(os.path.join(temp_dir, img_name))
    os.rmdir(temp_dir)

    return zip_path

def main():
    iface = gr.Interface(
        fn=pdf_to_images,
        inputs=gr.File(label="Upload PDF File"),
        outputs=gr.File(label="Download ZIP File"),
        title="PDF to Images Converter",
        description="Upload a PDF file and download a ZIP file containing all the pages as images."
    )
    iface.launch()

if __name__ == "__main__":
    main()


作者:大橙子打游戏
链接:https://juejin.cn/post/7382480523846467595
来源:稀土掘金
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
相关推荐
CryptoRzz8 分钟前
欧美(美股、加拿大股票、墨西哥股票)股票数据接口文档
java·服务器·开发语言·数据库·区块链
im_AMBER38 分钟前
CSS 01【基础语法学习】
前端·css·笔记·学习
摇滚侠38 分钟前
Spring Boot 3零基础教程,深度理解 Spring Boot 自动配置原理,笔记11
spring boot·笔记·后端
APItesterCris1 小时前
构建弹性数据管道:利用淘宝商品 API 进行流式数据采集与处理
linux·数据库·windows
九河云1 小时前
TOS + 数字孪生:集装箱码头的智能进化密码
大数据·服务器·网络·数据库·数字化转型
向阳花开_miemie1 小时前
Android音频学习(二十二)——音频接口
学习·音视频
胡萝卜3.01 小时前
深入理解string底层:手写高效字符串类
开发语言·c++·学习·学习笔记·string类·string模拟实现
手握风云-1 小时前
MySQL数据库精研之旅第十九期:存储过程,数据处理的全能工具箱(二)
数据库·mysql
孟意昶1 小时前
Doris专题17- 数据导入-文件格式
大数据·数据库·分布式·sql·doris
fanstering1 小时前
腾讯混元P3-SAM: Native 3D Part Segmentation
笔记·学习·3d·点云