0.23

pause_resume_stop.py

@@ -0,0 +1,63 @@
+import time
+from aiohttp import web
+from server import PromptServer
+import logging
+from pydub import AudioSegment
+from pydub.playback import play
+import os
+
+class Everything(str):
+    def __ne__(self, __value: object) -> bool:
+        return False
+
+class PauseResume:
+    is_paused = True
+    should_stop = False
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "input": (Everything("*"), {"forceInput": True})
+            }
+        }
+
+    RETURN_TYPES = (Everything("*"),)
+    RETURN_NAMES = ("output",)
+    FUNCTION = "loop_resume_or_stop"
+    CATEGORY = "Bjornulf"
+    
+    def play_audio(self):
+        audio_file = os.path.join(os.path.dirname(__file__), 'bell.m4a')
+        sound = AudioSegment.from_file(audio_file, format="m4a")
+        play(sound)
+
+    def loop_resume_or_stop(self, input):
+        self.play_audio()
+        self.input = input
+        while PauseResume.is_paused and not PauseResume.should_stop:
+            logging.info(f"PauseResume.is_paused: {PauseResume.is_paused}, PauseResume.should_stop: {PauseResume.should_stop}")
+            time.sleep(1)  # Sleep to prevent busy waiting
+        
+        if PauseResume.should_stop:
+            PauseResume.should_stop = False  # Reset for next run
+            PauseResume.is_paused = True
+            PauseResume.should_stop = False
+            raise Exception("Workflow stopped by user")
+        
+        PauseResume.is_paused = True
+        PauseResume.should_stop = False
+        return (self.input,)
+
+@PromptServer.instance.routes.get("/bjornulf_resume")
+async def resume_node(request):
+    logging.info("Resume node called")
+    PauseResume.is_paused = False
+    return web.Response(text="Node resumed")
+
+@PromptServer.instance.routes.get("/bjornulf_stop")
+async def stop_node(request):
+    logging.info("Stop node called")
+    PauseResume.should_stop = True
+    PauseResume.is_paused = False  # Ensure the loop exits
+    return web.Response(text="Workflow stopped")