From d174f71c593606deb3a6513ffe350373d620e198 Mon Sep 17 00:00:00 2001 From: pixxy Date: Thu, 17 Jul 2025 22:29:47 +0100 Subject: [PATCH] save metadata to images --- wgp.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/wgp.py b/wgp.py index bdb7327..37651b5 100644 --- a/wgp.py +++ b/wgp.py @@ -4738,11 +4738,15 @@ def generate_video( if metadata_choice == "json": with open(path.replace(f'.{extension}', '.json'), 'w') as f: json.dump(configs, f, indent=4) - elif metadata_choice == "metadata" and not is_image: - from mutagen.mp4 import MP4 - file = MP4(path) - file.tags['©cmt'] = [json.dumps(configs)] - file.save() + elif metadata_choice == "metadata": + if is_image: + with Image.open(path) as img: + img.save(path, comment=json.dumps(configs)) + else: + from mutagen.mp4 import MP4 + file = MP4(path) + file.tags['©cmt'] = [json.dumps(configs)] + file.save() if is_image: print(f"New image saved to Path: "+ path) else: @@ -7621,7 +7625,7 @@ def generate_configuration_tab(state, blocks, header, model_choice, prompt_enhan metadata_choice = gr.Dropdown( choices=[ ("Export JSON files", "json"), - ("Add metadata to video", "metadata"), + ("Embed metadata (Exif tag)", "metadata"), ("Neither", "none") ], value=server_config.get("metadata_type", "metadata"),