Update app.py

app.py

@@ -4,72 +4,83 @@ from PIL import Image
 import vtracer
 import io
 import os
-import cairosvg
 
 app = Flask(__name__)
 CORS(app, resources={r"/convert-to-vector": {"origins": "*"}})
 
-def convert_to_vector(image):
+def convert_to_vector(image, params):
     input_path = "temp_input.jpg"
     output_svg_path = "temp_output.svg"
-    output_png_path = "temp_output.png"
 
     # Save the input image to a temporary file
     image.save(input_path)
 
-    # Convert the image to SVG using VTracer with default parameters
+    # Convert the image to SVG using VTracer with provided parameters
     vtracer.convert_image_to_svg_py(
         input_path,
         output_svg_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
+        colormode=params.get("colormode", "color"),
+        hierarchical=params.get("hierarchical", "stacked"),
+        mode=params.get("mode", "spline"),
+        filter_speckle=int(params.get("filter_speckle", 4)),
+        color_precision=int(params.get("color_precision", 6)),
+        layer_difference=int(params.get("layer_difference", 16)),
+        corner_threshold=int(params.get("corner_threshold", 60)),
+        length_threshold=float(params.get("length_threshold", 4.0)),
+        max_iterations=int(params.get("max_iterations", 10)),
+        splice_threshold=int(params.get("splice_threshold", 45)),
+        path_precision=int(params.get("path_precision", 3))
     )
 
-    # Convert SVG to PNG
-    cairosvg.svg2png(url=output_svg_path, write_to=output_png_path)
-
-    # Read the PNG output
-    with open(output_png_path, "rb") as f:
-        png_content = f.read()
+    # Read the SVG output
+    with open(output_svg_path, "rb") as f:
+        svg_content = f.read()
 
     # Clean up temporary files
     os.remove(input_path)
     os.remove(output_svg_path)
-    os.remove(output_png_path)
 
-    return png_content
+    return svg_content
 
 @app.route('/convert-to-vector', methods=['POST'])
 def convert_to_vector_endpoint():
     try:
-        # Handle raw image data in the request body
-        if request.data:
+        # Handle image data
+        if request.data and not request.is_json:
             image = Image.open(io.BytesIO(request.data)).convert('RGB')
-        # Fallback for multipart/form-data
         elif 'image' in request.files:
             file = request.files['image']
             image = Image.open(file).convert('RGB')
         else:
             return jsonify({'error': 'No image data provided'}), 400
 
+        # Get parameters from JSON payload or query parameters
+        if request.is_json:
+            params = request.get_json()
+        else:
+            params = {
+                "colormode": request.args.get("colormode", "color"),
+                "hierarchical": request.args.get("hierarchical", "stacked"),
+                "mode": request.args.get("mode", "spline"),
+                "filter_speckle": request.args.get("filter_speckle", 4),
+                "color_precision": request.args.get("color_precision", 6),
+                "layer_difference": request.args.get("layer_difference", 16),
+                "corner_threshold": request.args.get("corner_threshold", 60),
+                "length_threshold": request.args.get("length_threshold", 4.0),
+                "max_iterations": request.args.get("max_iterations", 10),
+                "splice_threshold": request.args.get("splice_threshold", 45),
+                "path_precision": request.args.get("path_precision", 3)
+            }
+
         # Process the image
-        png_content = convert_to_vector(image)
+        svg_content = convert_to_vector(image, params)
 
-        # Return the PNG as a response
+        # Return the SVG as a response
         return send_file(
-            io.BytesIO(png_content),
-            mimetype='image/png',
+            io.BytesIO(svg_content),
+            mimetype='image/svg+xml',
             as_attachment=True,
-            download_name='vector_output.png'
+            download_name='vector_output.svg'
         )
     except Exception as e:
         return jsonify({'error': str(e)}), 500