Update app.py

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
-from PIL import Image
+from PIL import Image, ImageFilter
 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
 
@@ -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,11 +40,12 @@ 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)
 
+
         # Save the result as bytes for Gradio
         output_buffer = io.BytesIO()
         outlined_img.save(output_buffer, format="PNG")
@@ -53,7 +62,7 @@ def process_image(image, outline_size):
 # Gradio Interface
 with gr.Blocks(title="Sticker Maker") as interface:
     gr.Markdown("# Sticker Maker")
-    gr.Markdown("Upload an image to remove the background and add a white outline.")
+    gr.Markdown("Upload an image to remove the background and add an outline.")
 
     with gr.Row():
         with gr.Column():
@@ -61,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(visible=True):  # Ensure this column is always visible
@@ -74,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;">
@@ -101,7 +115,7 @@ with gr.Blocks(title="Sticker Maker") as interface:
 
     create_btn.click(
         fn=update_image,
-        inputs=[image_upload, outline_size],
+        inputs=[image_upload, outline_size, outline_color],
         outputs=image_output_html,
     )