From ff2c4b9a1b773b95686d5f3e546e1194de054694 Mon Sep 17 00:00:00 2001
From: Web Dev 9000 <rirath@gmail.com>
Date: Sun, 4 Sep 2022 17:50:13 -0700
Subject: [PATCH] Add ability to recreate variations via image click

---
 static/dream_web/index.js | 21 +++++++++++++++++----
 1 file changed, 17 insertions(+), 4 deletions(-)

diff --git a/static/dream_web/index.js b/static/dream_web/index.js
index ceaef32f09..27086ebd17 100644
--- a/static/dream_web/index.js
+++ b/static/dream_web/index.js
@@ -19,10 +19,24 @@ function appendOutput(src, seed, config) {
     outputNode.addEventListener('click', () => {
         let form = document.querySelector("#generate-form");
         for (const [k, v] of new FormData(form)) {
-	    if (k == 'initimg') { continue; }
-	    form.querySelector(`*[name=${k}]`).value = config[k];
+            form.querySelector(`*[name=${k}]`).value = config[k];
+        }
+        if (config.variation_amount > 0 || config.with_variations != '') {
+            document.querySelector("#seed").value = config.seed;
+        } else {
+            document.querySelector("#seed").value = seed;
+        }
+
+        if (config.variation_amount > 0) {
+            let oldVarAmt = document.querySelector("#variation_amount").value
+            let oldVariations = document.querySelector("#with_variations").value
+            let varSep = ''
+            document.querySelector("#variation_amount").value = 0;
+            if (document.querySelector("#with_variations").value != '') {
+                varSep = ","
+            }
+            document.querySelector("#with_variations").value = oldVariations + varSep + seed + ':' + config.variation_amount
         }
-        document.querySelector("#seed").value = seed;
 
         saveFields(document.querySelector("#generate-form"));
     });
@@ -60,7 +74,6 @@ async function generateSubmit(form) {
 
     // Convert file data to base64
     let formData = Object.fromEntries(new FormData(form));
-    formData.initimg_name = formData.initimg.name
     formData.initimg = formData.initimg.name !== '' ? await toBase64(formData.initimg) : null;
 
     let strength = formData.strength;