Update app.py

app.py

@@ -1,44 +1,45 @@
-import cairosvg
-import gradio as gr
+import xml.etree.ElementTree as ET
 from PIL import Image
-import os
-from io import BytesIO
+import cairosvg
 
 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
+    # Read the SVG content
     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)
+    # 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 'px' units and convert to float
+            width = float(width.replace('px', ''))
+            height = float(height.replace('px', ''))
+        else:
+            width = height = None
+    except ET.ParseError:
+        # Handle invalid SVG
+        width = height = None
     
-    # 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")
+    # Define output path
+    output_path = "./test.png"
     
-    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")
+    # 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 if no size is specified
+        cairosvg.svg2png(bytestring=svg_content, write_to=output_path)
     
-    # Connect the conversion function
-    convert_btn.click(
-        fn=show,
-        inputs=svg_input,
-        outputs=[png_output, download_btn]
-    )
-
-bl.launch()
+    # Load the PNG as a PIL image
+    png_image = Image.open(output_path)
+    return png_image, output_path