Update app.py

app.py

@@ -61,7 +61,7 @@ def preprocess_image(image, blur_radius, sharpen_radius, noise_reduction, detail
         if enhance_with_ai:
             image_np = np.array(image)
             # AI-based enhancement for smoothing edges without background removal
-            # Optionally apply background removal only if remove_bg is checked
+            image_np = cv2.detailEnhance(image_np, sigma_s=10, sigma_r=0.15)
             if remove_bg:
                 image_np = remove(image_np)
             image = Image.fromarray(image_np)
@@ -78,20 +78,20 @@ def convert_image(image, blur_radius, sharpen_radius, noise_reduction, detail_le
                   enhance_with_ai, remove_bg, upscale_factor):
     """Convert an image to SVG using vtracer with customizable and advanced parameters."""
     
-    # Preprocess the image with additional detail level settings
-    image = preprocess_image(image, blur_radius, sharpen_radius, noise_reduction, detail_level, edge_method, color_quantization, enhance_with_ai, remove_bg)
+    try:
+        # Preprocess the image with additional detail level settings
+        image = preprocess_image(image, blur_radius, sharpen_radius, noise_reduction, detail_level, edge_method, color_quantization, enhance_with_ai, remove_bg)
 
-    # Upscale the image if needed
-    if upscale_factor > 1:
-        new_size = (int(image.width * upscale_factor), int(image.height * upscale_factor))
-        image = image.resize(new_size, Image.LANCZOS)
+        # Upscale the image if needed
+        if upscale_factor > 1:
+            new_size = (int(image.width * upscale_factor), int(image.height * upscale_factor))
+            image = image.resize(new_size, Image.LANCZOS)
 
-    # 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()
+        # 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()
 
-    try:
         # Perform the conversion
         svg_str = vtracer.convert_raw_image_to_svg(
             img_bytes,
@@ -169,8 +169,8 @@ with iface:
         fn=convert_image,
         inputs=[
             image_input, blur_radius_input, sharpen_radius_input, noise_reduction_input, detail_level_input, edge_method_input, color_quantization_input,
-            color_mode_input, hierarchical_input, mode_input, filter_speckle_input, 
-            color_precision_input, length_threshold_input, max_iterations_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, enhance_with_ai_input, remove_bg_input, upscale_factor_input
         ],
         outputs=[svg_output, download_output]