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

handler.py

@@ -78,13 +78,31 @@ class EndpointHandler:
                 prompt, n_particle, num_iter, guidance_scale, width, height, style
             )
             
-            return particles
+            # Convert the first particle to PIL Image for HF API compatibility
+            if particles and len(particles) > 0:
+                main_svg = particles[0]["svg"]
+                pil_image = self.svg_to_pil_image(main_svg, width, height)
+                
+                # Store all particles data as metadata
+                pil_image.info['particles'] = json.dumps(particles)
+                pil_image.info['prompt'] = prompt
+                pil_image.info['style'] = style
+                pil_image.info['n_particle'] = str(n_particle)
+                
+                return pil_image
+            else:
+                # Fallback
+                fallback_svg = self.create_fallback_svg(prompt, width, height, style)
+                return self.svg_to_pil_image(fallback_svg, width, height)
             
         except Exception as e:
             print(f"Error in handler: {e}")
-            # Return fallback particles
-            fallback_particles = self.create_fallback_particles(prompt, n_particle, width, height, style)
-            return fallback_particles
+            # Return fallback image
+            fallback_svg = self.create_fallback_svg(prompt, width, height, style)
+            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_particles(self, prompt, n_particle, num_iter, guidance_scale, width, height, style):
         """Generate multiple SVG particles using SVGDreamer approach"""
@@ -527,6 +545,37 @@ class EndpointHandler:
         
         return particles
 
+    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
+            from PIL import 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
+            from PIL import Image
+            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
+            from PIL import Image
+            image = Image.new('RGB', (width, height), 'white')
+            return image
+
     def create_fallback_svg(self, prompt, width, height, style):
         """Create simple fallback SVG"""
         dwg = svgwrite.Drawing(size=(width, height))