benny/Buffteks-Website
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix error when no generated image response

Browse Source
This commit is contained in:
Carl Zhang
2025-10-03 21:42:53 -05:00
parent 0375584d53
commit c4066ed15e
1 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
webpages/bufftools_pages/ai_image_editor.py
View File

@@ -75,14 +75,14 @@ def ai_image_editor():
st.info(text_response) st.info(text_response)
if image_response: if image_response:
st.image(image_response, width=300) st.image(image_response, width=300)
# Convert PIL image to bytes for download # Convert PIL image to bytes for download
img_byte_arr = BytesIO() img_byte_arr = BytesIO()
image_response.save(img_byte_arr, format="JPEG") image_response.save(img_byte_arr, format="JPEG")
st.download_button( st.download_button(
label="Download Edited Image", label="Download Edited Image",
data=img_byte_arr.getvalue(), data=img_byte_arr.getvalue(),
file_name="edited_image.jpeg", file_name="edited_image.jpeg",
mime="image/jpeg" mime="image/jpeg"
) )