oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2025-12-06 07:12:10 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

UI: Fix an index error in the new image gallery

Browse source
This commit is contained in:
oobabooga 2025-12-02 11:20:52 -08:00
parent 9d07d3a229
commit 97281ff831
1 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
modules/ui_image_generation.py
View file

@ -290,8 +290,11 @@ def on_gallery_select(evt: gr.SelectData, current_page):
if evt.index is None:
return "", "Select an image to view its settings"
if not _image_cache:
get_all_history_images()
# Get the current page's images to find the actual file path
all_images = get_all_history_images()
all_images = _image_cache
total_images = len(all_images)
# Calculate the actual index in the full list
@ -506,9 +509,7 @@ def create_ui():
info="Enter HuggingFace path. Use : for branch, e.g. user/model:main"
)
shared.gradio['image_download_btn'] = gr.Button("Download", variant='primary')
shared.gradio['image_model_status'] = gr.Markdown(
value=f"Model: **{shared.settings['image_model_menu']}** (not loaded)" if shared.settings['image_model_menu'] != 'None' else "No model selected"
)
shared.gradio['image_model_status'] = gr.Markdown(value="")
def create_event_handlers():