Initial commit

Author: receyuki

.gitignore

@@ -0,0 +1,4 @@
+/venv/
+/test/
+/build/
+/dist/

.idea/.gitignore

@@ -0,0 +1,3 @@
+from PyInstaller.utils.hooks import collect_data_files
+
+datas = collect_data_files('tkinterdnd2')

.idea/inspectionProfiles/Project_Default.xml

@@ -0,0 +1,155 @@
+from PIL import Image, ImageTk
+from tkinter import TOP, Label, Frame, Text, LEFT, Scrollbar, VERTICAL, RIGHT, Y, BOTH, X, Canvas
+from tkinterdnd2 import *
+
+
+def read_info_from_image(image):
+    items = image.info or {}
+    geninfo = items.pop('parameters', None)
+    if "exif" in items:
+        exif = piexif.load(items["exif"])
+        exif_comment = (exif or {}).get("Exif", {}).get(piexif.ExifIFD.UserComment, b'')
+        try:
+            exif_comment = piexif.helper.UserComment.load(exif_comment)
+        except ValueError:
+            exif_comment = exif_comment.decode('utf8', errors="ignore")
+
+        if exif_comment:
+            items['exif comment'] = exif_comment
+            geninfo = exif_comment
+
+        for field in ['jfif', 'jfif_version', 'jfif_unit', 'jfif_density', 'dpi', 'exif',
+                      'loop', 'background', 'timestamp', 'duration']:
+            items.pop(field, None)
+
+    if items.get("Software", None) == "NovelAI":
+        try:
+            json_info = json.loads(items["Comment"])
+            sampler = sd_samplers.samplers_map.get(json_info["sampler"], "Euler a")
+
+            geninfo = f"""{items["Description"]}
+Negative prompt: {json_info["uc"]}
+Steps: {json_info["steps"]}, Sampler: {sampler}, CFG scale: {json_info["scale"]}, Seed: {json_info["seed"]}, Size: {image.width}x{image.height}, Clip skip: 2, ENSD: 31337"""
+        except Exception:
+            print("Error parsing NovelAI image generation parameters:", file=sys.stderr)
+            print(traceback.format_exc(), file=sys.stderr)
+
+    return geninfo, items
+
+
+def image_data(file):
+    try:
+        image = Image.open(file)
+        image.info
+        textinfo, _ = read_info_from_image(image)
+        return textinfo, None
+    except Exception:
+        pass
+
+    try:
+        text = data.decode('utf8')
+        assert len(text) < 10000
+        return text, None
+
+    except Exception:
+        pass
+
+    return '', None
+
+
+def image_info_format(text):
+    prompt_index = [text.index("\nNegative prompt:"),
+                    text.index("\nSteps:")]
+    positive = text[:prompt_index[0]]
+    negative = text[prompt_index[0] + 1 + len("Negative prompt: "):prompt_index[1]]
+    setting = text[prompt_index[1] + 1:]
+    return positive, negative, setting
+
+
+def display_info(event):
+    global image, image_tk, image_canvas
+    # clear text
+    positive_box.delete("1.0", "end")
+    negative_box.delete("1.0", "end")
+    setting_box.delete("1.0", "end")
+    if event.data.endswith(".png"):
+        with open(event.data, "rb") as f:
+            text_line, _ = image_data(f)
+            info = image_info_format(text_line)
+            # insert prompt
+            positive_box.insert("end", info[0])
+            negative_box.insert("end", info[1])
+            setting_box.insert("end", info[2])
+            image = Image.open(f)
+            # resize image to window size
+            if image.size[0] < image.size[1]:
+                resized = image.resize((int(image.size[0] / image.size[1] * image_frame.winfo_height()),
+                                        image_frame.winfo_height()), Image.LANCZOS)
+            else:
+                resized = image.resize((image_frame.winfo_width(),
+                                        int(image.size[1] / image.size[0] * image_frame.winfo_width())), Image.LANCZOS)
+            # display image
+            image_tk = ImageTk.PhotoImage(resized)
+            image_canvas.create_image(image_frame.winfo_width() / 2, 0, anchor="n", image=image_tk)
+            image_canvas.pack(side=LEFT, fill=BOTH, expand=True)
+
+
+def resize_image(event):
+    # resize image to window size
+    global image, image_canvas, image_tk
+    if image:
+        if image.size[0] < image.size[1]:
+            resized = image.resize((int(image.size[0] / image.size[1] * image_frame.winfo_height()),
+                                    image_frame.winfo_height()), Image.NEAREST)
+        else:
+            resized = image.resize((image_frame.winfo_width(),
+                                    int(image.size[1] / image.size[0] * image_frame.winfo_width())), Image.NEAREST)
+        image_tk = ImageTk.PhotoImage(resized)
+        image_canvas.create_image(image_frame.winfo_width() / 2, 0, anchor="n", image=image_tk)
+        # image_canvas.pack(fill=BOTH)
+
+
+window = TkinterDnD.Tk()
+window.title('SD Prompt Reader')
+window.geometry("800x400")
+# window.attributes('-alpha', 0.95)
+# window.config(bg='white')
+
+prompt_frame = Frame(window)
+prompt_frame.pack(side=RIGHT, fill=BOTH, expand=True, padx=10, pady=10)
+
+positive_box = Text(prompt_frame, height=10)
+positive_box.pack(side=TOP, fill=BOTH, expand=True)
+
+negative_box = Text(prompt_frame, height=10)
+negative_box.pack(side=TOP, fill=BOTH, expand=True)
+
+setting_box = Text(prompt_frame, height=5)
+setting_box.pack(side=TOP, fill=BOTH, expand=True)
+
+scrollbar_positive = Scrollbar(positive_box, orient=VERTICAL)
+scrollbar_positive.pack(side=RIGHT, fill=Y)
+positive_box.configure(yscrollcommand=scrollbar_positive.set)
+scrollbar_positive.config(command=positive_box.yview)
+
+scrollbar_negative = Scrollbar(negative_box, orient=VERTICAL)
+scrollbar_negative.pack(side=RIGHT, fill=Y)
+negative_box.configure(yscrollcommand=scrollbar_negative.set)
+scrollbar_negative.config(command=negative_box.yview)
+
+image_frame = Frame(window)
+image_frame.pack(side=LEFT, fill=BOTH, expand=True, padx=10, pady=10)
+image_canvas = Canvas(image_frame)
+
+image = None
+image_tk = None
+# image = Image.open("dnd.png")
+# image_tk = ImageTk.PhotoImage(image)
+# image_canvas.create_image(0, 0, anchor="nw", image=image_tk)
+# image_canvas.pack(fill=BOTH, expand=True)
+
+window.drop_target_register(DND_FILES)
+window.dnd_bind('<<Drop>>', display_info)
+window.bind("<Configure>", resize_image)
+
+window.mainloop()

.idea/inspectionProfiles/profiles_settings.xml

@@ -0,0 +1,50 @@
+# -*- mode: python ; coding: utf-8 -*-
+
+
+block_cipher = None
+
+
+a = Analysis(
+    ['main.py'],
+    pathex=[],
+    binaries=[],
+    datas=[('dnd.png', '.')],
+    hiddenimports=["collect_data_files"],
+    hookspath=['.'],
+    hooksconfig={},
+    runtime_hooks=[],
+    excludes=[],
+    win_no_prefer_redirects=False,
+    win_private_assemblies=False,
+    cipher=block_cipher,
+    noarchive=False,
+)
+pyz = PYZ(a.pure, a.zipped_data, cipher=block_cipher)
+
+exe = EXE(
+    pyz,
+    a.scripts,
+    a.binaries,
+    a.zipfiles,
+    a.datas,
+    [],
+    name='main',
+    debug=False,
+    bootloader_ignore_signals=False,
+    strip=False,
+    upx=True,
+    upx_exclude=[],
+    runtime_tmpdir=None,
+    windowed=True,
+    disable_windowed_traceback=False,
+    argv_emulation=False,
+    target_arch=None,
+    codesign_identity=None,
+    entitlements_file=None,
+)
+app = BUNDLE(
+    exe,
+    name='SD Prompt Reader.app',
+    icon='icon.png',
+    bundle_identifier=None,
+)

.idea/misc.xml

@@ -0,0 +1,6 @@
+altgraph==0.17.3
+macholib==1.16.2
+Pillow==9.4.0
+pyinstaller==5.9.0
+pyinstaller-hooks-contrib==2023.1
+tkinterdnd2==0.3.0