Update app.py

app.py

@@ -1,42 +1,85 @@
+import io
 import gradio as gr
-import cairosvg
 from PIL import Image
-import io
+import vtracer
+import tempfile
+
+def convert_image(image, color_mode, hierarchical, mode, filter_speckle,
+                  color_precision, layer_difference, corner_threshold,
+                  length_threshold, max_iterations, splice_threshold, path_precision):
+    """Converts an image to SVG using vtracer with customizable parameters."""
+
+    # Convert Gradio image to bytes for vtracer compatibility
+    img_byte_array = io.BytesIO()
+    image.save(img_byte_array, format='PNG')
+    img_bytes = img_byte_array.getvalue()
+
+    # Perform the conversion
+    svg_str = vtracer.convert_raw_image_to_svg(
+        img_bytes,
+        img_format='png',
+        colormode=color_mode.lower(),
+        hierarchical=hierarchical.lower(),
+        mode=mode.lower(),
+        filter_speckle=int(filter_speckle),
+        color_precision=int(color_precision),
+        layer_difference=int(layer_difference),
+        corner_threshold=int(corner_threshold),
+        length_threshold=float(length_threshold),
+        max_iterations=int(max_iterations),
+        splice_threshold=int(splice_threshold),
+        path_precision=int(path_precision)
+    )
+
+    # Save the SVG string to a temporary file
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.svg')
+    temp_file.write(svg_str.encode('utf-8'))
+    temp_file.close()
+
+    # Display the SVG in the Gradio interface and provide the download link
+    svg_html = f'<svg viewBox="0 0 {image.width} {image.height}">{svg_str}</svg>'
+    return gr.HTML(svg_html), temp_file.name
+
+# Gradio interface
+iface = gr.Blocks()
+
+with iface:
+    gr.Markdown("# Convert Image to SVG Vectors")
+    gr.Markdown("Upload an image and customize the conversion parameters for powerful, high-quality vector results.")
+
+    with gr.Row():
+        image_input = gr.Image(type="pil", label="Upload Image")
+        color_mode_input = gr.Radio(choices=["Color", "Binary"], value="Color", label="Color Mode")
+        hierarchical_input = gr.Radio(choices=["Stacked", "Cutout"], value="Stacked", label="Hierarchical")
+        mode_input = gr.Radio(choices=["Spline", "Polygon", "None"], value="Spline", label="Mode")
+
+    with gr.Row():
+        filter_speckle_input = gr.Slider(minimum=1, maximum=100, value=4, step=1, label="Filter Speckle")
+        color_precision_input = gr.Slider(minimum=1, maximum=100, value=6, step=1, label="Color Precision")
+        layer_difference_input = gr.Slider(minimum=1, maximum=100, value=16, step=1, label="Layer Difference")
+
+    with gr.Row():
+        corner_threshold_input = gr.Slider(minimum=1, maximum=100, value=60, step=1, label="Corner Threshold")
+        length_threshold_input = gr.Slider(minimum=1, maximum=100, value=4.0, step=0.5, label="Length Threshold")
+        max_iterations_input = gr.Slider(minimum=1, maximum=100, value=10, step=1, label="Max Iterations")
+
+    with gr.Row():
+        splice_threshold_input = gr.Slider(minimum=1, maximum=100, value=45, step=1, label="Splice Threshold")
+        path_precision_input = gr.Slider(minimum=1, maximum=100, value=8, step=1, label="Path Precision")
+
+    convert_button = gr.Button("Convert Image to SVG")
+    svg_output = gr.HTML(label="SVG Output")
+    download_output = gr.File(label="Download SVG")
+
+    convert_button.click(
+        fn=convert_image,
+        inputs=[
+            image_input, color_mode_input, hierarchical_input, mode_input,
+            filter_speckle_input, color_precision_input, layer_difference_input,
+            corner_threshold_input, length_threshold_input, max_iterations_input,
+            splice_threshold_input, path_precision_input
+        ],
+        outputs=[svg_output, download_output]
+    )
 
-def convert_image_to_svg(image, scale=1.0):
-    # Convert the uploaded image to bytes
-    image_bytes = io.BytesIO()
-    image.save(image_bytes, format='PNG')
-    image_bytes = image_bytes.getvalue()
-    
-    # Convert the PNG image to SVG using cairosvg
-    svg_bytes = cairosvg.svg2svg(bytestring=image_bytes, scale=scale)
-    
-    # Save the SVG output to a file
-    output_svg = 'output_image.svg'
-    with open(output_svg, 'wb') as f:
-        f.write(svg_bytes)
-    
-    return output_svg
-
-# Set up Gradio interface with customizable options
-def create_gradio_interface():
-    with gr.Blocks() as interface:
-        gr.Markdown("# Image to SVG Converter")
-        gr.Markdown("Upload an image and customize the options to convert it to a high-quality SVG file.")
-        
-        with gr.Row():
-            image_input = gr.Image(type="pil", label="Upload Image")
-            scale_input = gr.Slider(0.5, 5.0, value=1.0, label="Scale")
-        
-        output_file = gr.File(label="Download SVG")
-        
-        convert_button = gr.Button("Convert Image to SVG")
-        convert_button.click(fn=convert_image_to_svg, inputs=[image_input, scale_input], outputs=output_file)
-    
-    return interface
-
-# Launch the Gradio interface
-if __name__ == "__main__":
-    interface = create_gradio_interface()
-    interface.launch()
+iface.launch()