Update app.py

app.py

@@ -1,8 +1,11 @@
-import xml.etree.ElementTree as ET
-from PIL import Image
 import cairosvg
+import gradio as gr
+from PIL import Image
+import os
+import xml.etree.ElementTree as ET
 
 def show(svg_file):
+    # Check if no file is uploaded
     if svg_file is None:
         return None, None
     
@@ -16,9 +19,9 @@ def show(svg_file):
         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', ''))
+            # 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:
@@ -37,8 +40,13 @@ def show(svg_file):
             output_height=height
         )
     else:
-        # Fallback to default rendering if no size is specified
-        cairosvg.svg2png(bytestring=svg_content, write_to=output_path)
+        # Fallback to default rendering with a reasonable size
+        cairosvg.svg2png(
+            bytestring=svg_content,
+            write_to=output_path,
+            output_width=800,  # Default width if not specified
+            output_height=600  # Default height if not specified
+        )
     
     # Load the PNG as a PIL image
     png_image = Image.open(output_path)
@@ -53,7 +61,7 @@ with gr.Blocks() as bl:
             convert_btn = gr.Button("Convert")
         
         with gr.Column():
-            png_output = gr.Image(type='pil', label="Converted PNG")
+            png_output = gr.Image(type='pil', label="Converted PNG", height=800)
             download_btn = gr.File(label="Download PNG")
     
     # Connect the conversion function