fix two bug

This commit is contained in:
yfszzx 2022-10-17 22:28:43 +08:00
parent 9d702b16f0
commit c408a0b41c
2 changed files with 2 additions and 3 deletions

View File

@ -11,7 +11,6 @@ python = sys.executable
git = os.environ.get('GIT', "git") git = os.environ.get('GIT', "git")
index_url = os.environ.get('INDEX_URL', "") index_url = os.environ.get('INDEX_URL', "")
def extract_arg(args, name): def extract_arg(args, name):
return [x for x in args if x != name], name in args return [x for x in args if x != name], name in args

View File

@ -181,6 +181,7 @@ def delete_image(delete_num, name, filenames, image_index, visible_num):
return new_file_list, 1, visible_num return new_file_list, 1, visible_num
def save_image(file_name): def save_image(file_name):
if file_name is not None and os.path.exists(file_name):
shutil.copy2(file_name, opts.outdir_save) shutil.copy2(file_name, opts.outdir_save)
def get_recent_images(page_index, step, filenames): def get_recent_images(page_index, step, filenames):
@ -327,7 +328,6 @@ def create_history_tabs(gr, sys_opts, run_pnginfo, switch_dict):
opts = sys_opts opts = sys_opts
loads_files_num = int(opts.images_history_num_per_page) loads_files_num = int(opts.images_history_num_per_page)
num_of_imgs_per_page = int(opts.images_history_num_per_page * opts.images_history_pages_num) num_of_imgs_per_page = int(opts.images_history_num_per_page * opts.images_history_pages_num)
backup_flag = opts.images_history_backup
with gr.Blocks(analytics_enabled=False) as images_history: with gr.Blocks(analytics_enabled=False) as images_history:
with gr.Tabs() as tabs: with gr.Tabs() as tabs:
for tab in ["txt2img", "img2img", "extras", "saved"]: for tab in ["txt2img", "img2img", "extras", "saved"]: