Update app.py

app.py

@@ -4,32 +4,41 @@ from PIL import Image, ImageFilter
 import vtracer
 import tempfile
 import numpy as np
-from skimage import color, filters, feature, io as sk_io
+from skimage import color, filters, feature, io as sk_io, morphology
 
-def preprocess_image(image, blur_radius, edge_enhance):
-    """Applies preprocessing steps to the image before tracing."""
+def preprocess_image(image, blur_radius, edge_enhance, edge_threshold):
+    """Applies advanced preprocessing steps to the image before tracing."""
     if blur_radius > 0:
         image = image.filter(ImageFilter.GaussianBlur(blur_radius))
     
     if edge_enhance:
         image = image.filter(ImageFilter.EDGE_ENHANCE_MORE)
+    
+    # Convert image to grayscale and apply edge detection
+    grayscale_image = image.convert('L')
+    edge_image = grayscale_image.filter(ImageFilter.FIND_EDGES)
+    
+    # Binarize the edges based on the threshold
+    edge_array = np.array(edge_image)
+    binary_edge_array = edge_array > edge_threshold
+    binary_edge_image = Image.fromarray(binary_edge_array.astype('uint8') * 255)
+    
+    return binary_edge_image
 
-    return image
-
-def convert_image(image, blur_radius, edge_enhance, color_mode, hierarchical, mode, filter_speckle,
+def convert_image(image, blur_radius, edge_enhance, edge_threshold, 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."""
+    """Converts an image to SVG using vtracer with customizable parameters, aiming for higher quality vector results."""
 
     # Preprocess the image
-    image = preprocess_image(image, blur_radius, edge_enhance)
+    image = preprocess_image(image, blur_radius, edge_enhance, edge_threshold)
 
     # 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
+    # Perform the conversion with enhanced vector settings
     svg_str = vtracer.convert_raw_image_to_svg(
         img_bytes,
         img_format='png',
@@ -59,13 +68,14 @@ def convert_image(image, blur_radius, edge_enhance, color_mode, hierarchical, mo
 iface = gr.Blocks()
 
 with iface:
-    gr.Markdown("# Enhanced Convert Image to SVG Vectors")
-    gr.Markdown("Upload an image and customize the advanced conversion parameters for high-quality, detailed vector results.")
+    gr.Markdown("# Advanced Vector Image Converter")
+    gr.Markdown("Upload an image and use advanced settings to produce vector images with clear, software-like lines.")
 
     with gr.Row():
         image_input = gr.Image(type="pil", label="Upload Image")
         blur_radius_input = gr.Slider(minimum=0, maximum=10, value=0, step=0.5, label="Blur Radius (for smoothing)")
         edge_enhance_input = gr.Checkbox(value=False, label="Edge Enhance")
+        edge_threshold_input = gr.Slider(minimum=0, maximum=255, value=100, step=5, label="Edge Detection Threshold")
 
     with gr.Row():
         color_mode_input = gr.Radio(choices=["Color", "Binary"], value="Color", label="Color Mode")
@@ -93,7 +103,7 @@ with iface:
     convert_button.click(
         fn=convert_image,
         inputs=[
-            image_input, blur_radius_input, edge_enhance_input, color_mode_input, hierarchical_input, mode_input,
+            image_input, blur_radius_input, edge_enhance_input, edge_threshold_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
         ],