app.py

import cairosvg
import gradio as gr
from PIL import Image
import os
import xml.etree.ElementTree as ET

def initial_render(svg_file):
    """Render SVG at a large size for preview and cropping."""
    if svg_file is None:
        return None, None
    
    # Read the SVG content
    with open(svg_file.name, 'rb') as f:
        svg_content = f.read()
    
    # Store SVG content in a temporary file for later use
    temp_svg_path = "./temp.svg"
    with open(temp_svg_path, 'wb') as f:
        f.write(svg_content)
    
    # Initial large render
    output_path = "./initial_preview.png"
    cairosvg.svg2png(
        bytestring=svg_content,
        write_to=output_path,
        output_width=3000,  # Large initial size to capture all content
        output_height=3000
    )
    
    # Load the preview image
    preview_image = Image.open(output_path)
    return preview_image, temp_svg_path

def final_convert(svg_path, crop_box, size_input):
    """Convert SVG to PNG using either crop box or user-specified dimensions."""
    if svg_path is None:
        return None, None
    
    # Read the stored SVG content
    with open(svg_path, 'rb') as f:
        svg_content = f.read()
    
    # Final output path
    output_path = "./final_output.png"
    
    # Determine output dimensions
    if size_input and size_input.strip():  # Check if size input is provided
        try:
            width, height = map(int, size_input.split('x'))
            if width <= 0 or height <= 0:
                raise ValueError("Width and height must be positive")
        except ValueError:
            return gr.Warning("Invalid size format. Use 'width x height' (e.g., '800x600')"), None
    elif crop_box and 'left' in crop_box:  # Use crop box if size input is empty
        left = crop_box['left']
        top = crop_box['top']
        width = crop_box['right'] - left
        height = crop_box['bottom'] - top
    else:
        # Fallback if neither is provided (shouldn't happen with UI flow)
        width, height = 2000, 2000
    
    # Convert SVG to PNG with determined dimensions
    cairosvg.svg2png(
        bytestring=svg_content,
        write_to=output_path,
        output_width=width,
        output_height=height
    )
    
    # Load the final PNG
    final_image = Image.open(output_path)
    return final_image, output_path

with gr.Blocks() as bl:
    gr.Markdown("# SVG to PNG Converter with Custom Crop or Size")
    
    with gr.Row():
        with gr.Column():
            svg_input = gr.File(label="Upload SVG File", file_types=[".svg"])
            preview_btn = gr.Button("Generate Preview")
            size_input = gr.Textbox(
                label="Output Size (width x height)",
                placeholder="e.g., 800x600 (optional, overrides crop)"
            )
        
        with gr.Column():
            preview_output = gr.Image(
                type='pil',
                label="Preview (Draw a box to crop if no size specified)",
                interactive=True,
                tool="sketch",
                height=800
            )
            crop_btn = gr.Button("Convert")
            final_output = gr.Image(type='pil', label="Final PNG", height=800)
            download_btn = gr.File(label="Download Final PNG")
    
    # State to store the temporary SVG path
    svg_state = gr.State()
    
    # Step 1: Generate initial preview
    preview_btn.click(
        fn=initial_render,
        inputs=svg_input,
        outputs=[preview_output, svg_state]
    )
    
    # Step 2: Convert using crop box or specified size
    crop_btn.click(
        fn=final_convert,
        inputs=[svg_state, preview_output, size_input],
        outputs=[final_output, download_btn]
    )

bl.launch()