show point

app.py

@@ -1,21 +1,38 @@
 import gradio as gr
 from gradio_image_prompter import ImagePrompter
+from PIL import Image, ImageDraw
+
+
+# Function to draw points on the image
+def draw_points_on_image(prompts):
+    image = Image.fromarray(prompts["image"])  # Convert numpy array to PIL Image
+    points = prompts["points"]  # Get the list of points
+
+    draw = ImageDraw.Draw(image)
+    radius = 5  # Radius of the circle to draw
+
+    # Draw each point on the image
+    for point in points:
+        x, y = point
+        draw.ellipse(
+            (x - radius, y - radius, x + radius, y + radius), fill="red", outline="red"
+        )
+
+    return image, points
+
 
 # Define the Gradio interface
 demo = gr.Interface(
-    fn=lambda prompts: (
-        prompts["image"],
-        prompts["points"],
-    ),  # Extract image and points from the ImagePrompter
+    fn=draw_points_on_image,  # Function that handles the image and points
     inputs=ImagePrompter(
         show_label=False
     ),  # ImagePrompter for image input and point selection
     outputs=[
         gr.Image(show_label=False),
         gr.Dataframe(label="Points"),
-    ],  # Outputs: Image and DataFrame of points
-    title="Image Point Collector",
-    description="Upload an image, click on it, and get the coordinates of the clicked points.",
+    ],  # Outputs: Image with points and DataFrame of points
+    title="Image Point Marker",
+    description="Upload an image, click on it, and see the points marked.",
 )
 
 # Launch the Gradio app

requirements.txt

@@ -1,2 +1,3 @@
 gradio
 gradio-image-prompter
+Pillow