Update app.py

app.py

@@ -4,7 +4,6 @@ from PIL import Image
 import vtracer
 import io
 import os
-import cairosvg
 
 app = Flask(__name__)
 CORS(app, resources={r"/convert-to-vector": {"origins": "*"}})
@@ -12,12 +11,11 @@ CORS(app, resources={r"/convert-to-vector": {"origins": "*"}})
 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
+    # Convert the image to SVG using VTracer
     vtracer.convert_image_to_svg_py(
         input_path,
         output_svg_path,
@@ -34,19 +32,15 @@ def convert_to_vector(image):
         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, "r") 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():
@@ -62,14 +56,14 @@ def convert_to_vector_endpoint():
             return jsonify({'error': 'No image data provided'}), 400
 
         # Process the image
-        png_content = convert_to_vector(image)
+        svg_content = convert_to_vector(image)
 
-        # 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.encode('utf-8')),
+            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