svg and return png

app.py

@@ -25,7 +25,8 @@ from gradio_imageslider import ImageSlider
 
 from huggingface_hub import hf_hub_download
 import base64
-from io import BytesIO
+from io import BytesIO, StringIO
+import svgwrite
 USE_TORCH_COMPILE = False
 ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
 
@@ -152,6 +153,30 @@ def create_hdr_effect(original_image, hdr):
     hdr_image_8bit = np.clip(hdr_image * 255, 0, 255).astype('uint8')
     return Image.fromarray(cv2.cvtColor(hdr_image_8bit, cv2.COLOR_BGR2RGB))
 
+def save_as_png(image):
+    """Save image as PNG and return as base64."""
+    if isinstance(image, np.ndarray):
+        image = Image.fromarray(image)
+    
+    buffered = BytesIO()
+    image.save(buffered, format="PNG")
+    img_str = base64.b64encode(buffered.getvalue()).decode()
+    return img_str
+
+def vectorize_image(image):
+    svg_io = StringIO()
+    dwg = svgwrite.Drawing(size=(image.width, image.height))
+    for y in range(image.height):
+        for x in range(image.width):
+            r, g, b = image.getpixel((x, y))
+            color = f"rgb({r},{g},{b})"
+            dwg.add(dwg.rect(insert=(x, y), size=(1, 1), fill=color))
+    dwg.write(svg_io)
+
+    # Return the SVG as a response
+    svg_io.seek(0)
+    return svg_io.getvalue()
+
 lazy_pipe = LazyLoadPipeline()
 lazy_pipe.load()
 
@@ -193,8 +218,9 @@ def gradio_process_image(input_image, input_base64, resolution, num_inference_st
     # Convert input_image and result to numpy arrays
     input_array = np.array(input_image)
     result_array = np.array(result)
-    
-    return [input_array, result_array]
+    png_base64 = save_as_png(result)
+    svg = vectorize_image(result)
+    return [input_array, result_array, png_base64, svg]
 
 title = """<h1 align="center">Image Upscaler with Tile Controlnet</h1>
 <p align="center">The main ideas come from</p>

requirements.txt

@@ -12,4 +12,5 @@ peft
 gradio
 pillow
 gradio-imageslider
-pydantic==2.10.6
+pydantic==2.10.6
+svgwrite