gstaff commited on
Commit
4c37e0d
1 Parent(s): 947e208

Remove imgkit dependency.

Browse files
Files changed (2) hide show
  1. app.py +11 -20
  2. requirements.txt +0 -1
app.py CHANGED
@@ -1,11 +1,9 @@
1
- # pip install imgkit
2
  # pip install html2image
3
  import base64
4
  import random
5
- import uuid
6
  from io import BytesIO
7
 
8
- import imgkit
9
  import os
10
  import pathlib
11
  import re
@@ -265,23 +263,16 @@ def html_to_png(card_name, html):
265
 
266
  path = os.path.join('rendered_cards', save_name)
267
  try:
268
- css = ['./css/mana.css', './css/keyrune.css',
269
- './css/mtg_custom.css']
270
- imgkit.from_string(html, path, {"xvfb": "", "enable-local-file-access": ""}, css=css)
271
- except Exception as e:
272
- try:
273
- # For Windows local, requires 'html2image' package from pip.
274
- from html2image import Html2Image
275
- rendered_card_dir = 'rendered_cards'
276
- hti = Html2Image(output_path=rendered_card_dir)
277
- paths = hti.screenshot(html_str=html,
278
- css_file=['./css/mtg_custom.css', './css/mana.css',
279
- './css/keyrune.css'],
280
- save_as=save_name, size=(450, 600))
281
- print(paths)
282
- path = paths[0]
283
- except:
284
- pass
285
  print('OPENING IMAGE FROM FILE')
286
  img = Image.open(path)
287
  print('CROPPING BACKGROUND')
 
 
1
  # pip install html2image
2
  import base64
3
  import random
 
4
  from io import BytesIO
5
 
6
+ from html2image import Html2Image
7
  import os
8
  import pathlib
9
  import re
 
263
 
264
  path = os.path.join('rendered_cards', save_name)
265
  try:
266
+ rendered_card_dir = 'rendered_cards'
267
+ hti = Html2Image(output_path=rendered_card_dir)
268
+ paths = hti.screenshot(html_str=html,
269
+ css_file=['./css/mtg_custom.css', './css/mana.css',
270
+ './css/keyrune.css'],
271
+ save_as=save_name, size=(450, 600))
272
+ print(paths)
273
+ path = paths[0]
274
+ except:
275
+ pass
 
 
 
 
 
 
 
276
  print('OPENING IMAGE FROM FILE')
277
  img = Image.open(path)
278
  print('CROPPING BACKGROUND')
requirements.txt CHANGED
@@ -4,4 +4,3 @@ ftfy
4
  gradio-client
5
  torch
6
  transformers
7
- imgkit
 
4
  gradio-client
5
  torch
6
  transformers