Create app.py

app.py

@@ -0,0 +1,99 @@
+import gradio as gr
+import vtracer
+import os
+import tempfile
+
+def image_to_svg(image, colormode, hierarchical, mode, filter_speckle, color_precision, layer_difference, corner_threshold, length_threshold, max_iterations, splice_threshold, path_precision):
+    """
+    Converts an input image to an SVG string using vtracer.
+    """
+    if image is None:
+        return None, None
+
+    # Create a temporary directory to store the output file
+    with tempfile.TemporaryDirectory() as temp_dir:
+        # Use the uploaded file's original name for the output, but with an .svg extension
+        base, _ = os.path.splitext(os.path.basename(image.name))
+        output_filename = f"{base}.svg"
+        output_path = os.path.join(temp_dir, output_filename)
+
+        # Convert the image with the specified parameters
+        vtracer.convert_image_to_svg_py(
+            image.name,
+            output_path,
+            colormode=colormode,
+            hierarchical=hierarchical,
+            mode=mode,
+            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)
+        )
+
+        # Read the content of the generated SVG file
+        with open(output_path, "r") as f:
+            svg_content = f.read()
+
+        return output_path, svg_content
+
+# --- Gradio User Interface ---
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# IMG2SVG: Image to SVG Converter")
+    gr.Markdown("Upload your JPG, PNG, or other image file to convert it to a scalable vector graphic (SVG). Adjust the settings below for finer control over the conversion process.")
+
+    with gr.Row():
+        with gr.Column(scale=1):
+            image_input = gr.Image(type="filepath", label="Upload Image")
+            
+            with gr.Accordion("Conversion Settings", open=True):
+                colormode = gr.Dropdown(["color", "binary"], value="color", label="Color Mode")
+                hierarchical = gr.Dropdown(["stacked", "cutout"], value="stacked", label="Hierarchical")
+                mode = gr.Dropdown(["spline", "polygon", "none"], value="spline", label="Mode")
+                filter_speckle = gr.Slider(0, 16, value=4, step=1, label="Filter Speckle")
+                color_precision = gr.Slider(0, 8, value=6, step=1, label="Color Precision")
+                layer_difference = gr.Slider(0, 32, value=16, step=1, label="Layer Difference")
+                corner_threshold = gr.Slider(0, 180, value=60, step=1, label="Corner Threshold")
+                length_threshold = gr.Slider(0.0, 10.0, value=4.0, step=0.5, label="Length Threshold")
+                max_iterations = gr.Slider(1, 20, value=10, step=1, label="Max Iterations")
+                splice_threshold = gr.Slider(0, 90, value=45, step=1, label="Splice Threshold")
+                path_precision = gr.Slider(1, 8, value=3, step=1, label="Path Precision")
+            
+            convert_button = gr.Button("Convert to SVG", variant="primary")
+
+        with gr.Column(scale=2):
+            svg_file_output = gr.File(label="Download SVG")
+            svg_text_output = gr.Code(label="SVG Code", language="xml")
+
+    # Connect the button to the conversion function
+    convert_button.click(
+        fn=image_to_svg,
+        inputs=[
+            image_input,
+            colormode,
+            hierarchical,
+            mode,
+            filter_speckle,
+            color_precision,
+            layer_difference,
+            corner_threshold,
+            length_threshold,
+            max_iterations,
+            splice_threshold,
+            path_precision
+        ],
+        outputs=[svg_file_output, svg_text_output]
+    )
+    
+    with gr.Accordion("About vtracer", open=False):
+        gr.Markdown(
+            "This application uses the `vtracer` library to convert raster images into vector graphics. [1] "
+            "It offers a powerful alternative to tools like Adobe Illustrator's Image Trace, with a focus on creating compact SVG files. [1] "
+            "The core of `vtracer` is built in Rust for high performance. [1]"
+        )
+
+# To launch the application
+demo.launch()