gstaff commited on
Commit
1e131a5
1 Parent(s): cdc7658

Gracefully handle card render fail.

Browse files
Files changed (1) hide show
  1. app.py +12 -9
app.py CHANGED
@@ -214,15 +214,18 @@ def html_to_png(card_name, html):
214
  css = ['./colab-data-test/css/mana.css', './colab-data-test/css/keyrune.css', './colab-data-test/css/mtg_custom.css']
215
  imgkit.from_string(html, path, {"xvfb": ""}, css=css)
216
  except:
217
- # For Windows local, requires 'html2image' package from pip.
218
- from html2image import Html2Image
219
- rendered_card_dir = 'rendered_cards'
220
- hti = Html2Image(output_path=rendered_card_dir)
221
- paths = hti.screenshot(html_str=html,
222
- css_file=['./colab-data-test/css/mtg_custom.css', './colab-data-test/css/mana.css', './colab-data-test/css/keyrune.css'],
223
- save_as=save_name, size=(450, 600))
224
- print(paths)
225
- path = paths[0]
 
 
 
226
  print('OPENING IMAGE FROM FILE')
227
  img = Image.open(path)
228
  print('CROPPING BACKGROUND')
 
214
  css = ['./colab-data-test/css/mana.css', './colab-data-test/css/keyrune.css', './colab-data-test/css/mtg_custom.css']
215
  imgkit.from_string(html, path, {"xvfb": ""}, css=css)
216
  except:
217
+ try:
218
+ # For Windows local, requires 'html2image' package from pip.
219
+ from html2image import Html2Image
220
+ rendered_card_dir = 'rendered_cards'
221
+ hti = Html2Image(output_path=rendered_card_dir)
222
+ paths = hti.screenshot(html_str=html,
223
+ css_file=['./colab-data-test/css/mtg_custom.css', './colab-data-test/css/mana.css', './colab-data-test/css/keyrune.css'],
224
+ save_as=save_name, size=(450, 600))
225
+ print(paths)
226
+ path = paths[0]
227
+ except:
228
+ pass
229
  print('OPENING IMAGE FROM FILE')
230
  img = Image.open(path)
231
  print('CROPPING BACKGROUND')