# -*- coding: utf-8 -*- # Prompt Parser Analyzer — расширенная версия с интеграцией генерации (txt2img) # # Требования: modules.prompt_parser = «второй файл» (schedule_parser, get_schedule, at_step_from_schedule, ...) # Установка: положите этот файл в папку scripts/ (или в extensions/<...>/scripts/), перезапустите/Reload scripts. import os import re import json import difflib import importlib from typing import Any, List, Tuple, Optional import gradio as gr from modules import script_callbacks, scripts, shared from modules.shared import opts # Для генерации from modules import processing, sd_samplers # Импортируем модуль парсера (второй файл), чтобы при смене ENV можно было reload(...) from modules import prompt_parser as pp # --------------------------------------------------------------------- # Утилиты: подсветка ошибок, советы # --------------------------------------------------------------------- def _highlight_at(text: str, line: int, column: int, context: int = 2) -> str: lines = text.splitlines() i = max(0, line - 1) start = max(0, i - context) end = min(len(lines), i + context + 1) out = [] for idx in range(start, end): prefix = ">> " if idx == i else " " out.append(f"{prefix}{idx+1:>4}: {lines[idx]}") if idx == i: caret = " " * (column + 7 + len(str(idx+1))) + "^" out.append(caret) return "\n".join(out) def _guess_suggestion(src: str) -> List[str]: tips = [] if src.count("(") != src.count(")"): tips.append("Скобки () несбалансированы — проверьте пары.") if src.count("[") != src.count("]"): tips.append("Скобки [] несбалансированы — проверьте пары.") if src.count("{") != src.count("}"): tips.append("Скобки {} несбалансированы — проверьте пары.") if "::" in src and not any(term in src for term in ["!", ";", "!!"]): tips.append("Похоже, вы используете sequence (::), но не поставили терминатор (! или ; / !! для top-level).") if re.search(r":[A-Za-z]", src): tips.append("После ':' обычно ожидается число для веса (например, :1.2) — проверьте места с двоеточием.") if re.search(r"\[[^]]+\]!\s*!", src): tips.append("alternate_distinct пишется как [a|b]! — одно '!' сразу после ']'.") return tips def format_lark_error(prompt: str, err: Exception) -> str: title = f"**Ошибка парсинга:** `{type(err).__name__}`: {getattr(err, 'message', str(err))}" try: line = getattr(err, 'line', None) column = getattr(err, 'column', None) pos = getattr(err, 'pos_in_stream', None) except Exception: line = column = pos = None block = "" if line is not None and column is not None: block = "```\n" + _highlight_at(prompt, line, column) + "\n```" elif pos is not None: prefix = prompt[:pos] l = prefix.count("\n") + 1 c = len(prefix.split("\n")[-1]) + 1 block = "```\n" + _highlight_at(prompt, l, c) + "\n```" tips = _guess_suggestion(prompt) tips_md = "\n".join([f"- {t}" for t in tips]) if tips else "- Проверьте синтаксис во вкладке «Справка и примеры»." return f"{title}\n\n{block}\n\n**Как исправить:**\n{tips_md}\n" # --------------------------------------------------------------------- # ENV и перегрузка парсера # --------------------------------------------------------------------- class EnvState: def __init__(self): self.allow_empty_alt = os.environ.get("ALLOW_EMPTY_ALTERNATE", "0") self.expand_alt_per_step = os.environ.get("EXPAND_ALTERNATE_PER_STEP", "1") self.group_combo_limit = os.environ.get("GROUP_COMBO_LIMIT", "100") def apply(self, allow_empty_alt: bool, expand_alt: bool, combo_limit: int) -> bool: changed = False new_allow = "1" if allow_empty_alt else "0" new_expand = "1" if expand_alt else "0" new_limit = str(int(combo_limit)) if new_allow != self.allow_empty_alt: os.environ["ALLOW_EMPTY_ALTERNATE"] = new_allow self.allow_empty_alt = new_allow changed = True if new_expand != self.expand_alt_per_step: os.environ["EXPAND_ALTERNATE_PER_STEP"] = new_expand self.expand_alt_per_step = new_expand changed = True if new_limit != self.group_combo_limit: os.environ["GROUP_COMBO_LIMIT"] = new_limit self.group_combo_limit = new_limit changed = True return changed ENV_STATE = EnvState() def maybe_reload_prompt_parser(allow_empty_alt: bool, expand_alt: bool, combo_limit: int): changed = ENV_STATE.apply(allow_empty_alt, expand_alt, combo_limit) if changed: global pp from modules import prompt_parser as _pp pp = importlib.reload(_pp) # --------------------------------------------------------------------- # Парс/расписания/превью # --------------------------------------------------------------------- def _safe_parse_tree(prompt: str): try: tree = pp.schedule_parser.parse(prompt) tree_str = tree.pretty() if hasattr(tree, "pretty") else str(tree) return tree_str, "" except Exception as e: return "", format_lark_error(prompt, e) def _make_schedule(prompt: str, steps: int, seed: Optional[int], use_visitor: bool) -> Tuple[List[List[Any]], str]: try: schedule = pp.get_schedule(prompt, steps, True, seed, use_visitor=use_visitor) if not schedule: return [], "**Пустое расписание.** Проверьте синтаксис." return schedule, "" except Exception as e: return [], f"**Ошибка при построении расписания:** {e}" def _schedule_table_md(schedule: List[List[Any]]) -> str: rows = ["| end_step | text |", "|---:|---|"] for end_step, text in schedule: safe_text = text.replace("|", "\\|") rows.append(f"| {end_step} | {safe_text} |") return "\n".join(rows) def _per_step_timeline_md(schedule: List[List[Any]], steps: int) -> str: rows = ["| step | active text |", "|---:|---|"] for s in range(1, steps + 1): txt = pp.at_step_from_schedule(s, schedule) safe_txt = txt.replace("|", "\\|") rows.append(f"| {s} | {safe_txt} |") return "\n".join(rows) def _compare_vis_vs_trans(prompt: str, steps: int, seed: Optional[int]) -> str: vis, e1 = _make_schedule(prompt, steps, seed, True) trn, e2 = _make_schedule(prompt, steps, seed, False) if e1 or e2: return (e1 or "") + (e2 or "") md = ["**Visitor (сбор в визиторе)**", _schedule_table_md(vis), "", "**Transformer (пересбор на end_step)**", _schedule_table_md(trn)] return "\n\n".join(md) # --------------------------------------------------------------------- # Линтер и автофиксы # --------------------------------------------------------------------- class Issue: def __init__(self, kind: str, msg: str, fix_hint: str = "", fixer=None): self.kind = kind # "Error"/"Warning"/"Hint" self.msg = msg self.fix_hint = fix_hint self.fixer = fixer # callable(src) -> src def lint_prompt(src: str, use_visitor: bool) -> List[Issue]: issues: List[Issue] = [] # Несбалансированные скобки if src.count("(") != src.count(")"): issues.append(Issue("Error", "Несбалансированы круглые скобки ().", "Проверьте пары () или удалите лишние.", lambda s: auto_balance(s, "(", ")"))) if src.count("[") != src.count("]"): issues.append(Issue("Error", "Несбалансированы квадратные скобки [].", "Проверьте пары [] или удалите лишние.", lambda s: auto_balance(s, "[", "]"))) if src.count("{") != src.count("}"): issues.append(Issue("Warning", "Несбалансированы фигурные скобки {}.", "Если это 'dead group', лучше убрать фигурные скобки.", lambda s: auto_balance(s, "{", "}"))) # Sequence без терминатора if "::" in src and not any(t in src for t in ["!", ";", "!!"]): issues.append(Issue("Error", "Обнаружен sequence (::), но нет завершающего терминатора (! или ; / !!).", "Добавьте '!' или ';' в конец sequence.", ensure_sequence_terminator)) # Колон после которого не число if re.search(r":[A-Za-z]", src): issues.append(Issue("Warning", "После ':' встречается буква — вероятно ожидалось число веса.", "Исправьте на ':1.2' или уберите ':'.", lambda s: s)) # 'a | b' вне квадратных скобок if re.search(r"(?:^|[^[])\b[^]\n]*\|[^[]", src) and "[" not in src: issues.append(Issue("Hint", "Используется '|' вне '[]' — это alternate1/2 (рандом), а не карусель по шагам.", "Заменить на [ a | b ] для перебора по шагам.", bars_to_square_alternate)) # '&' в plain if "&" in src: issues.append(Issue("Hint", "Символ '&' активирует and_rule (логическая сцепка).", "Если это текст, замените на 'and' или запятую.", replace_amp_with_and)) # Dead group: {a, b, c} без | if "{" in src and "|" not in src: issues.append(Issue("Hint", "Группа без '|' (dead group) — просто склейка, фигурные скобки не нужны.", "Уберите {} — оставьте плоский список.", remove_dead_groups)) # alternate_distinct ']]!!' if re.search(r"\[[^]]+\]!\s*!", src): issues.append(Issue("Warning", "Лишний '!' после alternate_distinct.", "Должно быть '[a|b]!' (одно '!').", lambda s: re.sub(r"\]!\s*!", "]!", s))) # numbered: N! {a|b} с недостатком опций for m in re.finditer(r"(\d+)\s*!?\s*{([^}]+)}", src): n = int(m.group(1)) body = m.group(2) opts = [o.strip() for o in body.split("|") if o.strip()] if "!" in m.group(0) and len(opts) < n: issues.append(Issue("Warning", f"В numbered distinct запрошено {n}, но вариантов всего {len(opts)}.", "Уменьшите N или уберите '!'.", lambda s, _n=n: re.sub(rf"\b{_n}\s*!", str(_n), s, count=1))) # alternate2 под use_visitor=True if use_visitor and re.search(r"(^|[^[])\b[^]\n]+\|[^[]", src): issues.append(Issue("Hint", "Под use_visitor=True бар 'a|b' может остаться буквальным. " "Для случайного выбора используйте [a|b] или выключите use_visitor.", "Заменить на [a|b] или переключить режим.", bars_to_square_alternate)) return issues def auto_balance(src: str, l: str, r: str) -> str: c_l = src.count(l) c_r = src.count(r) if c_l > c_r: return src + (r * (c_l - c_r)) if c_r > c_l: return src return src def ensure_sequence_terminator(src: str) -> str: if "::" in src and not re.search(r"(!|;|!!)\s*$", src): return src.rstrip() + " !" return src def bars_to_square_alternate(src: str) -> str: if "[" in src and "]" in src: return src return "[ " + src.strip() + " ]" def replace_amp_with_and(src: str) -> str: return src.replace("&", "and") def remove_dead_groups(src: str) -> str: def repl(m): inner = m.group(1) if "|" in inner: return m.group(0) return inner return re.sub(r"\{\s*([^{}]+?)\s*\}", repl, src) def apply_safe_fixes(src: str, use_visitor: bool) -> Tuple[str, List[str]]: issues = lint_prompt(src, use_visitor) changes = [] out = src for issue in issues: if callable(issue.fixer): new_out = issue.fixer(out) if new_out != out: changes.append(f"{issue.kind}: {issue.msg} → {issue.fix_hint}") out = new_out return out, changes # --------------------------------------------------------------------- # Конструкторы/Конвертеры/Стратегии # --------------------------------------------------------------------- def build_group(items_csv: str) -> str: items = [s.strip() for s in items_csv.split(",") if s.strip()] return "{ " + ", ".join(items) + " }" if items else "{}" def build_alternate(items_bar: str, distinct: bool) -> str: core = items_bar.strip() if not core: return "[]" return f"[ {core} ]!" if distinct else f"[ {core} ]" def build_numbered(n: int, options_bar_or_group: str, distinct: bool) -> str: n = max(1, int(n)) body = options_bar_or_group.strip() mark = "!" if distinct else "" return f"{n}{mark} {body}" def build_sequence(owner: str, pairs_text: str, terminator: str) -> str: owner = owner.strip() pairs = [] for chunk in pairs_text.split(";"): chunk = chunk.strip() if not chunk: continue if ":" not in chunk: pairs.append(chunk) else: k, v = chunk.split(":", 1) pairs.append(f"{k.strip()} :: {v.strip()}") term = "!" if terminator == "!" else ";" return f"{owner} :: " + ", ".join(pairs) + f" {term}" def build_top_level_sequence(owner: str, pairs_text: str, trailing_plain: str) -> str: seq = build_sequence(owner, pairs_text, "!") inner = seq.replace(f"{owner} :: ", "").rstrip("!;").strip() tail = (", " + trailing_plain.strip()) if trailing_plain.strip() else "" return f"{owner.strip()} ::: {inner} !!{tail}" def build_scheduled_block(variants_multiline: str, ranges_text: str, reverse: bool) -> str: variants = [v.strip() for v in variants_multiline.split("\n") if v.strip()] if not variants: return "" core = " : ".join(variants) extra = f" : {ranges_text.strip()}" if ranges_text.strip() else "" suffix = " reverse" if reverse else "" return f"[ {core} ]{extra}{suffix}" def build_alternate_every_k(options_bar: str, steps: int, k: int) -> str: opts = [o.strip() for o in options_bar.split("|") if o.strip()] if not opts: return "" boundaries = [] for i in range(1, len(opts)): b = i * k if b < steps: boundaries.append(str(b)) extra = f" : {', '.join(boundaries)}" if boundaries else "" return f"[ {' : '.join(opts)} ]{extra}" # Конвертеры def conv_dead_groups_to_plain(src: str) -> str: return remove_dead_groups(src) def conv_bars_to_square(src: str) -> str: return bars_to_square_alternate(src) def conv_add_seq_terminator(src: str) -> str: return ensure_sequence_terminator(src) # --------------------------------------------------------------------- # Экспорт и дифф # --------------------------------------------------------------------- def schedule_to_csv(schedule: List[List[Any]]) -> str: lines = ["end_step,text"] for end_step, text in schedule: safe = text.replace('"', '""') lines.append(f'{end_step},"{safe}"') return "\n".join(lines) def timeline_to_csv(schedule: List[List[Any]], steps: int) -> str: lines = ["step,text"] for s in range(1, steps + 1): t = pp.at_step_from_schedule(s, schedule) safe = t.replace('"', '""') lines.append(f'{s},"{safe}"') return "\n".join(lines) def schedule_to_json(schedule: List[List[Any]]) -> str: data = [{"end_step": end, "text": txt} for end, txt in schedule] return json.dumps(data, ensure_ascii=False, indent=2) def md_block(title: str, content: str) -> str: return f"### {title}\n\n```\n{content}\n```" def mk_diff(a: str, b: str, from_name="original", to_name="fixed") -> str: diff = difflib.unified_diff( a.splitlines(keepends=False), b.splitlines(keepends=False), fromfile=from_name, tofile=to_name, lineterm="" ) text = "\n".join(diff) return "```\n" + (text if text else "# (no changes)\n") + "\n```" # --------------------------------------------------------------------- # Генерация (интеграция с A1111) # --------------------------------------------------------------------- def _list_samplers() -> List[str]: # Возвращаем список доступных сэмплеров по именам try: return [x.name for x in sd_samplers.all_samplers] except Exception: return ["Euler a", "Euler", "DPM++ 2M Karras"] def _generate_txt2img(prompt: str, negative: str, steps: int, width: int, height: int, cfg_scale: float, seed: int, sampler_name: str, batch_count: int, batch_size: int, restore_faces: bool, tiling: bool) -> Tuple[List[Any], str]: """ Классический txt2img через A1111 API. Важное: второй prompt_parser автоматически работает внутри process_images. """ try: p = processing.StableDiffusionProcessingTxt2Img( sd_model=shared.sd_model, prompt=prompt, negative_prompt=negative, steps=int(steps), width=int(width), height=int(height), cfg_scale=float(cfg_scale), seed=int(seed), sampler_name=sampler_name, n_iter=int(batch_count), batch_size=int(batch_size), restore_faces=restore_faces, tiling=tiling, do_not_save_grid=True, do_not_save_samples=True, ) processed = processing.process_images(p) return processed.images, processed.info except Exception as e: return [], f"Ошибка генерации: {e}" def _generate_variants_with_placeholder(prompt: str, negative: str, placeholder: str, options_multiline: str, steps: int, width: int, height: int, cfg_scale: float, seed: int, sampler_name: str, restore_faces: bool, tiling: bool) -> Tuple[List[Any], str]: """ Генерация серии вариантов: плейсхолдер {{ALT}} меняется на каждую опцию (каждая опция с новой строки). batch_size=1, n_iter = len(options) """ opts_list = [o.strip() for o in options_multiline.split("\n") if o.strip()] if not opts_list: return [], "Нет вариантов (пустой список)." images_all = [] infos = [] try: for i, opt_val in enumerate(opts_list, 1): pr = prompt.replace(placeholder, opt_val) p = processing.StableDiffusionProcessingTxt2Img( sd_model=shared.sd_model, prompt=pr, negative_prompt=negative, steps=int(steps), width=int(width), height=int(height), cfg_scale=float(cfg_scale), seed=int(seed + i - 1), # разносим сиды по вариантам, чтобы тайлы отличались sampler_name=sampler_name, n_iter=1, batch_size=1, restore_faces=restore_faces, tiling=tiling, do_not_save_grid=True, do_not_save_samples=True, ) processed = processing.process_images(p) images_all.extend(processed.images or []) infos.append(f"[{i}] {opt_val} — seed={int(seed + i - 1)}") return images_all, "\n".join(infos) except Exception as e: return [], f"Ошибка генерации вариантов: {e}" # --------------------------------------------------------------------- # Основной анализ # --------------------------------------------------------------------- def analyze(prompt: str, steps: int, seed: Optional[int], use_visitor: bool, allow_empty_alt: bool, expand_alt_per_step: bool, group_combo_limit: int, view_mode: str, at_step_n: int) -> Tuple[str, str, str, str, str]: try: maybe_reload_prompt_parser(allow_empty_alt, expand_alt_per_step, group_combo_limit) except Exception as e: err = f"**Не удалось применить ENV-флаги:** {e}" return err, "", "", "", "" status_md = "" tree_md = "" schedule_md = "" timeline_md = "" text_at_step_md = "" tree_str, err_md = _safe_parse_tree(prompt) if err_md: status_md = err_md else: status_md = "**OK:** Парсинг прошёл успешно." if view_mode == "Parse tree": tree_md = "```\n" + tree_str + "\n```" schedule, sched_err = _make_schedule(prompt, steps, seed, use_visitor) if sched_err: status_md += "\n\n" + sched_err return status_md, "", "", "", tree_md if view_mode in ("Schedule", "All"): schedule_md = _schedule_table_md(schedule) if view_mode in ("Timeline (per step)", "All"): timeline_md = _per_step_timeline_md(schedule, steps) try: t = pp.at_step_from_schedule(at_step_n, schedule) text_at_step_md = f"**Шаг {at_step_n}:** `{t}`" except Exception as e: text_at_step_md = f"Ошибка получения текста на шаге {at_step_n}: {e}" if view_mode == "Visitor vs Transformer": tree_md = _compare_vis_vs_trans(prompt, steps, seed) return status_md, schedule_md, timeline_md, text_at_step_md, tree_md # --------------------------------------------------------------------- # Gradio UI # --------------------------------------------------------------------- class PromptParserAnalyzerScript(scripts.Script): def title(self): return "Prompt Parser Analyzer" def show(self, is_txt2img): return scripts.AlwaysVisible def ui(self, is_txt2img): with gr.Group(): with gr.Accordion("Prompt Parser Analyzer", open=False): with gr.Tabs(): # ---------------- Analyzer ---------------- with gr.Tab("Analyzer"): prompt_in = gr.Textbox(label="Source Prompt", lines=6, placeholder="Вставьте промпт…") with gr.Row(): steps = gr.Slider(1, 300, value=50, step=1, label="Steps") seed = gr.Number(value=42, precision=0, label="Seed (для случайных конструкций)") at_step_n = gr.Slider(1, 300, value=25, step=1, label="Показать текст на шаге N") with gr.Row(): use_visitor = gr.Checkbox(value=True, label="use_visitor (быстрее; alternate2 может отличаться)") view_mode = gr.Dropdown( choices=["All", "Schedule", "Timeline (per step)", "Parse tree", "Visitor vs Transformer"], value="All", label="Режим вывода", ) with gr.Row(): allow_empty_alt = gr.Checkbox(value=False, label="ALLOW_EMPTY_ALTERNATE (разрешать пустые [a||b])") expand_alt = gr.Checkbox(value=True, label="EXPAND_ALTERNATE_PER_STEP (чередовать по шагам)") group_limit = gr.Slider(1, 5000, value=100, step=1, label="GROUP_COMBO_LIMIT") analyze_btn = gr.Button("Analyze", variant="primary") status_md = gr.Markdown() schedule_md = gr.Markdown() timeline_md = gr.Markdown() text_at_step_md = gr.Markdown() tree_md = gr.Markdown() # ---------------- Lint/Fix ---------------- with gr.Tab("Lint / Fix"): lint_src = gr.Textbox(label="Prompt для проверки", lines=6, placeholder="Вставьте промпт…") with gr.Row(): lint_use_visitor = gr.Checkbox(value=True, label="use_visitor (для эвристик)") lint_btn = gr.Button("Run Lint", variant="primary") fix_btn = gr.Button("Apply Safe Fixes") lint_report = gr.Markdown() fixed_out = gr.Textbox(label="Fixed Prompt (предпросмотр)", lines=6) diff_md = gr.Markdown() # ---------------- Scheduler Builder ---------------- with gr.Tab("Scheduler"): gr.Markdown("Соберите `scheduled`: варианты по строкам, интервалы (%, шаги), опционально `reverse`.") sched_variants = gr.Textbox(label="Варианты (каждый с новой строки)", value="A\nB\nC", lines=6) sched_ranges = gr.Textbox(label="Интервалы", placeholder="30, 70 или 10%-30%, 60-80") sched_reverse = gr.Checkbox(value=False, label="reverse") build_sched_btn = gr.Button("Build [ ... ] : ranges", variant="primary") sched_expr = gr.Textbox(label="Результат", lines=2) gr.Markdown("---") gr.Markдин("Альтернативы: детерминированный переключатель каждые K шагов (строится через scheduled).") alt_bar = gr.Textbox(label="Опции (через |)", value="red|blue|green") k_steps = gr.Slider(1, 100, value=10, step=1, label="K (каждые K шагов)") total_steps_for_k = gr.Slider(1, 300, value=50, step=1, label="Steps (для расчёта границ)") build_every_k = gr.Button("Build alternate every K", variant="secondary") every_k_expr = gr.Textbox(label="Результат", lines=2) # ---------------- Converters ---------------- with gr.Tab("Converters"): conv_src = gr.Textbox(label="Prompt для преобразования", lines=6) with gr.Row(): conv_dead = gr.Button("Dead groups → plain") conv_bars = gr.Button("a|b → [a|b]") conv_seq_term = gr.Button("Sequence: добавить терминатор") conv_out = gr.Textbox(label="Результат", lines=6) gr.Markdown("---") gr.Markdown("Конструкторы Lark") with gr.Row(): grp_items = gr.Textbox(label="Group items (через запятую)", placeholder="a, b, c") grp_out = gr.Textbox(label="Результат", lines=2) grp_btn = gr.Button("Build {group}") with gr.Row(): alt_items = gr.Textbox(label="Alternates (через |)", placeholder="a|b|c") alt_distinct = gr.Checkbox(value=False, label="distinct (!)") alt_out = gr.Textbox(label="Результат", lines=2) alt_btn = gr.Button("Build [alternate]") with gr.Row(): num_n = gr.Number(value=3, precision=0, label="N (сколько взять)") num_body = gr.Textbox(label="Опции: '{...}' или 'a|b|c'", placeholder="{ a | b | c } или a|b|c") num_distinct = gr.Checkbox(value=True, label="distinct (!)") num_out = gr.Textbox(label="Результат", lines=2) num_btn = gr.Button("Build numbered") with gr.Row(): seq_owner = gr.Textbox(label="Sequence owner", value="character") seq_pairs = gr.Textbox(label="Sequence pairs (k:v; k:v)", value="outfit: leather jacket; mood: brooding") seq_term = gr.Dropdown(choices=["!", ";"], value="!", label="Terminator") seq_out = gr.Textbox(label="Результат", lines=2) seq_btn = gr.Button("Build sequence") with gr.Row(): top_owner = gr.Textbox(label="Top-level owner", value="portrait") top_pairs = gr.Textbox(label="Pairs (k:v; k:v)", value="hair: auburn; eyes: blue") top_tail = gr.Textbox(label="Trailing plain (опционально)", value="ultra-detailed skin texture") top_out = gr.Textbox(label="Результат", lines=2) top_btn = gr.Button("Build top-level sequence") # ---------------- Export & Diff ---------------- with gr.Tab("Export & Diff"): exp_src = gr.Textbox(label="Prompt (источник для экспорта)", lines=6) with gr.Row(): exp_steps = gr.Slider(1, 300, value=50, step=1, label="Steps") exp_seed = gr.Number(value=42, precision=0, label="Seed") exp_use_visitor = gr.Checkbox(value=True, label="use_visitor") exp_btn = gr.Button("Build schedule/timeline", variant="primary") exp_sched_csv = gr.Textbox(label="Schedule CSV", lines=6) exp_sched_json = gr.Textbox(label="Schedule JSON", lines=8) exp_timeline_csv = gr.Textbox(label="Timeline CSV", lines=8) copy_md_btn = gr.Button("Copy Markdown of schedule/timeline (ниже)") exp_sched_md = gr.Markdown() exp_timeline_md = gr.Markdown() # ---------------- Generate (txt2img) ---------------- with gr.Tab("Generate"): gr.Markdown("### Single") gen_prompt = gr.Textbox(label="Prompt", lines=6, placeholder="Возьмём из Analyzer → Source Prompt, можно отредактировать") gen_negative = gr.Textbox(label="Negative", lines=4, value="") with gr.Row(): gen_width = gr.Slider(64, 1536, value=512, step=64, label="Width") gen_height = gr.Slider(64, 1536, value=768, step=64, label="Height") gen_steps = gr.Slider(1, 300, value=30, step=1, label="Steps") with gr.Row(): gen_cfg = gr.Slider(1, 20, value=7.0, step=0.5, label="CFG") gen_seed = gr.Number(value=42, precision=0, label="Seed") gen_sampler = gr.Dropdown(choices=_list_samplers(), value=_list_samplers()[0], label="Sampler") with gr.Row(): gen_batch_count = gr.Slider(1, 16, value=1, step=1, label="Batch Count") gen_batch_size = gr.Slider(1, 8, value=1, step=1, label="Batch Size") with gr.Row(): gen_restore_faces = gr.Checkbox(value=False, label="Restore faces") gen_tiling = gr.Checkbox(value=False, label="Tiling") gen_btn = gr.Button("Generate", variant="primary") gen_gallery = gr.Gallery(label="Result").style(grid=4) gen_info = gr.Textbox(label="Info", lines=4) gr.Markdown("---") gr.Markdown("### A/B Variants (placeholder)") gr.Markdown("Задайте плейсхолдер в промпте, напр. `{{ALT}}`, и список вариантов построчно.") var_prompt = gr.Textbox(label="Prompt (с плейсхолдером)", lines=6) var_negative = gr.Textbox(label="Negative", lines=4, value="") var_placeholder = gr.Textbox(label="Placeholder", value="{{ALT}}") var_options = gr.Textbox(label="Options (каждая на новой строке)", value="red dress\nblue jeans\nblack coat", lines=6) with gr.Row(): var_width = gr.Slider(64, 1536, value=512, step=64, label="Width") var_height = gr.Slider(64, 1536, value=768, step=64, label="Height") var_steps = gr.Slider(1, 300, value=30, step=1, label="Steps") with gr.Row(): var_cfg = gr.Slider(1, 20, value=7.0, step=0.5, label="CFG") var_seed = gr.Number(value=1000, precision=0, label="Base Seed") var_sampler = gr.Dropdown(choices=_list_samplers(), value=_list_samplers()[0], label="Sampler") with gr.Row(): var_restore_faces = gr.Checkbox(value=False, label="Restore faces") var_tiling = gr.Checkbox(value=False, label="Tiling") var_btn = gr.Button("Generate A/B", variant="secondary") var_gallery = gr.Gallery(label="Variants").style(grid=6) var_info = gr.Textbox(label="Info", lines=6) # ------- Bindings ------- analyze_btn.click( fn=analyze, inputs=[prompt_in, steps, seed, use_visitor, allow_empty_alt, expand_alt, group_limit, view_mode, at_step_n], outputs=[status_md, schedule_md, timeline_md, text_at_step_md, tree_md], ) # Lint/Fix def _run_lint(src: str, use_vis: bool): issues = lint_prompt(src, use_vis) if not issues: return "**OK:** проблем не найдено.", "" lines = [] for it in issues: lines.append(f"- **{it.kind}**: {it.msg} — _{it.fix_hint or '…'}_") return "\n".join(lines), "" lint_btn.click(_run_lint, [lint_src, lint_use_visitor], [lint_report, fixed_out]) def _apply_fixes(src: str, use_vis: bool): fixed, changes = apply_safe_fixes(src, use_vis) report = "**Применены правки:**\n" + "\n".join([f"- {c}" for c in changes]) if changes else "_Нечего править (safe-fixes)._" return fixed, report, mk_diff(src, fixed) fix_btn.click(_apply_fixes, [lint_src, lint_use_visitor], [fixed_out, lint_report, diff_md]) # Scheduler build_sched_btn.click(lambda v, r, rev: build_scheduled_block(v, r, rev), [sched_variants, sched_ranges, sched_reverse], [sched_expr]) build_every_k.click(lambda bar, s, k: build_alternate_every_k(bar, int(s), int(k)), [alt_bar, total_steps_for_k, k_steps], [every_k_expr]) # Converters conv_dead.click(lambda s: conv_dead_groups_to_plain(s), [conv_src], [conv_out]) conv_bars.click(lambda s: conv_bars_to_square(s), [conv_src], [conv_out]) conv_seq_term.click(lambda s: conv_add_seq_terminator(s), [conv_src], [conv_out]) # Builders grp_btn.click(build_group, [grp_items], [grp_out]) alt_btn.click(build_alternate, [alt_items, alt_distinct], [alt_out]) num_btn.click(build_numbered, [num_n, num_body, num_distinct], [num_out]) seq_btn.click(build_sequence, [seq_owner, seq_pairs, seq_term], [seq_out]) top_btn.click(build_top_level_sequence, [top_owner, top_pairs, top_tail], [top_out]) # Export def _export_all(src: str, s: int, sd: Optional[int], use_vis: bool): sched, err = _make_schedule(src, s, sd, use_vis) if err: return "", "", "", f"**Ошибка:** {err}", "" csv_sched = schedule_to_csv(sched) json_sched = schedule_to_json(sched) csv_timeline = timeline_to_csv(sched, s) md1 = md_block("Schedule", _schedule_table_md(sched)) md2 = md_block("Timeline", _per_step_timeline_md(sched, s)) return csv_sched, json_sched, csv_timeline, md1, md2 exp_btn.click(_export_all, [exp_src, exp_steps, exp_seed, exp_use_visitor], [exp_sched_csv, exp_sched_json, exp_timeline_csv, exp_sched_md, exp_timeline_md]) def _copy_md(md1: str, md2: str): return md1 + "\n\n" + md2 copy_md_btn.click(_copy_md, [exp_sched_md, exp_timeline_md], [exp_sched_md]) # Generate — Single gen_btn.click( fn=_generate_txt2img, inputs=[gen_prompt, gen_negative, gen_steps, gen_width, gen_height, gen_cfg, gen_seed, gen_sampler, gen_batch_count, gen_batch_size, gen_restore_faces, gen_tiling], outputs=[gen_gallery, gen_info] ) # Generate — A/B Variants var_btn.click( fn=_generate_variants_with_placeholder, inputs=[var_prompt, var_negative, var_placeholder, var_options, var_steps, var_width, var_height, var_cfg, var_seed, var_sampler, var_restore_faces, var_tiling], outputs=[var_gallery, var_info] ) return [] # не нужно возвращать компоненты для A1111 core def on_ui_settings(): section = ('prompt-parser-analyzer', "Prompt Parser Analyzer") # Можно добавить глобальные Settings при желании. script_callbacks.on_ui_settings(on_ui_settings)