Stable Diffusion image metadata viewer

Show Stable Diffusion generated image's metadata

Versión del día 7/2/2023. Echa un vistazo a la versión más reciente.

Tendrás que instalar una extensión para tu navegador como Tampermonkey, Greasemonkey o Violentmonkey si quieres utilizar este script.

You will need to install an extension such as Tampermonkey to install this script.

Necesitarás instalar una extensión como Tampermonkey o Violentmonkey para instalar este script.

Necesitarás instalar una extensión como Tampermonkey o Userscripts para instalar este script.

Necesitará instalar una extensión como Tampermonkey para instalar este script.

Necesitarás instalar una extensión para administrar scripts de usuario si quieres instalar este script.

(Ya tengo un administrador de scripts de usuario, déjame instalarlo)

Necesitará instalar una extensión como Stylus para instalar este estilo.

Necesitará instalar una extensión como Stylus para instalar este estilo.

Necesitará instalar una extensión como Stylus para instalar este estilo.

Necesitará instalar una extensión del gestor de estilos de usuario para instalar este estilo.

Necesitará instalar una extensión del gestor de estilos de usuario para instalar este estilo.

Necesitará instalar una extensión del gestor de estilos de usuario para instalar este estilo.

(Ya tengo un administrador de estilos de usuario, déjame instalarlo)

// ==UserScript==
// @name         Stable Diffusion image metadata viewer
// @namespace    https://github.com/himuro-majika
// @version      0.1
// @description  Show Stable Diffusion generated image's metadata
// @author       himuro_majika
// @match        http://*/*.png
// @match        http://*/*.jpg
// @match        http://*/*.jpeg
// @match        https://*/*.png
// @match        https://*/*.jpg
// @match        https://*/*.jpeg
// @match        file:///*.png
// @match        file:///*.jpg
// @match        file:///*.jpeg
// @require      https://cdn.jsdelivr.net/npm/[email protected]/exif.min.js
// @license      MIT
// @grant        none
// ==/UserScript==

(function() {
    'use strict';

    const img = document.images[0];
    const ext = img.src.match(/[^.]+$/)[0].toLowerCase();

    if ( ext == "png") {
        getImgData(img)
    } else if ( ext == "jpg" || ext == "jpeg") {
        getExif(img);
    }
    async function getImgData(img) {
        const imgtext = (await(await fetch(img.src)).text());
        // console.log(imgtext);
        const text = imgtext.match(/(?<=Xt(?:parameters|Description|Comment)\0*)([^\0]+)/ug);
        if (!text) return;
        // console.log(text);
        makeButton();
        makeData(text.join(""));
    }

    function getExif(img) {
        EXIF.getData(img, function() {
            const comment = EXIF.getTag(this, "UserComment");
            if (!comment) return;

            const decode = decodeUnicode(comment);
            if (!decode) return;

            makeButton();
            makeData(decode);
        });
    }

    function decodeUnicode(array) {
        const plain = array.map(t => t.toString(16).padStart(2, "0")).join("");
        if (!plain.match(/^554e49434f44450/)) {
            // console.log(array);
            return;
        }
        const hex = plain.replace(/^554e49434f44450[0-9]/, "").replace(/[0-9a-f]{4}/g, ",0x$&").replace(/^,/, "");
        const arhex = hex.split(",");
        let decode = "";
        arhex.forEach(v => {
            decode += String.fromCodePoint(v);
        })
        return decode;
    }

    function makeButton() {
        const button = document.createElement("button");
        button.innerHTML = "Show SD metadata";
        button.addEventListener("click", showModal);
        document.body.insertBefore(button, img);
    }

    function makeData(text) {
        const positive = extractPositivePrompt(text);
        const negative = extractNegativePrompt(text);
        const others = extractOthers(text);
        const container = document.createElement("div");
        container.id ="_gm_sipv_container";
        container.style.display = "none";
        container.style.width = "100%";
        const copybutton = location.protocol == "http:" ? "" : `<button class="_gm_sipv_copybutton" type="button" style="cursor: pointer; opacity: 0.5;">copy</button>`;
        container.innerHTML = `
<div style="color: #eee; width: 800px; margin-left: auto; margin-right: auto; z-index: 2; position: fixed; inset: auto 0; margin: auto; background: #000a; border-radius: 6px; box-shadow: #000 0px 0px 2px;">
    <div style="display:flex; justify-content: space-between; padding: 0px 10px;">
        <h5>Stable Diffusion image metadata</h5>
        <button id="_gm_sipv_closebutton" type="button" style="cursor: pointer; height: 4em; opacity: 0.5; padding: 1em; background: #0000; border: 0; width: 3em;">❎</button>
    </div>
    <div style="padding: 10px;">
		<div>
		    <div style="display:flex; justify-content: space-between;">
                <label>Positive Prompt</label>
                ${copybutton}
            </div>
			<textarea rows="6" style="display: block; width: 774px; background: #aaaa; border: 0px none; margin: 10px 0;">${positive}</textarea>
		</div>
		<div>
		    <div style="display:flex; justify-content: space-between;">
		    	<label>Negative Prompt</label>
                ${copybutton}
            </div>
            <textarea rows="6" style="display: block; width: 774px; background: #aaaa; border: 0px none; margin: 10px 0;">${negative}</textarea>
		</div>
		<div>
		    <div style="display:flex; justify-content: space-between;">
    			<label>Other info</label>
                ${copybutton}
            </div>
            <textarea rows="3" style="display: block; width: 774px; background: #aaaa; border: 0px none; margin: 10px 0;">${others}</textarea>
		</div>
	</div>
</div>`;
        document.body.insertBefore(container, img);
        document.getElementById("_gm_sipv_closebutton").addEventListener("click", closeModal);
        document.querySelectorAll("._gm_sipv_copybutton").forEach(item => {
            item.addEventListener("click", copyText);
        });
    }

    function extractPositivePrompt(text) {
        try {
            let matchtext = text.match(/([^]+)Negative prompt: /) || text.match(/([^]+)Steps: /) || text.match(/([^]+){"steps"/);
            return matchtext[1];
        } catch (e) {
            // console.log(text);
            return "";
        }
    }

    function extractNegativePrompt(text) {
        try {
            let matchtext = text.match(/Negative prompt: ([^]+)Steps: /) || text.match(/"uc": "([^]+)"}/);
            return matchtext[1];
        } catch (e) {
            // console.log(text);
            return "";
        }
    }

    function extractOthers(text) {
        try {
            let matchtext = text.match(/(Steps: [^]+)/) || text.match(/{("steps"[^]+)"uc": /);
            return matchtext[1];
        } catch (e) {
            // console.log(text);
            return "";
        }
    }

    function showModal() {
        document.getElementById("_gm_sipv_container").style.display = "block";
    }

    function closeModal() {
        document.getElementById("_gm_sipv_container").style.display = "none";
    }

    function copyText() {
        const value = this.parentNode.parentNode.querySelector("textarea").value;
        navigator.clipboard.writeText(value);
    }

})();