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

feat: embed file public status into image output

Browse Source
This commit is contained in:
bennykok 2024-09-03 23:07:48 -07:00
parent 9cd626e1f6
commit 4df9d38e56
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
custom_routes.py
View File

@ -1244,6 +1244,8 @@ async def upload_file(prompt_id, filename, subfolder=None, content_type="image/p
if file_download_url is not None: if file_download_url is not None:
item["url"] = file_download_url item["url"] = file_download_url
item["upload_duration"] = end_time - start_time item["upload_duration"] = end_time - start_time
if ok.get("is_public") is not None:
item["is_public"] = ok.get("is_public")
def have_pending_upload(prompt_id): def have_pending_upload(prompt_id):
if prompt_id in prompt_metadata and len(prompt_metadata[prompt_id].uploading_nodes) > 0: if prompt_id in prompt_metadata and len(prompt_metadata[prompt_id].uploading_nodes) > 0: