Spaces:
Sleeping
Sleeping
File size: 1,686 Bytes
9898ee7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
import os
import gradio as gr
import tempfile
import shutil
def generate_file(file_obj):
global tmpdir
print('临时文件夹地址:{}'.format(tmpdir))
print('上传文件的地址:{}'.format(file_obj.name)) # 输出上传后的文件在gradio中保存的绝对地址
#获取到上传后的文件的绝对路径后,其余的操作就和平常一致了
# 将文件复制到临时目录中
shutil.copy(file_obj.name, tmpdir)
# 获取上传Gradio的文件名称
FileName=os.path.basename(file_obj.name)
# 获取拷贝在临时目录的新的文件地址
NewfilePath=os.path.join(tmpdir,FileName)
print(NewfilePath)
# 打开复制到新路径后的文件
with open(NewfilePath, 'rb') as file_obj:
#在本地电脑打开一个新的文件,并且将上传文件内容写入到新文件
outputPath=os.path.join(tmpdir,"New"+FileName)
with open(outputPath,'wb') as w:
w.write(file_obj.read())
# 返回新文件的的地址(注意这里)
return outputPath
def main():
global tmpdir
with tempfile.TemporaryDirectory(dir='.') as tmpdir:
# 定义输入和输出
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="上传任何文件都可以,只要大小别超过你电脑的内存即可"
)
# 启动应用程序
app.launch(share=True)
if __name__=="__main__":
main() |