Update app.py

app.py

@@ -4,8 +4,8 @@ from PIL import Image
 import os
 import xml.etree.ElementTree as ET
 
-def show(svg_file):
-    # Check if no file is uploaded
+def initial_render(svg_file):
+    """Render SVG at a large size for preview and cropping."""
     if svg_file is None:
         return None, None
     
@@ -13,62 +13,91 @@ def show(svg_file):
     with open(svg_file.name, 'rb') as f:
         svg_content = f.read()
     
-    # Parse SVG to extract width and height
-    try:
-        root = ET.fromstring(svg_content)
-        width = root.get('width')
-        height = root.get('height')
-        if width and height:
-            # Remove units (like 'px') and convert to float
-            width = float(''.join(filter(str.isdigit, width)) or width)
-            height = float(''.join(filter(str.isdigit, height)) or height)
-        else:
-            width = height = None
-    except ET.ParseError:
-        # Handle invalid SVG
-        width = height = None
+    # 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)
     
-    # Define output path
-    output_path = "./test.png"
+    # 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):
+    """Convert SVG to PNG using user-specified crop box dimensions."""
+    if svg_path is None or crop_box is None:
+        return None, None
+    
+    # Extract crop coordinates from the crop_box dictionary
+    # crop_box format: {'left': x1, 'top': y1, 'right': x2, 'bottom': y2}
+    left = crop_box['left']
+    top = crop_box['top']
+    width = crop_box['right'] - left
+    height = crop_box['bottom'] - top
+    
+    # Read the stored SVG content
+    with open(svg_path, 'rb') as f:
+        svg_content = f.read()
+    
+    # Final output path
+    output_path = "./final_output.png"
     
-    # Convert SVG to PNG with explicit size if available
-    if width and height:
-        cairosvg.svg2png(
-            bytestring=svg_content,
-            write_to=output_path,
-            output_width=width,
-            output_height=height
-        )
-    else:
-        # Fallback to default rendering with a reasonable size
-        cairosvg.svg2png(
-            bytestring=svg_content,
-            write_to=output_path,
-            output_width=1600,  # Default width if not specified
-            output_height=1200  # Default height if not specified
-        )
+    # Convert with user-specified dimensions
+    cairosvg.svg2png(
+        bytestring=svg_content,
+        write_to=output_path,
+        output_width=width,
+        output_height=height,
+        # Optional: adjust scale or offset if needed based on original SVG coordinates
+    )
     
-    # Load the PNG as a PIL image
-    png_image = Image.open(output_path)
-    return png_image, output_path
+    # 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")
+    gr.Markdown("# SVG to PNG Converter with Custom Crop")
     
     with gr.Row():
         with gr.Column():
             svg_input = gr.File(label="Upload SVG File", file_types=[".svg"])
-            convert_btn = gr.Button("Convert")
+            preview_btn = gr.Button("Generate Preview")
         
         with gr.Column():
-            png_output = gr.Image(type='pil', label="Converted PNG", height=800)
-            download_btn = gr.File(label="Download PNG")
+            preview_output = gr.Image(
+                type='pil',
+                label="Preview (Draw a box to crop)",
+                interactive=True,
+                tool="sketch",
+                height=800
+            )
+            crop_btn = gr.Button("Convert with Crop")
+            final_output = gr.Image(type='pil', label="Final PNG", height=800)
+            download_btn = gr.File(label="Download Final PNG")
     
-    # Connect the conversion function
-    convert_btn.click(
-        fn=show,
+    # 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=[png_output, download_btn]
+        outputs=[preview_output, svg_state]
+    )
+    
+    # Step 2: Convert using crop box
+    crop_btn.click(
+        fn=final_convert,
+        inputs=[svg_state, preview_output],
+        outputs=[final_output, download_btn]
     )
 
 bl.launch()