| """Statblock generator: input handling, generation, rendering, export.""" |
|
|
| import html |
| import random |
| import tempfile |
| from pathlib import Path |
|
|
| from .shared import ai_client |
| from .shared.ratelimit import RateLimiter |
| from .prompts import SCHEMA_SPEC, SYSTEM_PROMPT, build_user_prompt |
| from .fixtures import FIXTURE_RUSTBOUND_SENTINEL |
|
|
| limiter = RateLimiter() |
|
|
| CHALLENGE_RATINGS = ["1/8", "1/4", "1/2"] + [str(n) for n in range(1, 21)] |
| SIZES = ["Tiny", "Small", "Medium", "Large", "Huge", "Gargantuan"] |
| CREATURE_TYPES = [ |
| "Aberration", "Beast", "Celestial", "Construct", "Dragon", "Elemental", |
| "Fey", "Fiend", "Giant", "Humanoid", "Monstrosity", "Ooze", "Plant", "Undead", |
| ] |
|
|
| MAX_CONCEPT = 2000 |
| MAX_FIELD = 300 |
| MAX_EXTRA = 800 |
|
|
|
|
| def randomize(): |
| """Fill the framing dropdowns randomly; the concept stays the user's.""" |
| return ( |
| random.choice(CHALLENGE_RATINGS), |
| random.choice(SIZES), |
| random.choice(CREATURE_TYPES), |
| ) |
|
|
|
|
| def _clip(value: str, limit: int) -> str: |
| return (value or "").strip()[:limit] |
|
|
|
|
| def generate(concept, cr, size, creature_type, legendary, extra, request=None): |
| """Returns (markdown, plain_text, download_path, error_message).""" |
| allowed, message = limiter.check(request) |
| if not allowed: |
| return None, None, None, message |
|
|
| fields = { |
| "Creature concept / homebrew notes": _clip(concept, MAX_CONCEPT), |
| "Target challenge rating": _clip(cr, 20), |
| "Size": _clip(size, 40), |
| "Creature type": _clip(creature_type, MAX_FIELD), |
| "Include legendary actions": "yes" if legendary else "no", |
| "Extra mechanical requests": _clip(extra, MAX_EXTRA), |
| } |
| user_prompt = build_user_prompt(fields) |
|
|
| try: |
| data = ai_client.generate_json( |
| SYSTEM_PROMPT, user_prompt, SCHEMA_SPEC, |
| fixture=FIXTURE_RUSTBOUND_SENTINEL, |
| ) |
| except ai_client.AIUnavailable as exc: |
| return None, None, None, str(exc) |
|
|
| md = render_markdown(data) |
| path = Path(tempfile.mkdtemp()) / "dnd-statblock.md" |
| path.write_text(md, encoding="utf-8") |
| return md, md, str(path), None |
|
|
|
|
| def _has_content(value) -> bool: |
| """True unless the field is empty or the literal 'None' sentinel.""" |
| return (value or "").strip().rstrip(".").lower() not in ("", "none") |
|
|
|
|
| def render_markdown(d: dict) -> str: |
| traits = "\n".join(f"- {t}" for t in d["traits"]) |
| actions = "\n".join(f"- {a}" for a in d["actions"]) |
| md = f"""## {d['name']} |
| |
| *{d['size_type_alignment']}* |
| |
| **Armor Class** {d['armor_class']} \n**Hit Points** {d['hit_points']} \n**Speed** {d['speed']} \n**Ability Scores** {d['ability_scores']} \n**Saves & Skills** {d['saves_and_skills']} \n**Defenses** {d['defenses']} \n**Senses & Languages** {d['senses_and_languages']} \n**Challenge** {d['challenge']} |
| |
| ### Traits |
| {traits} |
| |
| ### Actions |
| {actions} |
| """ |
| if _has_content(d.get("bonus_and_reactions")): |
| md += f""" |
| ### Bonus Actions & Reactions |
| {d['bonus_and_reactions']} |
| """ |
| if _has_content(d.get("legendary_actions")): |
| md += f""" |
| ### Legendary Actions |
| {d['legendary_actions']} |
| """ |
| return md |
|
|
|
|
| def render_error(message: str) -> str: |
| return ( |
| '<div class="lf-output" role="alert" style="border-color:#A7343B;">' |
| f"<p>{html.escape(message)}</p></div>" |
| ) |
|
|
