From 4df9d38e5697445fba172346ffbeba5fdbb5bd63 Mon Sep 17 00:00:00 2001 From: bennykok Date: Tue, 3 Sep 2024 23:07:48 -0700 Subject: [PATCH] feat: embed file public status into image output --- custom_routes.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/custom_routes.py b/custom_routes.py index 4f4ff7c..6af88a3 100644 --- a/custom_routes.py +++ b/custom_routes.py @@ -1244,6 +1244,8 @@ async def upload_file(prompt_id, filename, subfolder=None, content_type="image/p if file_download_url is not None: item["url"] = file_download_url 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): if prompt_id in prompt_metadata and len(prompt_metadata[prompt_id].uploading_nodes) > 0: