| import re | |
| from typing import Dict | |
| def apply_style_args(data: Dict): | |
| prompt = data.get("prompt", None) | |
| if prompt is None: | |
| return | |
| result = re.match(r"\[style:(.*?)\]", prompt) | |
| if result is not None: | |
| style = result.group(1) | |
| data["style"] = style | |
| data["prompt"] = prompt.replace(f"[style:{style}]", "").strip() | |