Update app.py

app.py

@@ -1,21 +1,21 @@
-from flask import Flask, request, jsonify
+from flask import Flask, request, jsonify, send_file
 from flask_cors import CORS
 from PIL import Image
-import vtracer
 import io
+import vtracer
 import os
 
 app = Flask(__name__)
-CORS(app, resources={r"/convert-to-vector": {"origins": "*"}})
+CORS(app, resources={r"/vectorize": {"origins": "*"}})
 
 def convert_to_vector(image):
-    input_path = "temp_input.png"
-    output_path = "svg_output.svg"
+    input_path = "temp_input.jpg"
+    output_path = "temp_output.svg"
 
-    # Save input image
+    # Save the input image to a temporary file
     image.save(input_path)
 
-    # Convert to SVG using VTracer
+    # Convert the image to SVG using vtracer with default settings
     vtracer.convert_image_to_svg_py(
         input_path,
         output_path,
@@ -32,36 +32,45 @@ def convert_to_vector(image):
         path_precision=3
     )
 
-    # Read SVG content
-    with open(output_path, "r") as f:
+    # Read the SVG content
+    with open(output_path, "rb") as f:
         svg_content = f.read()
 
-    # Clean up
+    # Clean up temporary files
     os.remove(input_path)
     os.remove(output_path)
 
     return svg_content
 
-@app.route('/convert-to-vector', methods=['POST'])
-def convert_to_vector_endpoint():
+@app.route('/vectorize', methods=['POST'])
+def vectorize_endpoint():
     try:
-        if not request.data:
+        # 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
 
-        # Open image
-        image = Image.open(io.BytesIO(request.data)).convert('RGB')
-
-        # Convert to SVG
+        # Process the image to SVG
         svg_content = convert_to_vector(image)
 
-        return svg_content, 200, {'Content-Type': 'text/plain'}
-
+        # Return the SVG as a response
+        return send_file(
+            io.BytesIO(svg_content),
+            mimetype='image/svg+xml',
+            as_attachment=True,
+            download_name='vectorized.svg'
+        )
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
 @app.route('/')
 def index():
-    return jsonify({'status': 'Image to Vector API is running'})
+    return jsonify({'status': 'Image Vectorization API is running'})
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)