From 76f210b4de0e02002cd765517c356a49b707f717 Mon Sep 17 00:00:00 2001
From: Carl Worth <cworth@cworth.org>
Date: Wed, 7 Dec 2022 16:35:49 -0800
Subject: [PATCH] Hide image-generation form while generation is happening

Giving the user a queue that something is happening (and helping to
avoid queueing the generation of multiple identical images).
---
 index.html | 7 +++++++
 index.js   | 1 +
 2 files changed, 8 insertions(+)

diff --git a/index.html b/index.html
index 334a390..a0ee564 100644
--- a/index.html
+++ b/index.html
@@ -142,10 +142,17 @@ mute.addEventListener("click", () => {
 
     zombo_form.addEventListener('submit', function(e) {
         e.preventDefault();
+        /* Hide the form while generation is happening. */
+        zombo_form.style.display = "none";
         socket.emit('generate', {"prompt": prompt.value, "code": code.value});
         prompt.value = '';
     });
 
+    socket.on('generation-done', () => {
+        /* Re-display the form now that image-generation is over. */
+        zombo_form.style.display = "grid";
+    });
+
     // TODO: Dynamically generate many different prompts here
     safety.addEventListener("click", () => {
         prompt.value = "Matte painting of a Samurai warrior";
diff --git a/index.js b/index.js
index 0ee5e43..ba01d67 100644
--- a/index.js
+++ b/index.js
@@ -85,6 +85,7 @@ io.on('connection', (socket) => {
                 console.log("Error occurred during generate-image: " + data);
             });
             const { stdout, stderr } = await promise;
+            socket.emit('generation-done');
         }
         generate_image(request['code'], request['prompt']);
     });
-- 
2.43.0