Spaces:
Paused
Paused
File size: 3,246 Bytes
5893438 944cbf7 5893438 18639dd 5893438 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 |
import os
import gradio as gr
import toml
from math import ceil
from pnginfo import read_info_from_image
client_config = toml.load("config.toml")['client']
num_of_imgs_per_page = 60
def get_img_list(folder, page):
img_list = [os.path.join(client_config['save_path'], folder, f) for f in os.listdir(os.path.join(client_config['save_path'], folder))]
img_list.sort(key=lambda f: os.path.getctime(f), reverse=True)
if page == 0:
page = 1
elif page*num_of_imgs_per_page > len(img_list) or page < 0:
page = ceil(len(img_list)/num_of_imgs_per_page)
res = img_list[(page - 1)*num_of_imgs_per_page: page*num_of_imgs_per_page]
return gr.Gallery(value=res, selected_index=None), page
def update_img_dir():
all_dirs = os.listdir(client_config['save_path'])
all_dirs.sort(reverse=True)
img_dir_path = gr.update(choices=all_dirs)
return img_dir_path
def get_img_info(evt: gr.SelectData):
info, items = read_info_from_image(evt.value['image']['path'])
index = evt.index
return index, info, items
def img_history_ui(tab):
os.makedirs(client_config['save_path'], exist_ok=True)
all_dirs = os.listdir(client_config['save_path'])
all_dirs.sort(reverse=True)
with gr.Row():
img_dir_path = gr.Dropdown(choices=all_dirs, value=None, label='图片文件夹')
tab.select(fn=update_img_dir, inputs=None, outputs=img_dir_path)
with gr.Row():
with gr.Column(scale=2):
with gr.Row():
first_page = gr.Button('第一页')
prev_page = gr.Button('前一页')
page_index = gr.Number(value=1, label="页码", precision=0)
next_page = gr.Button('下一页')
end_page = gr.Button('末尾页')
history_gallery = gr.Gallery(show_label=False, columns=6)
with gr.Column(scale=1):
info = gr.HTML()
items = gr.JSON(value=None, visible=False)
selected_index = gr.Number(value=-1, visible=False, precision=0)
with gr.Row():
send_btn = gr.Button('参数发送到文生图')
img_dir_path.change(lambda path: get_img_list(path, 1) + ('', None), inputs=img_dir_path, outputs=[history_gallery, page_index, info, items])
img_dir_path.blur(update_img_dir, inputs=None, outputs=img_dir_path)
first_page.click(lambda path: get_img_list(path, 1) + ('', None), inputs=img_dir_path, outputs=[history_gallery, page_index, info, items])
prev_page.click(lambda path, p: get_img_list(path, p - 1) + ('', None), inputs=[img_dir_path, page_index], outputs=[history_gallery, page_index, info, items])
page_index.submit(lambda path, p: get_img_list(path, p) + ('', None), inputs=[img_dir_path, page_index], outputs=[history_gallery, page_index, info, items])
next_page.click(lambda path, p: get_img_list(path, p + 1) + ('', None), inputs=[img_dir_path, page_index], outputs=[history_gallery, page_index, info, items])
end_page.click(lambda path: get_img_list(path, -1) + ('', None), inputs=img_dir_path, outputs=[history_gallery, page_index, info, items])
history_gallery.select(get_img_info, inputs=None, outputs=[selected_index, info, items])
return send_btn, items
|