Fix: Update handler.py with robust cairosvg import and PNG conversion

handler.py

@@ -5,8 +5,23 @@ import json
 import torch
 import numpy as np
 from PIL import Image
-import cairosvg
-from diffusers import StableDiffusionPipeline
+
+# Safely import cairosvg with fallback
+try:
+    import cairosvg
+except ImportError:
+    print("Warning: cairosvg not found. Installing...")
+    import subprocess
+    subprocess.check_call(["pip", "install", "cairosvg", "cairocffi", "cssselect2", "defusedxml", "tinycss2"])
+    import cairosvg
+
+try:
+    from diffusers import StableDiffusionPipeline
+except ImportError:
+    print("Warning: diffusers not found. Installing...")
+    import subprocess
+    subprocess.check_call(["pip", "install", "diffusers", "transformers", "accelerate"])
+    from diffusers import StableDiffusionPipeline
 
 class ModelHandler:
     def __init__(self):
@@ -68,9 +83,19 @@ class ModelHandler:
         svg_bytes = inference_output.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')
+        except Exception as e:
+            print(f"Error converting SVG to PNG: {e}")
+            # Return a transparent 1x1 pixel PNG as fallback
+            png_base64 = "iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAQAAAC1HAwCAAAAC0lEQVR42mNkYAAAAAYAAjCB0C8AAAAASUVORK5CYII="
+        
         return {
             "svg": inference_output,
-            "svg_base64": svg_base64
+            "svg_base64": svg_base64,
+            "png_base64": png_base64
         }
     
     def handle(self, data):