Spaces:
Sleeping
Sleeping
File size: 3,094 Bytes
ecd0d00 5b034d2 09de40b 6154876 521e8e6 6154876 521e8e6 6154876 09de40b 6154876 09de40b ecd0d00 6154876 521e8e6 6154876 09de40b 6154876 ecd0d00 6154876 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 |
import gradio as gr
import imgkit
from PIL import Image
from io import BytesIO
def html_to_image(html_code, width, height):
options = {
'format': 'png',
'width': str(width),
'height': str(height),
'encoding': "UTF-8"
}
image = Image.open(BytesIO(imgkit.from_string(html_code, False, options=options)))
return image
def website_to_image(website_link, width, height):
if not website_link:
website_link = "https://www.google.com"
options = {
'format': 'png',
'width': str(width),
'height': str(height),
'encoding': "UTF-8"
}
image = Image.open(BytesIO(imgkit.from_url(website_link, False, options=options)))
return image
with gr.Blocks() as interface:
gr.Markdown("# HTML to Image Converter")
gr.Markdown("Enter HTML code and set dimensions to generate an image")
with gr.Row():
with gr.Column():
html_code_input = gr.Code(
label="HTML Code",
language="html",
lines=30,
interactive=True
)
width_input = gr.Number(
label="Width",
value=1280,
step=10
)
height_input = gr.Number(
label="Height",
value=720,
step=10,
)
with gr.Row(variant="panel", ):
with gr.Column():
website_link = gr.Textbox(
label="Website Link",
placeholder="https://www.google.com"
)
snapshotwebsite = gr.Button("Show Website", size="lg")
output_image = gr.Image(type="pil", format="PNG", show_label=False)
generate_button = gr.Button("Refresh")
html_code_input.change(
fn=html_to_image,
inputs=[html_code_input, width_input, height_input],
outputs=output_image
)
width_input.change(
fn=html_to_image,
inputs=[html_code_input, width_input, height_input],
outputs=output_image
)
height_input.change(
fn=html_to_image,
inputs=[html_code_input, width_input, height_input],
outputs=output_image
)
generate_button.click(
fn=html_to_image,
inputs=[html_code_input, width_input, height_input],
outputs=output_image
)
snapshotwebsite.click(
fn=website_to_image,
inputs=[website_link, width_input, height_input],
outputs=output_image
)
gr.Examples(
examples=[
["<div style='background: linear-gradient(45deg, #ff6b6b, #4ecdc4); padding: 20px; border-radius: 10px;'><h1 style='color: white; font-family: Arial;'>Hello, World!</h1></div>", 800, 400],
["<div style='background: #f0f0f0; padding: 20px;'><ul style='color: #333;'><li>Item 1</li><li>Item 2</li><li>Item 3</li></ul></div>", 600, 300]
],
inputs=[html_code_input, width_input, height_input]
)
if __name__ == "__main__":
interface.launch()
|