Update app.py

app.py

@@ -1,10 +1,11 @@
-from flask import Flask, request, jsonify, send_file
+from flask import Flask, request, jsonify
 from flask_cors import CORS
 import vtracer
 from PIL import Image
 import io
 import os
 import logging
+import base64
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -76,24 +77,30 @@ def convert_to_vector(
 @app.route('/convert', methods=['POST'])
 def convert_image():
     try:
-        # Handle image upload
-        if 'file' not in request.files:
-            return jsonify({'error': 'No image file provided'}), 400
-        file = request.files['file']
-        image = Image.open(file).convert('RGB')
+        # Handle JSON payload
+        data = request.get_json()
+        if not data or 'file' not in data:
+            return jsonify({'error': 'No image data provided'}), 400
+
+        # Decode base64 image
+        base64_string = data['file']
+        if base64_string.startswith('data:image'):
+            base64_string = base64_string.split(',')[1]
+        image_data = base64.b64decode(base64_string)
+        image = Image.open(io.BytesIO(image_data)).convert('RGB')
 
         # Get parameters (with defaults)
-        colormode = request.form.get('colormode', 'color')
-        hierarchical = request.form.get('hierarchical', 'stacked')
-        mode = request.form.get('mode', 'spline')
-        filter_speckle = int(request.form.get('filter_speckle', 4))
-        color_precision = int(request.form.get('color_precision', 6))
-        layer_difference = int(request.form.get('layer_difference', 16))
-        corner_threshold = int(request.form.get('corner_threshold', 60))
-        length_threshold = float(request.form.get('length_threshold', 4.0))
-        max_iterations = int(request.form.get('max_iterations', 10))
-        splice_threshold = int(request.form.get('splice_threshold', 45))
-        path_precision = int(request.form.get('path_precision', 3))
+        colormode = data.get('colormode', 'color')
+        hierarchical = data.get('hierarchical', 'stacked')
+        mode = data.get('mode', 'spline')
+        filter_speckle = int(data.get('filter_speckle', 4))
+        color_precision = int(data.get('color_precision', 6))
+        layer_difference = int(data.get('layer_difference', 16))
+        corner_threshold = int(data.get('corner_threshold', 60))
+        length_threshold = float(data.get('length_threshold', 4.0))
+        max_iterations = int(data.get('max_iterations', 10))
+        splice_threshold = int(data.get('splice_threshold', 45))
+        path_precision = int(data.get('path_precision', 3))
 
         logger.info("Received request to /convert")