check for text attr and save

18379dea · Chris · edcff9ab · 18379dea
Commit 18379dea authored Aug 29, 2023 by Chris
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

server.py server.py +3 -2

No files found.
--- a/server.py
+++ b/server.py
@@ -235,6 +235,7 @@ class PromptServer():
                if os.path.isfile(file):
                    with Image.open(file) as original_pil:
                        metadata = PngInfo()
+                        if hasattr(original_pil,'text'):
                            for key in original_pil.text:
                                metadata.add_text(key, original_pil.text[key])
                        original_pil = original_pil.convert('RGBA')