From 20c6edb4788a3a9d235ec0cc600e1b1029a9b4e7 Mon Sep 17 00:00:00 2001
From: mrm987 <99248187+mrm987@users.noreply.github.com>
Date: Sun, 28 Dec 2025 10:26:55 +0900
Subject: [PATCH] Add files via upload

---
 __init__.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/__init__.py b/__init__.py
index 34f27ea..c60cafe 100644
--- a/__init__.py
+++ b/__init__.py
@@ -461,7 +461,13 @@ class MultiPromptGenerator:
         all_images = []
         counter = self._get_next_counter(save_prefix)
         
+        # 인터럽트 체크용
+        import comfy.model_management as mm
+        
         for idx, line in enumerate(lines):
+            # 중단 체크
+            mm.throw_exception_if_processing_interrupted()
+            
             # skip_indices에 있으면 스킵 (1-based)
             if (idx + 1) in skip_set:
                 print(f"[MultiPrompt] Skipping {idx + 1}/{len(lines)}: {line[:40]}...")
@@ -851,7 +857,13 @@ class NAIMultiPromptGenerator:
         all_images = []
         counter = self._get_next_counter(save_prefix)
         
+        # 인터럽트 체크용
+        import comfy.model_management as mm
+        
         for idx, line in enumerate(lines):
+            # 중단 체크
+            mm.throw_exception_if_processing_interrupted()
+            
             if (idx + 1) in skip_set:
                 print(f"[NAI MultiPrompt] Skipping {idx + 1}/{len(lines)}: {line[:40]}...")
                 continue