Allow escaping commas in XYZ grid

AUTOMATIC1111 · Jun 9, 2024 · d13e2e0 · d13e2e0
1 parent feee37d
commit d13e2e0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/scripts/xyz_grid.py b/scripts/xyz_grid.py
@@ -212,7 +212,10 @@ def list_to_csv_string(data_list):
 
 
 def csv_string_to_list_strip(data_str):
-    return list(map(str.strip, chain.from_iterable(csv.reader(StringIO(data_str)))))
+    placeholder = "<<<COMMA>>>"
+    data_str = data_str.replace(r"\,", placeholder) # Replace escaped commas with the placeholder
+    values = chain.from_iterable(csv.reader(StringIO(data_str)))
+    return list(map(lambda s: s.replace(placeholder, ",").strip(), values))
 
 
 class AxisOption: