Update app.py

app.py

@@ -4,7 +4,7 @@ import rembg
 import io
 import base64
 
-def process_image(image, outline_size, outline_color):
+def process_image(image, outline_size):
     if image is None:
         return None
 
@@ -24,14 +24,6 @@ def process_image(image, outline_size, outline_color):
         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()
@@ -40,7 +32,7 @@ def process_image(image, outline_size, outline_color):
         for x in range(outlined_img.width):
             for y in range(outlined_img.height):
                 if alpha_pixels[x, y] > 0:
-                    pixels[x, y] = outline_rgb
+                    pixels[x, y] = (255, 255, 255, 255)  # White outline
 
         # Overlay original image
         outlined_img = Image.alpha_composite(outlined_img, processed_img)
@@ -62,7 +54,7 @@ def process_image(image, outline_size, outline_color):
 # 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 an outline.")
+    gr.Markdown("Upload an image to remove the background and add a white outline.")
 
     with gr.Row():
         with gr.Column():
@@ -70,14 +62,9 @@ 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
+        with gr.Column():
             # Use HTML to create an image and a download link
             image_output_html = gr.HTML(
                 """
@@ -88,9 +75,9 @@ with gr.Blocks(title="Sticker Maker") as interface:
                 """
             )
 
-    def update_image(image, outline_size, outline_color):
+    def update_image(image, outline_size):
         try:
-            data_url = process_image(image, outline_size, outline_color)
+            data_url = process_image(image, outline_size)
             if data_url:
                 return f"""
                     <div style="position: relative;">
@@ -113,9 +100,10 @@ with gr.Blocks(title="Sticker Maker") as interface:
                     </div>
                     """
 
+
     create_btn.click(
         fn=update_image,
-        inputs=[image_upload, outline_size, outline_color],
+        inputs=[image_upload, outline_size],
         outputs=image_output_html,
     )