Update images_history.py

images_history.py

@@ -28,6 +28,20 @@ def get_img_info(evt: gr.SelectData):
     index = evt.index
     return index, info, items
 
+def del_img(index, folder, page):
+    if index < 0:
+        return gr.Gallery(selected_index=None)
+    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]
+    os.remove(res[index])
+    res.pop(index)
+    return gr.Gallery(value=res, selected_index=None)
+
 def img_history_ui(tab):
     os.makedirs(client_config['save_path'], exist_ok=True)
     all_dirs = os.listdir(client_config['save_path'])
@@ -51,6 +65,7 @@ def img_history_ui(tab):
             selected_index = gr.Number(value=-1, visible=False, precision=0)
             with gr.Row():
                 send_btn = gr.Button('参数发送到文生图')
+                del_btn = gr.Button('Delete')
     
     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)
@@ -60,5 +75,7 @@ def img_history_ui(tab):
     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])
+    history_gallery.change(lambda: -1, outputs=selected_index)
+    del_btn.click(del_img, inputs=[selected_index, img_dir_path, page_index], outputs=history_gallery)
 
     return send_btn, items