OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-04-06 21:09:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Make gallery return images

Browse Source
This commit is contained in:
missionfloyd 2024-04-22 00:54:16 -06:00
parent dcb73d4373
commit 4e8dfa3af5
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
modules/ui_common.py
View File

@ -5,7 +5,6 @@ import html
import os import os
import gradio as gr import gradio as gr
from PIL import Image
from modules import call_queue, shared, ui_tempdir, util from modules import call_queue, shared, ui_tempdir, util
import modules.images import modules.images
@ -113,10 +112,8 @@ def save_files(js_data, images, do_make_zip, index):
writer.writerow(fields) writer.writerow(fields)
for image_index, filedata in enumerate(images, start_index): for image_index, filedata in enumerate(images, start_index):
image = Image.open(filedata[0]) image = filedata[0]
is_grid = image_index < p.index_of_first_image is_grid = image_index < p.index_of_first_image
p.batch_index = image_index-1 p.batch_index = image_index-1
parameters = parameters_copypaste.parse_generation_parameters(data["infotexts"][image_index], []) parameters = parameters_copypaste.parse_generation_parameters(data["infotexts"][image_index], [])
@ -181,7 +178,7 @@ def create_output_panel(tabname, outdir, toprow=None):
with gr.Column(variant='panel', elem_id=f"{tabname}_results_panel"): with gr.Column(variant='panel', elem_id=f"{tabname}_results_panel"):
with gr.Group(elem_id=f"{tabname}_gallery_container"): with gr.Group(elem_id=f"{tabname}_gallery_container"):
res.gallery = gr.Gallery(label='Output', show_label=False, elem_id=f"{tabname}_gallery", columns=4, preview=True, height=shared.opts.gallery_height or None, interactive=False) res.gallery = gr.Gallery(label='Output', show_label=False, elem_id=f"{tabname}_gallery", columns=4, preview=True, height=shared.opts.gallery_height or None, interactive=False, type="pil")
with gr.Row(elem_id=f"image_buttons_{tabname}", elem_classes="image-buttons"): with gr.Row(elem_id=f"image_buttons_{tabname}", elem_classes="image-buttons"):
open_folder_button = ToolButton(folder_symbol, elem_id=f'{tabname}_open_folder', visible=not shared.cmd_opts.hide_ui_dir_config, tooltip="Open images output directory.") open_folder_button = ToolButton(folder_symbol, elem_id=f'{tabname}_open_folder', visible=not shared.cmd_opts.hide_ui_dir_config, tooltip="Open images output directory.")