Add exception handling to execution of image generation script

[zombocom-ai] / index.js

diff --git a/index.js b/index.js
index 0ee5e4359255f4afe794114cd043950d0da2c85e..638c59dc6543fbab851e179d9b59b74aa123c51c 100644 (file)
--- a/index.js
+++ b/index.js
@@ -84,7 +84,12 @@ io.on('connection', (socket) => {
             child.stderr.on('data', (data) => {
                 console.log("Error occurred during generate-image: " + data);
             });
-            const { stdout, stderr } = await promise;
+            try {
+                const { stdout, stderr } = await promise;
+            } catch(e) {
+                console.error(e);
+            }
+            socket.emit('generation-done');
         }
         generate_image(request['code'], request['prompt']);
     });