break the realtime generation loop on error

typescripts/comfyui/comfyui.tsx

@@ -1627,6 +1627,7 @@ class ComfyWorkflowComponent extends React.Component<{}, { value?: number }> {
                                         outputs ?? {}
                                 } catch (e) {
                                     console.error(e)
+                                    break
                                 } finally {
                                     clearInterval(interval as NodeJS.Timeout)
                                     store.data.progress_value = 0

typescripts/comfyui/main_ui.tsx

@@ -467,12 +467,16 @@ async function generateComfyMode(
         //     }
         // }
         console.log('final_prompt: ', final_prompt)
-        const { outputs, separated_outputs } =
-            await comfyui_util.postPromptAndGetBase64JsonResult(
-                final_prompt,
-                separated_output_node_ids
-            )
-
+        try {
+            var { outputs, separated_outputs } =
+                await comfyui_util.postPromptAndGetBase64JsonResult(
+                    final_prompt,
+                    separated_output_node_ids
+                )
+        } catch (e) {
+            outputs = undefined
+            separated_outputs = undefined
+        }
         if (outputs) {
             image_url_list = Object.values(outputs).flat()
             image_base64_list = image_url_list.map((image_url) => {

typescripts/comfyui/util.ts

@@ -274,7 +274,7 @@ export async function postPromptAndGetBase64JsonResult(
     } catch (e) {
         console.error(e)
         app.showAlert(`${e}`)
-        return { outputs: undefined, separated_outputs: undefined }
+        throw e
     }
 }
 export const getFileFormat = (fileName: string): string =>
@@ -529,7 +529,7 @@ async function maskExpansion(
         },
     }
     try {
-        const { outputs, separated_outputs } =
+        var { outputs, separated_outputs } =
             await postPromptAndGetBase64JsonResult(prompt)
         if (outputs) {
             const expanded_mask = outputs['6'][0]
@@ -538,6 +538,8 @@ async function maskExpansion(
         // html_manip.setInitImageMaskSrc(expanded_mask)
     } catch (e) {
         console.error(e)
+        outputs = undefined
+        separated_outputs = undefined
     }
     return base64_mask
 }