Update app.py

app.py

@@ -4,75 +4,72 @@ 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, params):
+def convert_to_vector(image):
     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
     vtracer.convert_image_to_svg_py(
         input_path,
         output_svg_path,
-        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))
+        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
     )
 
-    with open(output_svg_path, "rb") as f:
-        svg_content = f.read()
+    # 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()
+
+    # Clean up temporary files
     os.remove(input_path)
     os.remove(output_svg_path)
+    os.remove(output_png_path)
 
-    return svg_content
+    return png_content
 
 @app.route('/convert-to-vector', methods=['POST'])
 def convert_to_vector_endpoint():
     try:
-        if request.data and not request.is_json:
+        # 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
         elif 'image' in request.files:
             file = request.files['image']
             image = Image.open(file).convert('RGB')
         else:
             return jsonify({'error': 'No image data provided'}), 400
 
-        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)
-            }
-
-        svg_content = convert_to_vector(image, params)
+        # Process the image
+        png_content = convert_to_vector(image)
 
+        # Return the PNG as a response
         return send_file(
-            io.BytesIO(svg_content),
-            mimetype='image/svg+xml',
+            io.BytesIO(png_content),
+            mimetype='image/png',
             as_attachment=True,
-            download_name='vector_output.svg'
+            download_name='vector_output.png'
         )
     except Exception as e:
         return jsonify({'error': str(e)}), 500