chunzhi/comfyui-deploy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(plugin): support preview image upload

Browse Source
This commit is contained in:
BennyKok 2023-12-21 19:52:39 +08:00
parent 04ca3760ce
commit 9cc3c6ffe3
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
custom_routes.py
View File

@ -201,7 +201,7 @@ def update_run(prompt_id, status: Status):
requests.post(status_endpoint, json=body) requests.post(status_endpoint, json=body)
async def upload_file(prompt_id, filename, subfolder=None, type="image/png"): async def upload_file(prompt_id, filename, subfolder=None, content_type="image/png", type="output"):
""" """
Uploads file to S3 bucket using S3 client object Uploads file to S3 bucket using S3 client object
:return: None :return: None
@ -213,7 +213,7 @@ async def upload_file(prompt_id, filename, subfolder=None, type="image/png"):
return return
if output_dir is None: if output_dir is None:
output_dir = folder_paths.get_directory_by_type("output") output_dir = folder_paths.get_directory_by_type(type)
if output_dir is None: if output_dir is None:
return return
@ -231,8 +231,6 @@ async def upload_file(prompt_id, filename, subfolder=None, type="image/png"):
file_upload_endpoint = prompt_metadata[prompt_id]['file_upload_endpoint'] file_upload_endpoint = prompt_metadata[prompt_id]['file_upload_endpoint']
content_type = type
filename = quote(filename) filename = quote(filename)
prompt_id = quote(prompt_id) prompt_id = quote(prompt_id)
content_type = quote(content_type) content_type = quote(content_type)
@ -264,11 +262,11 @@ async def update_run_with_output(prompt_id, data):
try: try:
images = data.get('images', []) images = data.get('images', [])
for image in images: for image in images:
await upload_file(prompt_id, image.get("filename"), subfolder=image.get("subfolder"), type=image.get("type", "image/png")) await upload_file(prompt_id, image.get("filename"), subfolder=image.get("subfolder"), type=image.get("type"), content_type=image.get("content_type", "image/png"))
files = data.get('files', []) files = data.get('files', [])
for file in files: for file in files:
await upload_file(prompt_id, file.get("filename"), subfolder=file.get("subfolder"), type=file.get("type", "image/png")) await upload_file(prompt_id, file.get("filename"), subfolder=file.get("subfolder"), type=file.get("type"), content_type=image.get("content_type", "image/png"))
except Exception as e: except Exception as e:
error_type = type(e).__name__ error_type = type(e).__name__
@ -276,9 +274,11 @@ async def update_run_with_output(prompt_id, data):
body = { body = {
"run_id": prompt_id, "run_id": prompt_id,
"output_data": { "output_data": {
"type": error_type, "error": {
"message": str(e), "type": error_type,
"stack_trace": stack_trace "message": str(e),
"stack_trace": stack_trace
}
} }
} }
print(body) print(body)