Update chatgpt-ad-maker.py

chatgpt-ad-maker.py

@@ -1,13 +1,15 @@
 import gradio as gr
 import numpy as np
-from PIL import Image
 
-def dot_effect(input_image):
+def dot_effect(input_image, dot_size, threshold, dot_color):
     """
-    Convert input image to dotted effect similar to the example
+    Convert input image to customizable dotted effect
     
     Args:
         input_image (ndarray): Input image
+        dot_size (int): Size of dots
+        threshold (float): Brightness threshold for dot appearance
+        dot_color (list): RGB color of dots
     
     Returns:
         ndarray: Image with dotted effect
@@ -19,13 +21,12 @@ def dot_effect(input_image):
     normalized = (gray_image - gray_image.min()) / (gray_image.max() - gray_image.min())
     
     # Create dot mask
-    dot_size = 5  # Adjust dot size as needed
     height, width = gray_image.shape
     dot_mask = np.zeros((height, width, 3), dtype=np.uint8)
     
     for y in range(0, height, dot_size):
         for x in range(0, width, dot_size):
-            if normalized[y, x] > 0.3:  # Threshold for dot appearance
+            if normalized[y, x] > threshold:
                 radius = int(dot_size * normalized[y, x])
                 center_x, center_y = x + dot_size//2, y + dot_size//2
                 
@@ -34,18 +35,23 @@ def dot_effect(input_image):
                         if dx*dx + dy*dy <= radius*radius:
                             px, py = center_x + dx, center_y + dy
                             if 0 <= px < width and 0 <= py < height:
-                                dot_mask[py, px] = [255, 255, 255]
+                                dot_mask[py, px] = dot_color
     
     return dot_mask
 
 # Create Gradio interface
 iface = gr.Interface(
     fn=dot_effect,
-    inputs=gr.Image(type="numpy"),
+    inputs=[
+        gr.Image(type="numpy", label="Input Image"),
+        gr.Slider(minimum=1, maximum=20, value=5, label="Dot Size"),
+        gr.Slider(minimum=0, maximum=1, value=0.3, label="Dot Threshold"),
+        gr.ColorPicker(label="Dot Color", value="#FFFFFF")
+    ],
     outputs=gr.Image(type="numpy"),
     title="ChatGPT Ad Maker",
-    description="Transform images into dotted effect"
+    description="Transform images into customizable dotted effect"
 )
 
 # Launch the app
-iface.launch()
+iface.launch()