Update app.py

app.py

@@ -4,13 +4,15 @@ 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):
     input_path = "temp_input.jpg"
-    output_path = "temp_output.svg"
+    output_svg_path = "temp_output.svg"
+    output_png_path = "temp_output.png"
 
     # Save the input image to a temporary file
     image.save(input_path)
@@ -18,7 +20,7 @@ def convert_to_vector(image):
     # Convert the image to SVG using VTracer with default parameters
     vtracer.convert_image_to_svg_py(
         input_path,
-        output_path,
+        output_svg_path,
         colormode="color",
         hierarchical="stacked",
         mode="spline",
@@ -32,15 +34,19 @@ def convert_to_vector(image):
         path_precision=3
     )
 
-    # Read the SVG output
-    with open(output_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_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():
@@ -48,7 +54,7 @@ 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)
+        # Fallback for multipart/form-data
         elif 'image' in request.files:
             file = request.files['image']
             image = Image.open(file).convert('RGB')
@@ -56,14 +62,14 @@ def convert_to_vector_endpoint():
             return jsonify({'error': 'No image data provided'}), 400
 
         # Process the image
-        svg_content = convert_to_vector(image)
+        png_content = convert_to_vector(image)
 
-        # Return the SVG as a response
+        # 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