From c4066ed15e444bbafe567cfb72050ff145119c41 Mon Sep 17 00:00:00 2001 From: Carl Zhang Date: Fri, 3 Oct 2025 21:42:53 -0500 Subject: [PATCH] fix error when no generated image response --- webpages/bufftools_pages/ai_image_editor.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/webpages/bufftools_pages/ai_image_editor.py b/webpages/bufftools_pages/ai_image_editor.py index 8991eacd7..c57989644 100644 --- a/webpages/bufftools_pages/ai_image_editor.py +++ b/webpages/bufftools_pages/ai_image_editor.py @@ -75,14 +75,14 @@ def ai_image_editor(): st.info(text_response) if image_response: st.image(image_response, width=300) - # Convert PIL image to bytes for download - img_byte_arr = BytesIO() - image_response.save(img_byte_arr, format="JPEG") - st.download_button( - label="Download Edited Image", - data=img_byte_arr.getvalue(), - file_name="edited_image.jpeg", - mime="image/jpeg" - ) + # Convert PIL image to bytes for download + img_byte_arr = BytesIO() + image_response.save(img_byte_arr, format="JPEG") + st.download_button( + label="Download Edited Image", + data=img_byte_arr.getvalue(), + file_name="edited_image.jpeg", + mime="image/jpeg" + )