svg2png / app.py
codelion's picture
Update app.py
1b43408 verified
raw
history blame
1.24 kB
import cairosvg
import gradio as gr
from PIL import Image
import os
from io import BytesIO
def show(svg_file):
# Check if svg_file is None (no file uploaded)
if svg_file is None:
return None, None
# Get the file content from the Gradio file object
# svg_file is a tempfile._TemporaryFileWrapper or similar object
with open(svg_file.name, 'rb') as f:
svg_content = f.read()
# Convert SVG content to PNG
output_path = "./test.png"
cairosvg.svg2png(bytestring=svg_content, write_to=output_path)
# Open and return the PNG image
png_image = Image.open(output_path)
return png_image, output_path
with gr.Blocks() as bl:
gr.Markdown("# SVG to PNG Converter")
with gr.Row():
with gr.Column():
svg_input = gr.File(label="Upload SVG File", file_types=[".svg"])
convert_btn = gr.Button("Convert")
with gr.Column():
png_output = gr.Image(type='pil', label="Converted PNG")
download_btn = gr.File(label="Download PNG")
# Connect the conversion function
convert_btn.click(
fn=show,
inputs=svg_input,
outputs=[png_output, download_btn]
)
bl.launch()