Fix: Update handler to return PIL Image directly

handler.py

@@ -43,24 +43,31 @@ class EndpointHandler:
                 <text x="50%" y="50%" font-family="Arial" font-size="20" text-anchor="middle">{prompt}</text>
             </svg>"""
             
-            # Convert SVG to base64
-            svg_bytes = svg_content.encode("utf-8")
-            svg_base64 = base64.b64encode(svg_bytes).decode("utf-8")
-            
             # Convert SVG to PNG using cairosvg
             try:
-                png_data = cairosvg.svg2png(bytestring=svg_bytes)
-                png_base64 = base64.b64encode(png_data).decode("utf-8")
+                png_data = cairosvg.svg2png(bytestring=svg_content.encode("utf-8"))
+                # Create a PIL Image from the PNG data
+                image = Image.open(io.BytesIO(png_data))
             except Exception as e:
                 print(f"Error converting SVG to PNG: {e}")
-                # Return a transparent 1x1 pixel PNG as fallback
-                png_base64 = "iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAQAAAC1HAwCAAAAC0lEQVR42mNkYAAAAAYAAjCB0C8AAAAASUVORK5CYII="
+                # Create a simple placeholder image
+                image = Image.new("RGB", (width, height), color="#f0f0f0")
+                # Add text to the image
+                from PIL import ImageDraw, ImageFont
+                draw = ImageDraw.Draw(image)
+                try:
+                    font = ImageFont.truetype("Arial", 20)
+                except:
+                    font = ImageFont.load_default()
+                draw.text((width/2, height/2), prompt, fill="black", font=font, anchor="mm")
             
-            # Return the results
-            return {
-                "svg": svg_content,
-                "svg_base64": svg_base64,
-                "png_base64": png_base64
-            }
+            # Return the PIL Image directly
+            return image
         except Exception as e:
-            return {"error": str(e)}
+            print(f"Error in handler: {e}")
+            # Return a simple error image
+            image = Image.new("RGB", (512, 512), color="#ff0000")
+            from PIL import ImageDraw
+            draw = ImageDraw.Draw(image)
+            draw.text((256, 256), f"Error: {str(e)}", fill="white", anchor="mm")
+            return image