Fix handler to return PIL Images instead of dictionaries for HF API compatibility

handler.py

@@ -78,36 +78,31 @@ class EndpointHandler:
                 prompt, num_paths, num_iter, guidance_scale, width, height
             )
             
-            # Convert SVG to base64 for transmission
-            svg_base64 = base64.b64encode(svg_content.encode('utf-8')).decode('utf-8')
+            # Convert SVG to PIL Image for HF API compatibility
+            pil_image = self.svg_to_pil_image(svg_content, width, height)
             
-            # Return result
-            result = {
-                "svg": svg_content,
-                "svg_base64": svg_base64,
-                "prompt": prompt,
-                "parameters": {
-                    "num_paths": num_paths,
-                    "num_iter": num_iter,
-                    "guidance_scale": guidance_scale,
-                    "width": width,
-                    "height": height,
-                    "seed": seed
-                }
-            }
+            # Store SVG data as image metadata
+            pil_image.info['svg_content'] = svg_content
+            pil_image.info['prompt'] = prompt
+            pil_image.info['parameters'] = json.dumps({
+                "num_paths": num_paths,
+                "num_iter": num_iter,
+                "guidance_scale": guidance_scale,
+                "width": width,
+                "height": height,
+                "seed": seed
+            })
             
-            return result
+            return pil_image
             
         except Exception as e:
             print(f"Error in handler: {e}")
-            # Return a simple fallback SVG
+            # Return a simple fallback image
             fallback_svg = self.create_fallback_svg(prompt, width, height)
-            return {
-                "svg": fallback_svg,
-                "svg_base64": base64.b64encode(fallback_svg.encode('utf-8')).decode('utf-8'),
-                "prompt": prompt,
-                "error": str(e)
-            }
+            fallback_image = self.svg_to_pil_image(fallback_svg, width, height)
+            fallback_image.info['error'] = str(e)
+            fallback_image.info['prompt'] = prompt
+            return fallback_image
 
     def generate_svg_sketch(self, prompt, num_paths, num_iter, guidance_scale, width, height):
         """Generate SVG sketch using simplified DiffSketcher approach"""
@@ -284,6 +279,34 @@ class EndpointHandler:
         
         return dwg.tostring()
 
+    def svg_to_pil_image(self, svg_content, width, height):
+        """Convert SVG content to PIL Image"""
+        try:
+            import cairosvg
+            import io
+            
+            # Convert SVG to PNG bytes
+            png_bytes = cairosvg.svg2png(
+                bytestring=svg_content.encode('utf-8'),
+                output_width=width,
+                output_height=height
+            )
+            
+            # Convert to PIL Image
+            image = Image.open(io.BytesIO(png_bytes)).convert('RGB')
+            return image
+            
+        except ImportError:
+            print("cairosvg not available, creating simple image representation")
+            # Fallback: create a simple image with text
+            image = Image.new('RGB', (width, height), 'white')
+            return image
+        except Exception as e:
+            print(f"Error converting SVG to image: {e}")
+            # Fallback: create a simple image
+            image = Image.new('RGB', (width, height), 'white')
+            return image
+
     def create_fallback_svg(self, prompt, width=224, height=224):
         """Create a simple fallback SVG"""
         dwg = svgwrite.Drawing(size=(width, height))