Add drag/drop param loading.

Drop an image or generational text onto the prompt bar, it loads the info for parsing.
This commit is contained in:
d8ahazard
2022-10-12 18:17:26 -05:00
parent cc5803603b
commit 54e0051bdd
4 changed files with 74 additions and 1 deletions

View File

@@ -463,3 +463,23 @@ def save_image(image, path, basename, seed=None, prompt=None, extension='png', i
txt_fullfn = None
return fullfn, txt_fullfn
def image_data(image_path):
file, ext = os.path.splitext(image_path.name)
data = {}
if "png" in ext:
image = Image.open(image_path.name, "r")
print(f"Image data requested for {image_path.name} {image.format} of {type(image)}")
try:
data = image.text["parameters"]
except Exception as e:
print(f"Exception: {e}")
pass
print(f"Image data: {data}")
if "txt" in ext:
myfile = open(image_path.name, 'r')
data = myfile.read()
myfile.close()
return data, None