Update app.py

app.py

@@ -4,14 +4,14 @@ import rembg
 import io
 import base64
 
-def process_image(image, outline_size):
+def process_image(image, outline_size, outline_color):
     if image is None:
         return None
 
     try:
         # Convert image to bytes for rembg
         img_byte_arr = io.BytesIO()
-        image.save(img_byte_arr, format="PNG")
+        image.save(img_byte_arr, format="PNG", dpi=(300, 300))  # Set DPI here
         img_byte_arr.seek(0)
 
         # Remove background
@@ -24,6 +24,14 @@ def process_image(image, outline_size):
         blurred = alpha.filter(ImageFilter.GaussianBlur(radius=outline_size))
         new_alpha = blurred.point(lambda x: 0 if x == 0 else 255)  # Binary mask
 
+        # Define outline color based on selection
+        if outline_color == "white":
+            outline_rgb = (255, 255, 255, 255)  # White
+        elif outline_color == "black":
+            outline_rgb = (0, 0, 0, 255)  # Black
+        else:
+            outline_rgb = (255, 255, 255, 255) #Default white
+
         # Apply outline to the transparent image
         outlined_img = Image.new("RGBA", processed_img.size, (0, 0, 0, 0))  # Fully transparent
         pixels = outlined_img.load()
@@ -32,7 +40,7 @@ def process_image(image, outline_size):
         for x in range(outlined_img.width):
             for y in range(outlined_img.height):
                 if alpha_pixels[x, y] > 0:
-                    pixels[x, y] = (255, 255, 255, 255)  # White outline
+                    pixels[x, y] = outline_rgb
 
         # Overlay original image
         outlined_img = Image.alpha_composite(outlined_img, processed_img)
@@ -40,7 +48,7 @@ def process_image(image, outline_size):
 
         # Save the result as bytes for Gradio
         output_buffer = io.BytesIO()
-        outlined_img.save(output_buffer, format="PNG")
+        outlined_img.save(output_buffer, format="PNG", dpi=(300, 300))  # Set DPI here
         output_buffer.seek(0)
         img_data = base64.b64encode(output_buffer.read()).decode("utf-8")
         data_url = f"data:image/png;base64,{img_data}"
@@ -62,6 +70,11 @@ with gr.Blocks(title="Sticker Maker") as interface:
             outline_size = gr.Slider(
                 label="Outline Thickness", minimum=0, maximum=20, value=5, step=1
             )
+            outline_color = gr.Radio(
+                choices=["white", "black"],
+                value="white",
+                label="Outline Color",
+            )
             create_btn = gr.Button("Create Sticker", variant="primary")
 
         with gr.Column():
@@ -75,9 +88,9 @@ with gr.Blocks(title="Sticker Maker") as interface:
                 """
             )
 
-    def update_image(image, outline_size):
+    def update_image(image, outline_size, outline_color):
         try:
-            data_url = process_image(image, outline_size)
+            data_url = process_image(image, outline_size, outline_color)
             if data_url:
                 return f"""
                     <div style="position: relative;">
@@ -100,10 +113,9 @@ with gr.Blocks(title="Sticker Maker") as interface:
                     </div>
                     """
 
-
     create_btn.click(
         fn=update_image,
-        inputs=[image_upload, outline_size],
+        inputs=[image_upload, outline_size, outline_color],
         outputs=image_output_html,
     )