re

CSMNode.py

@@ -380,7 +380,7 @@ class CSMDialogRun:
     def get_speaker_text(self, text):
         import re
         
-        pattern = r'(\[s?S?1\]|\[s?S?2\])\s*(.*)'
+        pattern = r'(\[s?S?1\]|\[s?S?2\])\s*([\s\S]*?)(?=\[s?S?[12]\]|$)'
         matches = re.findall(pattern, text)
         
         labels = []

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "csm_mw"
 description = "ComfyUI node of Conversational Speech Model (CSM)."
-version = "1.1.2"
+version = "1.1.3"
 license = {file = "LICENSE"}
 
 [project.urls]