From 269fcf92d9f2dfe699eb1950176eeebbc1df6620 Mon Sep 17 00:00:00 2001
From: tesseractcat <tesseractcats@gmail.com>
Date: Wed, 24 Aug 2022 21:38:47 -0400
Subject: [PATCH] Reapply prompt config on image click

---
 scripts/dream_web.py      |  2 +-
 scripts/static/index.html | 17 +++++++++++++++--
 2 files changed, 16 insertions(+), 3 deletions(-)

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"));
       };
     </script>
@@ -143,6 +155,7 @@
           <span>&bull;</span>
           <label title="Set to -1 for random seed" for="seed">Seed:</label>
           <input value="-1" type="number" id="seed" name="seed">
+          <button type="button" id="reset">&olarr;</button>
         </fieldset>
       </form>
     </div>