Improve card formatting, render html into png for return

app.py

@@ -1,6 +1,7 @@
 import base64
 import pathlib
 import re
+import time
 from io import BytesIO
 
 import gradio as gr
@@ -10,6 +11,7 @@ from PIL import Image
 from fastai.callback.core import Callback
 from fastai.learner import *
 from fastai.torch_core import TitledStr
+from html2image import Html2Image
 from min_dalle import MinDalle
 from torch import tensor, Tensor, float16, float32
 from torch.distributions import Transform
@@ -113,6 +115,13 @@ def extract_text_for_header(text, header):
     return match.group(1)
 
 
+def remove_section(html, html_class):
+    match = re.search(f'<li class="{html_class}"([\w\W])*?li>', html)
+    if match is not None:
+        html = html.replace(match.group(0), '')
+        return html
+
+
 def format_monster_card(monster_text, image_data):
     print('FORMATTING MONSTER TEXT')
     # see giffyglyph's monster maker https://giffyglyph.com/monstermaker/app/
@@ -146,14 +155,40 @@ def format_monster_card(monster_text, image_data):
     card = card.replace('{cha_stat}', cha_stat)
     saving_throws = extract_text_for_header(monster_text, 'Saving Throws')
     card = card.replace('{saving_throws}', saving_throws)
+    if not saving_throws:
+        card = remove_section(card, 'monster-saves')
+    skills = extract_text_for_header(monster_text, 'Skills')
+    card = card.replace('{skills}', skills)
+    if not skills:
+        card = remove_section(card, 'monster-skills')
+    damage_vulnerabilities = extract_text_for_header(monster_text, 'Damage Vulnerabilities')
+    card = card.replace('{damage_vulnerabilities}', damage_vulnerabilities)
+    if not damage_vulnerabilities:
+        card = remove_section(card, 'monster-vulnerabilities')
+    damage_resistances = extract_text_for_header(monster_text, 'Damage Resistances')
+    card = card.replace('{damage_resistances}', damage_resistances)
+    if not damage_resistances:
+        card = remove_section(card, 'monster-resistances')
+    damage_immunities = extract_text_for_header(monster_text, 'Damage Immunities')
+    card = card.replace('{damage_immunities}', damage_immunities)
+    if not damage_immunities:
+        card = remove_section(card, 'monster-immunities')
+    condition_immunities = extract_text_for_header(monster_text, 'Condition Immunities')
+    card = card.replace('{condition_immunities}', condition_immunities)
+    if not condition_immunities:
+        card = remove_section(card, 'monster-conditions')
     senses = extract_text_for_header(monster_text, 'Senses')
     card = card.replace('{senses}', senses)
+    if not senses:
+        card = remove_section(card, 'monster-senses')
     languages = extract_text_for_header(monster_text, 'Languages')
     card = card.replace('{languages}', languages)
+    if not languages:
+        card = remove_section(card, 'monster-languages')
     challenge = extract_text_for_header(monster_text, 'Challenge')
     card = card.replace('{challenge}', challenge)
-    # TODO: Skills, immunities etc., remove unavailable
-    # include Melee or Ranged Weapon Attack:
+    if not challenge:
+        card = remove_section(card, 'monster-challenge')
 
     description = extract_text_for_header(monster_text, 'Description')
     card = card.replace('{description}', description)
@@ -180,22 +215,30 @@ def format_monster_card(monster_text, image_data):
     else:
         card = card.replace('{passives}', f'<div class="monster-trait"><p>{passives}</p></div>')
 
-    # TODO: multiple lines
-    match = re.search(r"Actions:\n(.*)", monster_text)
+    match = re.search(r"Actions:\n([\w\W]*)", monster_text)
     if match is None:
         actions = ''
     else:
         actions = match.group(1)
     a = actions.split(':')
     if len(a) > 1:
-        action = a[0]
-        detail = ":".join(a[1:])
-        card = card.replace('{actions}',
-                            f'<div class="monster-action"><p><span class="name">{action}</span> <span class="detail">{detail}</span></p></div>')
+        a = ":".join(a)
+        a = a.split('\n')
+        actions_data = ''
+        for x in a:
+            x = x.split(':')
+            if len(x) > 1:
+                action = x[0]
+                if 'Passive' in action:
+                    break
+                detail = ":".join(x[1:])
+                actions_data += f'<div class="monster-action"><p><span class="name">{action}</span> <span class="detail">{detail}</span></p></div>'
+        card = card.replace('{actions}', actions_data)
     else:
         card = card.replace('{actions}', f'<div class="monster-action"><p>{actions}</p></div>')
 
     card = card.replace('Melee Weapon Attack:', '<i>Melee Weapon Attack:</i>')
+    card = card.replace('Ranged Weapon Attack:', '<i>Ranged Weapon Attack:</i>')
     card = card.replace('Hit:', '<i>Hit:</i>')
 
     print('FORMATTING MONSTER TEXT COMPLETE')
@@ -211,14 +254,31 @@ def pil_to_base64(image):
     return img_str
 
 
+hti = Html2Image(output_path='rendered_cards')
+
+
+def html_to_png(html):
+    print('CONVERTING HTML CARD TO PNG IMAGE')
+    paths = hti.screenshot(html_str=html, css_file="monstermaker.css", save_as="test.png")
+    path = paths[0]
+    img = Image.open(path).convert("RGB")
+    print('CONVERTING HTML CARD TO PNG IMAGE COMPLETE')
+    return img
+
+
 def run(name):
+    start = time.time()
+    print(f'BEGINNING RUN FOR {name}')
     text = gen_monster_text(name)
     description = parse_monster_description(text)
     pil = gen_image(description)
     image_data = pil_to_base64(pil)
-    card = format_monster_card(text, image_data)
-    return text, pil, card
+    card_html = format_monster_card(text, image_data)
+    card_image = html_to_png(card_html)
+    end = time.time()
+    print(f'RUN COMPLETED IN {end - start} seconds')
+    return text, pil, card_image, card_html
 
 
-iface = gr.Interface(fn=run, inputs="text", outputs=["text", "pil", 'html'])
+iface = gr.Interface(fn=run, inputs="text", outputs=["text", "pil", "pil", 'html'])
 iface.launch()

monsterMakerTemplate.html

@@ -90,6 +90,36 @@
                         <span>{saving_throws}</span>
                     </p>
                 </li>
+                <li class="monster-skills">
+                    <p>
+                        <span class="label">Skills</span>
+                        <span>{skills}</span>
+                    </p>
+                </li>
+                <li class="monster-vulnerabilities">
+                    <p>
+                        <span class="label">Damage Vulnerabilities</span>
+                        <span>{damage_vulnerabilities}</span>
+                    </p>
+                </li>
+                <li class="monster-resistances">
+                    <p>
+                        <span class="label">Damage Resistances</span>
+                        <span>{damage_resistances}</span>
+                    </p>
+                </li>
+                <li class="monster-immunities">
+                    <p>
+                        <span class="label">Damage Immunities</span>
+                        <span>{damage_immunities}</span>
+                    </p>
+                </li>
+                <li class="monster-conditions">
+                    <p>
+                        <span class="label">Condition Immunities</span>
+                        <span>{condition_immunities}</span>
+                    </p>
+                </li>
                 <li class="monster-senses">
                     <p>
                         <span class="label">Senses</span>