Create app.py

app.py

@@ -0,0 +1,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()