Gradio学习(四)—————实现一个文件上传下载的页面

斯特凡今天也很帅

于 2024-02-28 10:28:24 发布

阅读量486

点赞数 2

分类专栏： Python 文章标签： python

本文链接：https://blog.csdn.net/weixin_41311528/article/details/136338792

版权

Python 专栏收录该内容

12 篇文章 0 订阅

订阅专栏

import os

import gradio as gr
import tempfile
import shutil

def generate_file(file_obj):
    # global tmpdir
    print('临时文件夹地址：{}'.format('tmpdir'))
    FilePath=file_obj.name
    print('上传文件的地址：{}'.format(file_obj.name)) # 输出上传后的文件在gradio中保存的绝对地址

    #获取到上传后的文件的绝对路径后，其余的操作就和平常一致了

    # 将文件复制到临时目录中
    shutil.copy(file_obj.name, 'tmpdir')

    # 获取上传Gradio的文件名称
    FileName=os.path.basename(file_obj.name)
print(FilePath)
    # 获取拷贝在临时目录的新的文件地址

    # 打开复制到新路径后的文件
    with open(FilePath, 'rb') as file_obj:

        #在本地电脑打开一个新的文件，并且将上传文件内容写入到新文件
        outputPath=os.path.join('想要保存的目录位置，写在这里！',FileName)
        with open(outputPath,'wb') as w:
            w.write(file_obj.read())

    # 返回新文件的的地址（注意这里）
    return outputPath


inputs = gr.components.File(label="上传文件")
outputs = gr.components.File(label="下载文件")

  # 创建 Gradio 应用程序g
app = gr.Interface(fn=generate_file, inputs=inputs, outputs=outputs,   title="文件上传、并生成可下载文件demo",
        description="上传任何文件都可以，只要大小别超过你电脑的内存即可"
)

# 因为在虚拟机中启动，而不是pycharm 所以指定主机ip (1.1.1.1)和端口号,如果是在pycharm 中，无需指定
app.launch(server_name=“1.1.1.1”,server_port=7860,share=True,inbrowser=True)