Update images_history.py

images_history.py

@@ -3,9 +3,11 @@ import gradio as gr
 import toml
 from math import ceil
 from pnginfo import read_info_from_image
+from huggingface_hub import HfApi
 
 client_config = toml.load("config.toml")['client']
 num_of_imgs_per_page = 60
+api = HfApi()
     
 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))]
@@ -28,6 +30,21 @@ 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])
+    api.delete_file(path_in_repo=res[index], repo_id="P01yH3dr0n/naimages", repo_type="dataset", token=os.environ.get("hf_token"))
+    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 +68,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)
@@ -61,5 +79,6 @@ 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