Update app.py

app.py

@@ -8,10 +8,7 @@ import os
 app = Flask(__name__)
 CORS(app, resources={r"/convert-to-vector": {"origins": "*"}})
 
-def convert_to_vector(image, 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):
+def convert_to_vector(image):
     input_path = "temp_input.jpg"
     output_path = "temp_output.svg"
 
@@ -19,21 +16,21 @@ def convert_to_vector(image, colormode="color", hierarchical="stacked", mode="sp
         # Save the input image to a temporary file
         image.save(input_path)
 
-        # Convert the image to SVG using VTracer
+        # Convert the image to SVG using VTracer with default parameters from Gradio app
         vtracer.convert_image_to_svg_py(
             input_path,
             output_path,
-            colormode=colormode,
-            hierarchical=hierarchical,
-            mode=mode,
-            filter_speckle=int(filter_speckle),
-            color_precision=int(color_precision),
-            layer_difference=int(layer_difference),
-            corner_threshold=int(corner_threshold),
-            length_threshold=float(length_threshold),
-            max_iterations=int(max_iterations),
-            splice_threshold=int(splice_threshold),
-            path_precision=int(path_precision)
+            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
         )
 
         # Read the SVG output
@@ -51,24 +48,25 @@ def convert_to_vector(image, colormode="color", hierarchical="stacked", mode="sp
 @app.route('/convert-to-vector', methods=['POST'])
 def convert_to_vector_endpoint():
     try:
-        # Check for raw image data
-        if not request.data:
+        # Handle raw image data in the request body
+        if request.data:
+            image = Image.open(io.BytesIO(request.data)).convert('RGB')
+        else:
             return jsonify({'error': 'No image data provided'}), 400
 
-        # Open image from raw data
-        image = Image.open(io.BytesIO(request.data)).convert('RGB')
-
-        # Convert to SVG
+        # Process the image
         svg_content = convert_to_vector(image)
 
-        # Return SVG as file
-        svg_bytes = io.BytesIO(svg_content.encode('utf-8'))
-        svg_bytes.seek(0)
-        return send_file(svg_bytes, mimetype='image/svg+xml', as_attachment=True, download_name='vector_output.svg')
+        # Save the result to a bytes buffer
+        svg_io = io.BytesIO(svg_content.encode('utf-8'))
+        svg_io.seek(0)
+
+        # Return the SVG as a response
+        return send_file(svg_io, mimetype='image/svg+xml', as_attachment=True, download_name='vector_output.svg')
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
-@app.route('/', methods=['GET'])
+@app.route('/')
 def index():
     return jsonify({'status': 'Image to Vector Converter API is running'})