digital-garbageandClaude Sonnet 5.5
5718a0d7e3
Co-Authored-By: Claude Sonnet 5.5 <noreply@anthropic.com>
46 lines
2.0 KiB
Python
46 lines
2.0 KiB
Python
"""The text actually sent as a scene's prompt at generation time.
|
|
|
|
`Scene.text` (core/projects.py) stays the user's literal typed words -- this
|
|
module never writes back to it. It only ever runs at the moment a prompt
|
|
value is about to leave the app for a real run, so editing a scene and
|
|
looking at what it produced always shows what was actually typed, not an
|
|
expanded copy nothing can trace back.
|
|
|
|
Order matches v4's: anchor and postfix are joined to the scene text FIRST,
|
|
then shortcuts expand across the whole combined string (a shortcut trigger
|
|
sitting right at the anchor/scene boundary still matches), then `$variables`
|
|
resolve last. Resolving last means a variable's OWN text is never re-scanned
|
|
for shortcut triggers -- only the anchor/scene/postfix text is -- but a
|
|
variable may still reference another variable, since resolve_variables()
|
|
recurses through the variable map itself (see its own docstring for the
|
|
cycle-safety that recursion needs).
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import re
|
|
|
|
from . import shortcuts as shortcuts_mod
|
|
|
|
|
|
#: A typed [A|B], found in the raw text (before any shortcut or variable has put brackets of its own in).
|
|
_TYPED_ALTERNATIVES = re.compile(r"(?<!\\)\[([^\[\]|]+(?:\|[^\[\]|]+)+)\]")
|
|
|
|
|
|
def _protect(m):
|
|
opts = [o.strip() for o in m.group(1).split("|") if o.strip()]
|
|
if len(opts) < 2:
|
|
return m.group(0)
|
|
return shortcuts_mod.ALT_OPEN + shortcuts_mod.ALT_SEP.join(opts) + shortcuts_mod.ALT_CLOSE
|
|
|
|
|
|
def build(scene_text: str, *, anchor: str = "", postfix: str = "",
|
|
postfix_enabled: bool = True, variables=None, shortcuts=None, seed: int = 0,
|
|
commit: bool = False) -> str:
|
|
anchor = (anchor or "").strip()
|
|
postfix = (postfix or "").strip() if postfix_enabled else ""
|
|
combined = " ".join(p for p in (anchor, str(scene_text or "").strip(), postfix) if p)
|
|
combined = _TYPED_ALTERNATIVES.sub(_protect, combined)
|
|
expanded = shortcuts_mod.expand(combined, shortcuts=shortcuts, seed=seed, commit=commit)
|
|
return shortcuts_mod.resolve_variables(expanded, variables)
|