Spaces:
Running
Running
File size: 2,362 Bytes
f71fab4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
import gradio as gr
from PIL import Image, ImageDraw
def halftone_effect(img, dot_size=10):
"""
Apply a halftone effect to the input image.
Parameters:
- img: PIL.Image object (input image)
- dot_size: int, spacing and maximum cell size for dot placement
Returns:
- PIL.Image object with halftone effect
"""
# Convert to grayscale
grayscale = img.convert('L')
width, height = grayscale.size
# Create a new white image for the halftone effect
halftone_image = Image.new('L', (width, height), color=255)
draw = ImageDraw.Draw(halftone_image)
# Loop through the image in steps of dot_size
for x in range(0, width, dot_size):
for y in range(0, height, dot_size):
# Determine the size of the current block (to handle image edges)
block_width = min(dot_size, width - x)
block_height = min(dot_size, height - y)
# Extract the region of interest and calculate its average brightness
region = grayscale.crop((x, y, x + block_width, y + block_height))
avg_brightness = sum(region.getdata()) / (block_width * block_height)
# Calculate radius based on brightness (brighter = smaller dot)
max_radius = dot_size // 2
radius = int((255 - avg_brightness) / 255 * max_radius)
radius = max(0, min(radius, max_radius)) # Clamp radius between 0 and max_radius
# Draw the circle at the center of the current cell
center_x = x + dot_size // 2
center_y = y + dot_size // 2
draw.ellipse(
(
center_x - radius,
center_y - radius,
center_x + radius,
center_y + radius
),
fill=0 # Black dot
)
return halftone_image
# Gradio Interface
interface = gr.Interface(
fn=halftone_effect,
inputs=[
gr.Image(type="pil", label="Upload an Image"),
gr.Slider(minimum=5, maximum=20, step=1, value=10, label="Dot Size")
],
outputs=gr.Image(label="Halftone Image"),
title="Halftone Image Generator",
description="Convert any image into a stylized halftone effect using black dots.",
allow_flagging="never"
)
if __name__ == "__main__":
interface.launch() |