Display spinner while image is being generated

[zombocom-ai] / index.js

diff --git a/index.js b/index.js
index 0ee5e4359255f4afe794114cd043950d0da2c85e..ba01d67ffbe60b85403b5026e514b51f34d9a389 100644 (file)
--- 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']);
     });