import gradio as gr
from PIL import Image, ImageFilter
import rembg
import io
import base64
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", dpi=(300, 300)) # Set DPI here
img_byte_arr.seek(0)
# Remove background
processed_bytes = rembg.remove(img_byte_arr.getvalue())
processed_img = Image.open(io.BytesIO(processed_bytes)).convert("RGBA")
# Create outline mask
alpha = processed_img.split()[-1]
alpha = Image.merge('L', [alpha])
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()
alpha_pixels = new_alpha.load()
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
# 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", 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}"
return data_url # Return data_url for display
except Exception as e:
print(f"Error processing image: {e}") # Log the error
return None
# 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.")
with gr.Row():
with gr.Column():
image_upload = gr.Image(label="Upload Image", type="pil")
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():
image_output_html = gr.HTML(
"""
"""
)
def update_image(image, outline_size, outline_color):
try:
data_url = process_image(image, outline_size, outline_color)
if data_url:
return f"""
"""
else:
return """
"""
except Exception as e:
print(f"Error in update_image: {e}") # Log the error
return f"""
Error: {e}
"""
create_btn.click(
fn=update_image,
inputs=[image_upload, outline_size, outline_color],
outputs=image_output_html,
)
if __name__ == "__main__":
interface.launch(debug=True)