Automatically crop card png output

app.py

@@ -7,7 +7,7 @@ from io import BytesIO
 import gradio as gr
 import numpy as np
 import torch
-from PIL import Image
+from PIL import Image, ImageChops, ImageDraw
 from fastai.callback.core import Callback
 from fastai.learner import *
 from fastai.torch_core import TitledStr
@@ -238,6 +238,8 @@ def format_monster_card(monster_text, image_data):
     else:
         card = card.replace('{actions}', f'<div class="monster-action"><p>{actions}</p></div>')
 
+    # TODO: Legendary actions, reactions
+
     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('Melee or Ranged Weapon Attack:', '<i>Melee or Ranged Weapon Attack:</i>')
@@ -259,12 +261,29 @@ def pil_to_base64(image):
 hti = Html2Image(output_path='rendered_cards')
 
 
+def trim(im, border):
+    bg = Image.new(im.mode, im.size, border)
+    diff = ImageChops.difference(im, bg)
+    bbox = diff.getbbox()
+    if bbox:
+        return im.crop(bbox)
+
+
+def crop_background(image):
+    white = (255, 255, 255)
+    ImageDraw.floodfill(image, (image.size[0] - 1, 0), white, thresh=50)
+    image = trim(image, white)
+    return image
+
+
 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", size=(400, 1200))
+    paths = hti.screenshot(html_str=html, css_file="monstermaker.css", save_as="test.png", size=(400, 1440))
     path = paths[0]
     print('OPENING IMAGE FROM FILE')
     img = Image.open(path).convert("RGB")
+    print('CROPPING BACKGROUND')
+    img = crop_background(img)
     print('CONVERTING HTML CARD TO PNG IMAGE COMPLETE')
     return img