Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-12-15 11:43:21 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

save metadata to images

Browse Source
This commit is contained in:
pixxy 2025-07-17 22:29:47 +01:00
parent 0be33acb57
commit d174f71c59
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
wgp.py
View File

@ -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"),