From 6b7d4877e69789cbed6cf3271cb060c97c6ec546 Mon Sep 17 00:00:00 2001
From: cmdr2 <secondary.cmdr2@gmail.com>
Date: Sat, 8 Oct 2022 15:56:56 +0530
Subject: [PATCH] Initial implementation of prompt matrix - creates multiple
 task entries for the different permutations

---
 ui/index.html    |  4 +--
 ui/media/main.js | 84 ++++++++++++++++++++++++++++++++++++++++++------
 2 files changed, 76 insertions(+), 12 deletions(-)

diff --git a/ui/index.html b/ui/index.html
index 8ec4fe61..aa524574 100644
--- a/ui/index.html
+++ b/ui/index.html
@@ -15,7 +15,7 @@
 <div id="container">
     <div id="top-nav">
         <div id="logo">
-            <h1>Stable Diffusion UI <small>v2.21 <span id="updateBranchLabel"></span></small></h1>
+            <h1>Stable Diffusion UI <small>v2.22 <span id="updateBranchLabel"></span></small></h1>
         </div>
         <ul id="top-nav-items">
             <li class="dropdown">
@@ -226,7 +226,7 @@
 </div>
 </body>
 
-<script src="media/main.js?v=32"></script>
+<script src="media/main.js?v=33"></script>
 <script>
 async function init() {
     await loadModifiers()
diff --git a/ui/media/main.js b/ui/media/main.js
index 0f2c7344..beac4891 100644
--- a/ui/media/main.js
+++ b/ui/media/main.js
@@ -596,16 +596,8 @@ function makeImage() {
         return
     }
 
-    let prompts = promptField.value
-    prompts = prompts.split('\n')
-    prompts.forEach(prompt => {
-        prompt = prompt.trim()
-        if (prompt === '') {
-            return
-        }
-
-        createTask(prompt)
-    })
+    let prompts = getPrompts()
+    prompts.forEach(createTask)
 
     initialText.style.display = 'none'
 }
@@ -747,6 +739,78 @@ function createTask(prompt) {
     taskQueue.unshift(task)
 }
 
+function getPrompts() {
+    let prompts = promptField.value
+    prompts = prompts.split('\n')
+
+    let promptsToMake = []
+
+    prompts.forEach(prompt => {
+        prompt = prompt.trim()
+        if (prompt === '') {
+            return
+        }
+
+        let promptMatrix = prompt.split('|')
+        prompt = promptMatrix.shift().trim()
+
+        promptsToMake.push(prompt)
+
+        promptMatrix = promptMatrix.map(p => p.trim())
+        promptMatrix = promptMatrix.filter(p => p !== '')
+
+        if (promptMatrix.length > 0) {
+            let promptPermutations = permutePrompts(prompt, promptMatrix)
+            promptsToMake = promptsToMake.concat(promptPermutations)
+        }
+    })
+
+    return promptsToMake
+}
+
+function permutePrompts(promptBase, promptMatrix) {
+    let prompts = []
+    let permutations = permute(promptMatrix)
+    permutations.forEach(perm => {
+        let prompt = promptBase
+
+        if (perm.length > 0) {
+            let promptAddition = perm.join(', ')
+            if (promptAddition.trim() === '') {
+                return
+            }
+
+            prompt += ', ' + promptAddition
+        }
+
+        prompts.push(prompt)
+    })
+
+    return prompts
+}
+
+function permute(arr) {
+    let permutations = []
+    let n = arr.length
+    let n_permutations = Math.pow(2, n)
+    for (let i = 0; i < n_permutations; i++) {
+        let perm = []
+        let mask = Number(i).toString(2).padStart(n, '0')
+
+        for (let idx = 0; idx < mask.length; idx++) {
+            if (mask[idx] === '1' && arr[idx].trim() !== '') {
+                perm.push(arr[idx])
+            }
+        }
+
+        if (perm.length > 0) {
+            permutations.push(perm)
+        }
+    }
+
+    return permutations
+}
+
 // create a file name with embedded prompt and metadata
 // for easier cateloging and comparison
 function createFileName(seed, outputFormat) {