Update app.py

app.py

@@ -1,14 +1,29 @@
 import io
 import gradio as gr
-from PIL import Image
+from PIL import Image, ImageFilter
 import vtracer
 import tempfile
+import numpy as np
+from skimage import color, filters, feature, io as sk_io
 
-def convert_image(image, color_mode, hierarchical, mode, filter_speckle,
-                  color_precision, layer_difference, corner_threshold,
-                  length_threshold, max_iterations, splice_threshold, path_precision):
+def preprocess_image(image, blur_radius, edge_enhance):
+    """Applies 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)
+
+    return image
+
+def convert_image(image, blur_radius, edge_enhance, 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."""
 
+    # Preprocess the image
+    image = preprocess_image(image, blur_radius, edge_enhance)
+
     # Convert Gradio image to bytes for vtracer compatibility
     img_byte_array = io.BytesIO()
     image.save(img_byte_array, format='PNG')
@@ -44,11 +59,15 @@ def convert_image(image, color_mode, hierarchical, mode, filter_speckle,
 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.")
+    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.")
 
     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")
+
+    with gr.Row():
         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")
@@ -74,10 +93,9 @@ with iface:
     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
+            image_input, blur_radius_input, edge_enhance_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]
     )