Update app.py

app.py

@@ -13,7 +13,7 @@ CORS(app, resources={r"/convert-to-vector": {"origins": "*"}})
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
 
 def convert_to_vector(image):
-    input_path = "temp_input.jpg"
+    input_path = "temp_input.png"  # Use .png to support PNG inputs
     output_svg_path = "temp_output.svg"
 
     try:
@@ -65,15 +65,34 @@ def convert_to_vector_endpoint():
         # Handle image data
         if request.data:
             app.logger.debug("Received raw image data")
-            image = Image.open(io.BytesIO(request.data)).convert('RGB')
+            # Try to open the image without assuming format
+            try:
+                image = Image.open(io.BytesIO(request.data))
+            except Exception as e:
+                app.logger.error(f"Failed to open image: {str(e)}")
+                return jsonify({'error': f"Invalid image data: {str(e)}"}), 400
         elif 'image' in request.files:
             app.logger.debug("Received multipart image data")
             file = request.files['image']
-            image = Image.open(file).convert('RGB')
+            try:
+                image = Image.open(file)
+            except Exception as e:
+                app.logger.error(f"Failed to open image: {str(e)}")
+                return jsonify({'error': f"Invalid image data: {str(e)}"}), 400
         else:
             app.logger.error("No image data provided")
             return jsonify({'error': 'No image data provided'}), 400
 
+        # Log image format and mode
+        app.logger.debug(f"Image format: {image.format}, mode: {image.mode}")
+
+        # Convert to RGB if needed (for vtracer compatibility)
+        if image.mode not in ['RGB', 'RGBA']:
+            image = image.convert('RGB')
+        elif image.mode == 'RGBA':
+            # Optionally convert RGBA to RGB to avoid alpha channel issues
+            image = image.convert('RGB')
+
         # Process image
         app.logger.debug("Processing image")
         svg_content = convert_to_vector(image)