Spaces:
Runtime error
Runtime error
File size: 1,987 Bytes
d4ea1a4 |
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 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
import os
import chainlit as cl
async def need_file_upload():
"""
When the user's question mentions handling files, you need to upload files, you can call this function.
Parameters: None
"""
if not os.path.exists('./tmp'):
os.mkdir('./tmp')
files = await cl.AskFileMessage(
content="Please upload a text file to begin!",
max_size_mb=50,
accept=[
"text/plain",
"image/png",
"image/jpeg",
"application/pdf",
"application/vnd.openxmlformats-officedocument.spreadsheetml.sheet", # for .xlsx files
"application/vnd.ms-excel", # for .xls files
"text/csv", # for .csv files
# More MIME types here as needed.
]).send()
file = files[0]
# 保存文件到paths目录下
# 判断paths目录是否存在
file_path = f"./tmp/{file.name}"
# 保存文件
content = file.content
file_name = file.name
file_type = file.type
# 保存文件
# content是bytes类型
with open(file_path, "wb") as f:
f.write(content)
return {
'type': 'file',
'path': file_path,
'name': file_name,
'file_type': file_type
}
async def show_images(paths: str):
"""
If your return contains images in png or jpg format, you can call this function to display the images.
Parameters: paths: The paths of the images as a comma-separated string.(required)
"""
path_list = paths.split(',')
elments = []
for i, path in enumerate(path_list):
tmp_image = cl.Image(name=f"image{i}",
path=path.strip(),
display="inline")
tmp_image.size = "large"
elments.append(tmp_image)
await cl.Message(content="",
elements=elments).send() # type: ignore
return {'description': '图片已经显示成功了，下面的回复中不再需要展示它了'}
|