diff --git a/scripts/dream_web.py b/scripts/dream_web.py
index 7c9155d926..a363dd32ad 100644
--- a/scripts/dream_web.py
+++ b/scripts/dream_web.py
@@ -45,7 +45,7 @@ class DreamServer(BaseHTTPRequestHandler):
steps = steps);
print(f"Prompt generated with output: {outputs}")
- outputs = [x + [prompt] for x in outputs] # Append prompt to each output
+ outputs = [x + [post_data] for x in outputs] # Append config to each output
result = {'outputs': outputs}
self.wfile.write(bytes(json.dumps(result), "utf-8"))
diff --git a/scripts/static/index.html b/scripts/static/index.html
index d5b27e83ef..6d8697846d 100644
--- a/scripts/static/index.html
+++ b/scripts/static/index.html
@@ -52,12 +52,20 @@
let output_node = document.createElement("img");
output_node.src = output[0];
- let alt_text = output[1].toString() + " | " + output[2];
+ let output_config = output[2];
+ let alt_text = output[1].toString() + " | " + output_config.prompt;
output_node.alt = alt_text;
output_node.title = alt_text;
- // Update seed on click
+
+ // Reload image config
output_node.addEventListener('click', () => {
+ let form = document.querySelector("#generate_form");
+ for (const [k, v] of new FormData(form)) {
+ form.querySelector(`*[name=${k}]`).value = output_config[k];
+ }
document.querySelector("#seed").value = output[1];
+
+ save_fields(document.querySelector("#generate_form"));
});
document.querySelector("#results").prepend(output_node);
@@ -115,6 +123,10 @@
document.querySelector("#generate_form").addEventListener('change', (e) => {
save_fields(e.target.form);
});
+ document.querySelector("#reset").addEventListener('click', (e) => {
+ document.querySelector("#seed").value = -1;
+ save_fields(e.target.form);
+ });
load_fields(document.querySelector("#generate_form"));
};
@@ -143,6 +155,7 @@
•
+