Remove imgkit dependency, update example commands, update LCM client URL.

app.py

@@ -1,11 +1,9 @@
-# pip install imgkit
 # pip install html2image
 import base64
 import random
-import uuid
 from io import BytesIO
 
-import imgkit
+from html2image import Html2Image
 import os
 import pathlib
 import re
@@ -16,8 +14,6 @@ from gradio_client import Client
 import torch
 from transformers import AutoModelForSpeechSeq2Seq, AutoProcessor, pipeline, Pipeline
 
-
-
 HF_TOKEN = os.getenv("HF_TOKEN")
 
 if not HF_TOKEN:
@@ -26,7 +22,7 @@ if not HF_TOKEN:
 API_URL = "https://api-inference.huggingface.co/models/HuggingFaceH4/zephyr-7b-beta"
 headers = {"Authorization": f"Bearer {HF_TOKEN}"}
 
-client = Client("https://latent-consistency-super-fast-lcm-lora-sd1-5.hf.space/")
+client = Client("https://latent-consistency-super-fast-lcm-lora-sd1-5.hf.space/--replicas/0867lltlv/")
 
 
 def init_speech_to_text_model() -> Pipeline:
@@ -334,22 +330,13 @@ def html_to_png(card_name, html):
 
     path = os.path.join('rendered_cards', save_name)
     try:
-        css = ['./css/mana.css', './css/keyrune.css',
-               './css/mtg_custom.css']
-        imgkit.from_string(html, path, {"xvfb": "", "enable-local-file-access": ""}, css=css)
-    except Exception as e:
-        try:
-            # For Windows local, requires 'html2image' package from pip.
-            from html2image import Html2Image
-            rendered_card_dir = 'rendered_cards'
-            hti = Html2Image(output_path=rendered_card_dir)
-            paths = hti.screenshot(html_str=html,
-                                   css_file='monstermaker.css',
-                                   save_as=save_name, size=(800, 1440))
-            print(paths)
-            path = paths[0]
-        except:
-            pass
+        rendered_card_dir = 'rendered_cards'
+        hti = Html2Image(output_path=rendered_card_dir)
+        paths = hti.screenshot(html_str=html, css_file='monstermaker.css', save_as=save_name, size=(800, 1440))
+        print(paths)
+        path = paths[0]
+    except:
+        pass
     print('OPENING IMAGE FROM FILE')
     img = Image.open(path).convert("RGB")
     print('CROPPING BACKGROUND')
@@ -448,7 +435,7 @@ with gr.Blocks(title='MonsterGen') as demo:
                 audio_in = gr.Microphone(label="Record a voice request (click or press ctrl + ` to start/stop)",
                                          type='filepath', elem_classes=["record-btn"])
                 prompt_in = gr.Textbox(label="Or type a text request and press Enter", interactive=True,
-                                       placeholder="Need an idea? Try one of these:\n- Create a creature card named 'WiFi Elemental'\n- Make it an instant\n- Change the color")
+                                       placeholder="Need an idea? Try one of these:\n- Create a monster named 'WiFi Elemental'\n- Make it legendary\n- Change the description")
             with gr.Accordion(label='🤖 Chat Assistant Response', open=False):
                 bot_text = gr.TextArea(label='Response', interactive=False)
     with gr.Row():

requirements.txt

@@ -4,4 +4,3 @@ ftfy
 gradio-client
 torch
 transformers
-imgkit