Update images_history.py

images_history.py

@@ -28,20 +28,6 @@ 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'])
@@ -65,7 +51,6 @@ 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)
@@ -76,6 +61,5 @@ def img_history_ui(tab):
     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