Update app.py

app.py

@@ -5,17 +5,19 @@ import vtracer
 import tempfile
 
 def convert_image(image, color_mode, hierarchical, mode, filter_speckle,
-                 color_precision, layer_difference, corner_threshold,
-                 length_threshold, max_iterations, splice_threshold, path_precision):
+                  color_precision, layer_difference, corner_threshold,
+                  length_threshold, max_iterations, splice_threshold, path_precision):
     """Converts an image to SVG using vtracer with customizable parameters."""
-    # Save image to temporary file, as vtracer 0.6.11 requires a file path
-    with tempfile.NamedTemporaryFile(suffix='.png', delete=False) as temp_img:
-        image.save(temp_img, format='PNG')
-        temp_img_path = temp_img.name
 
-    # Convert image to SVG using vtracer
-    svg_str = vtracer.convert_image_to_svg(
-        image_path=temp_img_path,
+    # 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(),
@@ -29,38 +31,41 @@ def convert_image(image, color_mode, hierarchical, mode, filter_speckle,
         path_precision=int(path_precision)
     )
 
-    # Save SVG to temporary file for download
+    # 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 temp_file.name
+    return gr.HTML(f'<svg viewBox="0 0 {image.width} {image.height}">{svg_str}</svg>'), temp_file.name
 
 # Gradio interface
 iface = gr.Blocks()
 
 with iface:
+    
     gr.Interface(
-        fn=convert_image,
-        inputs=[
-            gr.Image(type="pil", label="Upload Image"),
-            gr.Radio(choices=["color", "binary"], value="color", label="Color Mode"),
-            gr.Radio(choices=["stacked", "cutout"], value="stacked", label="Hierarchical"),
-            gr.Radio(choices=["spline", "polygon", "none"], value="spline", label="Mode"),
-            gr.Slider(minimum=1, maximum=10, value=4, step=1, label="Filter Speckle"),
-            gr.Slider(minimum=1, maximum=8, value=6, step=1, label="Color Precision"),
-            gr.Slider(minimum=1, maximum=32, value=16, step=1, label="Layer Difference"),
-            gr.Slider(minimum=10, maximum=90, value=60, step=1, label="Corner Threshold"),
-            gr.Slider(minimum=3.5, maximum=10, value=4.0, step=0.5, label="Length Threshold"),
-            gr.Slider(minimum=1, maximum=20, value=10, step=1, label="Max Iterations"),
-            gr.Slider(minimum=10, maximum=90, value=45, step=1, label="Splice Threshold"),
-            gr.Slider(minimum=1, maximum=10, value=8, step=1, label="Path Precision")
-        ],
-        outputs=[
-            gr.File(label="Download SVG")
-        ],
-        title="Convert Image to SVG vectors",
-        description="Upload an image and customize the conversion parameters as needed.<br><h2>Support me USDT (TRC-20): TAe7hsSVWtMEYz3G5V1UiUdYPQVqm28bKx</h2>",
-    )
+    fn=convert_image,
+    inputs=[
+        gr.Image(type="pil", label="Upload Image"),
+        gr.Radio(choices=["Color", "Binary"], value="Color", label="Color Mode"),
+        gr.Radio(choices=["Stacked", "Cutout"], value="Stacked", label="Hierarchical"),
+        gr.Radio(choices=["Spline", "Polygon", "None"], value="Spline", label="Mode"),
+        gr.Slider(minimum=1, maximum=10, value=4, step=1, label="Filter Speckle"),
+        gr.Slider(minimum=1, maximum=8, value=6, step=1, label="Color Precision"),
+        gr.Slider(minimum=1, maximum=32, value=16, step=1, label="Layer Difference"),
+        gr.Slider(minimum=10, maximum=90, value=60, step=1, label="Corner Threshold"),
+        gr.Slider(minimum=3.5, maximum=10, value=4.0, step=0.5, label="Length Threshold"),
+        gr.Slider(minimum=1, maximum=20, value=10, step=1, label="Max Iterations"),
+        gr.Slider(minimum=10, maximum=90, value=45, step=1, label="Splice Threshold"),
+        gr.Slider(minimum=1, maximum=10, value=8, step=1, label="Path Precision")
+    ],
+    outputs=[
+        gr.HTML(label="SVG Output"),
+        gr.File(label="Download SVG")
+    ],
+    title="Convert Image to SVG vectors",
+    description="Upload an image and customize the conversion parameters as needed.<br><h2>Support me USDT (TRC-20): TAe7hsSVWtMEYz3G5V1UiUdYPQVqm28bKx</h2>",
+)
 
-iface.launch(ssr=False)
+    
+iface.launch()