Create app.py

app.py

@@ -0,0 +1,124 @@
+import io
+import gradio as gr
+from PIL import Image
+import vtracer
+import tempfile
+
+# Localization dictionary
+TRANSLATIONS = {
+    'en': {
+        'title': 'Convert Image to SVG Vectors',
+        'description': 'Upload an image and customize the conversion parameters as needed.'
+    },
+    'de': {
+        'title': 'Bild in SVG-Vektoren umwandeln',
+        'description': 'Laden Sie ein Bild hoch und passen Sie die Konvertierungsparameter nach Bedarf an.'
+    }
+}
+
+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()
+
+    return (
+        gr.HTML(f'<svg viewBox="0 0 {image.width} {image.height}">{svg_str}</svg>'), 
+        temp_file.name
+    )
+
+# Create Gradio interface
+with gr.Blocks() as vector_converter_interface:
+    # Language selector at the top
+    language_dropdown = gr.Dropdown(
+        choices=['en', 'de'], 
+        value='en', 
+        label='Language / Sprache'
+    )
+    
+    # Title and description that will be updated
+    title_display = gr.Markdown(f"# {TRANSLATIONS['en']['title']}")
+    description_display = gr.Markdown(TRANSLATIONS['en']['description'])
+    
+    # Main interface components
+    with gr.Row():
+        with gr.Column():
+            image_input = gr.Image(type="pil", label="Upload Image")
+            
+        with gr.Column():
+            color_mode = gr.Radio(choices=["Color", "Binary"], value="Color", label="Color Mode")
+            hierarchical = gr.Radio(choices=["Stacked", "Cutout"], value="Stacked", label="Hierarchical")
+            mode = gr.Radio(choices=["Spline", "Polygon", "None"], value="Spline", label="Mode")
+    
+    with gr.Row():
+        with gr.Column():
+            filter_speckle = gr.Slider(minimum=1, maximum=10, value=4, step=1, label="Filter Speckle")
+            color_precision = gr.Slider(minimum=1, maximum=8, value=6, step=1, label="Color Precision")
+            layer_difference = gr.Slider(minimum=1, maximum=32, value=16, step=1, label="Layer Difference")
+            corner_threshold = gr.Slider(minimum=10, maximum=90, value=60, step=1, label="Corner Threshold")
+            
+        with gr.Column():
+            length_threshold = gr.Slider(minimum=3.5, maximum=10, value=4.0, step=0.5, label="Length Threshold")
+            max_iterations = gr.Slider(minimum=1, maximum=20, value=10, step=1, label="Max Iterations")
+            splice_threshold = gr.Slider(minimum=10, maximum=90, value=45, step=1, label="Splice Threshold")
+            path_precision = gr.Slider(minimum=1, maximum=10, value=8, step=1, label="Path Precision")
+    
+    # Convert button
+    convert_btn = gr.Button("Convert to SVG", variant="primary")
+    
+    # Output components
+    with gr.Row():
+        svg_output = gr.HTML(label="SVG Output")
+        file_output = gr.File(label="Download SVG")
+    
+    # Function to update language
+    def update_language(language):
+        new_title = f"# {TRANSLATIONS[language]['title']}"
+        new_description = TRANSLATIONS[language]['description']
+        return new_title, new_description
+    
+    # Connect the language dropdown to update title and description
+    language_dropdown.change(
+        fn=update_language,
+        inputs=[language_dropdown],
+        outputs=[title_display, description_display]
+    )
+    
+    # Connect the convert button to the conversion function
+    convert_btn.click(
+        fn=convert_image,
+        inputs=[
+            image_input, color_mode, hierarchical, mode, filter_speckle,
+            color_precision, layer_difference, corner_threshold,
+            length_threshold, max_iterations, splice_threshold, path_precision
+        ],
+        outputs=[svg_output, file_output]
+    )
+
+if __name__ == "__main__":
+    vector_converter_interface.launch()