Update app.py

app.py

@@ -12,38 +12,35 @@ def convert_to_vector(image):
     input_path = "temp_input.jpg"
     output_path = "temp_output.svg"
 
-    try:
-        # Save the input image to a temporary file
-        image.save(input_path)
+    # Save the input image to a temporary file
+    image.save(input_path)
 
-        # Convert the image to SVG using VTracer with default parameters from Gradio app
-        vtracer.convert_image_to_svg_py(
-            input_path,
-            output_path,
-            colormode="color",
-            hierarchical="stacked",
-            mode="spline",
-            filter_speckle=4,
-            color_precision=6,
-            layer_difference=16,
-            corner_threshold=60,
-            length_threshold=4.0,
-            max_iterations=10,
-            splice_threshold=45,
-            path_precision=3
-        )
+    # Convert the image to SVG using VTracer with default parameters
+    vtracer.convert_image_to_svg_py(
+        input_path,
+        output_path,
+        colormode="color",
+        hierarchical="stacked",
+        mode="spline",
+        filter_speckle=4,
+        color_precision=6,
+        layer_difference=16,
+        corner_threshold=60,
+        length_threshold=4.0,
+        max_iterations=10,
+        splice_threshold=45,
+        path_precision=3
+    )
+
+    # Read the SVG output
+    with open(output_path, "rb") as f:
+        svg_content = f.read()
 
-        # Read the SVG output
-        with open(output_path, "r") as f:
-            svg_content = f.read()
+    # Clean up temporary files
+    os.remove(input_path)
+    os.remove(output_path)
 
-        return svg_content
-    finally:
-        # Clean up temporary files
-        if os.path.exists(input_path):
-            os.remove(input_path)
-        if os.path.exists(output_path):
-            os.remove(output_path)
+    return svg_content
 
 @app.route('/convert-to-vector', methods=['POST'])
 def convert_to_vector_endpoint():
@@ -51,18 +48,23 @@ def convert_to_vector_endpoint():
         # Handle raw image data in the request body
         if request.data:
             image = Image.open(io.BytesIO(request.data)).convert('RGB')
+        # Fallback for multipart/form-data (if needed later)
+        elif 'image' in request.files:
+            file = request.files['image']
+            image = Image.open(file).convert('RGB')
         else:
             return jsonify({'error': 'No image data provided'}), 400
 
         # Process the image
         svg_content = convert_to_vector(image)
 
-        # Save the result to a bytes buffer
-        svg_io = io.BytesIO(svg_content.encode('utf-8'))
-        svg_io.seek(0)
-
         # Return the SVG as a response
-        return send_file(svg_io, mimetype='image/svg+xml', as_attachment=True, download_name='vector_output.svg')
+        return send_file(
+            io.BytesIO(svg_content),
+            mimetype='image/svg+xml',
+            as_attachment=True,
+            download_name='vector_output.svg'
+        )
     except Exception as e:
         return jsonify({'error': str(e)}), 500