Update app.py

app.py

@@ -31,18 +31,11 @@ def initial_render(svg_file):
     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:
+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
     
-    # 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()
@@ -50,13 +43,29 @@ def final_convert(svg_path, crop_box):
     # Final output path
     output_path = "./final_output.png"
     
-    # Convert with user-specified dimensions
+    # 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,
-        # Optional: adjust scale or offset if needed based on original SVG coordinates
+        output_height=height
     )
     
     # Load the final PNG
@@ -64,22 +73,26 @@ def final_convert(svg_path, crop_box):
     return final_image, output_path
 
 with gr.Blocks() as bl:
-    gr.Markdown("# SVG to PNG Converter with Custom Crop")
+    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)",
+                label="Preview (Draw a box to crop if no size specified)",
                 interactive=True,
                 tool="sketch",
                 height=800
             )
-            crop_btn = gr.Button("Convert with Crop")
+            crop_btn = gr.Button("Convert")
             final_output = gr.Image(type='pil', label="Final PNG", height=800)
             download_btn = gr.File(label="Download Final PNG")
     
@@ -93,10 +106,10 @@ with gr.Blocks() as bl:
         outputs=[preview_output, svg_state]
     )
     
-    # Step 2: Convert using crop box
+    # Step 2: Convert using crop box or specified size
     crop_btn.click(
         fn=final_convert,
-        inputs=[svg_state, preview_output],
+        inputs=[svg_state, preview_output, size_input],
         outputs=[final_output, download_btn]
     )