Update app.py

app.py

@@ -27,25 +27,36 @@ def upload_to_aws(local_file, s3_file):
         print("Credentials not available")
         return None
 
-def convert_image(image_url, color_mode, hierarchical, mode, filter_speckle,
+def convert_image(image_input, color_mode, hierarchical, mode, filter_speckle,
                   color_precision, layer_difference, corner_threshold,
                   length_threshold, max_iterations, splice_threshold, path_precision):
-    """Converts an image from a URL to SVG using vtracer with customizable parameters."""
-    
-    # Debugging: Print the received image_url and its type
-    print(f"Received image_url: {image_url}, type: {type(image_url)}")
-    
-    if not isinstance(image_url, str):
-        return gr.HTML(f'<p style="color: red;">Error: Expected a string URL, got {type(image_url)}</p>'), None, None
+    """Converts an image from a URL or file path to SVG using vtracer with customizable parameters."""
     
     try:
-        # Fetch the image from the URL
-        response = requests.get(image_url)
-        response.raise_for_status()  # Raise an error for bad status codes
-        image_data = response.content
+        # Debugging: Print the received input and its type
+        print(f"Received image_input: {image_input}, type: {type(image_input)}")
+        
+        # Handle both URL strings and Gradio FileData objects
+        if isinstance(image_input, dict) and 'path' in image_input:
+            # If the input is a Gradio FileData object, use the file path
+            image_path = image_input['path']
+            with open(image_path, 'rb') as f:
+                image_data = f.read()
+        elif isinstance(image_input, str):
+            # If the input is a URL, fetch the image
+            if image_input.startswith(('http://', 'https://')):
+                response = requests.get(image_input)
+                response.raise_for_status()  # Raise an error for bad status codes
+                image_data = response.content
+            else:
+                return gr.HTML(f'<p style="color: red;">Error: Invalid URL format. Expected a valid HTTP/HTTPS URL.</p>'), None, None
+        else:
+            return gr.HTML(f'<p style="color: red;">Error: Invalid input type. Expected a URL or file path, got {type(image_input)}</p>'), None, None
 
         # Load the image using PIL
         image = Image.open(io.BytesIO(image_data))
+        print(f"Image format: {image.format}, Size: {image.size}, Mode: {image.mode}")
+        image.show()  # Display the image for debugging
 
         # Convert the image to bytes for vtracer compatibility
         img_byte_array = io.BytesIO()
@@ -53,6 +64,14 @@ def convert_image(image_url, color_mode, hierarchical, mode, filter_speckle,
         image.save(img_byte_array, format=img_format)
         img_bytes = img_byte_array.getvalue()
 
+        # Debugging: Save the input image to a temporary file
+        with tempfile.NamedTemporaryFile(delete=False, suffix='.png') as temp_img_file:
+            image.save(temp_img_file, format='PNG')
+            print(f"Saved input image to: {temp_img_file.name}")
+
+        # Debugging: Print vtracer parameters
+        print(f"vtracer parameters: img_format={img_format}, colormode={color_mode}, hierarchical={hierarchical}, mode={mode}, filter_speckle={filter_speckle}, color_precision={color_precision}, layer_difference={layer_difference}, corner_threshold={corner_threshold}, length_threshold={length_threshold}, max_iterations={max_iterations}, splice_threshold={splice_threshold}, path_precision={path_precision}")
+
         # Perform the conversion
         svg_str = vtracer.convert_raw_image_to_svg(
             img_bytes,
@@ -69,6 +88,7 @@ def convert_image(image_url, color_mode, hierarchical, mode, filter_speckle,
             splice_threshold=int(splice_threshold),
             path_precision=int(path_precision)
         )
+        print(f"Generated SVG: {svg_str}")  # Debugging: Print the generated SVG
     except Exception as e:
         return gr.HTML(f'<p style="color: red;">Error during image processing or SVG conversion: {str(e)}</p>'), None, None
 
@@ -89,7 +109,7 @@ def convert_image(image_url, color_mode, hierarchical, mode, filter_speckle,
 
     # Return the SVG wrapped in HTML and the file URL
     return gr.HTML(f'<svg viewBox="0 0 {image.width} {image.height}">{svg_str}</svg>'), file_url, svg_str
-
+    
     
 # Gradio interface
 iface = gr.Blocks()