code snippets for getting useful exif metadata from generative art outputs

exif.py

###############
# DreamStudio #
###############

import PIL
from PIL import Image, ExifTags

def get_exif(img):
    outv = {}
    for k, v in img.getexif().items():
        tag_name = ExifTags.TAGS[k]
        outv[tag_name] = v
    return outv

def prompt_from_exif_dreamstudio(image:PIL.Image.Image) -> str:
    exif = get_exif(image)
    return exif['ImageDescription']

if __name__ == '__main__':
    #import sys
    #fpath = sys.argv[1]
    fpath = "my_image.png"
    img = Image.open(fpath)
    prompt = prompt_from_exif_dreamstudio(img)
    print(prompt)
    
############
# InvokeAI #
############

# ? talk to PBaylies 

###################
# Automatic WebUI #
###################

# ?

#########################
# Huggingface Diffusers #
#########################

# ?

##############
# Midjourney #
##############

# ?