Update app.py

app.py

@@ -19,8 +19,8 @@ def mask_to_rgb(mask):
     bg_transparent[mask == 1] = [0, 255, 0, 127]
     return bg_transparent
 
-def get_processed_inputs(image, points):
-    input_points = [points]
+def get_processed_inputs(image, x, y):
+    input_points = [[[x, y]]]
     inputs = processor(image, input_points, return_tensors="pt").to(device)
     with torch.no_grad():
         outputs = model(**inputs)
@@ -49,18 +49,18 @@ def inpaint(raw_image, input_mask, prompt, negative_prompt=None, seed=74294536,
     ).images[0]
     return image
 
-def gradio_interface(image, coordinates, positive_prompt, negative_prompt):
+def gradio_interface(image, x, y, positive_prompt, negative_prompt):
     raw_image = Image.fromarray(image).convert("RGB").resize((512, 512))
-    points = [int(coord) for coord in coordinates]
-    mask = get_processed_inputs(raw_image, points)
+    mask = get_processed_inputs(raw_image, x, y)
     processed_image = inpaint(raw_image, mask, positive_prompt, negative_prompt)
     return processed_image, mask_to_rgb(mask)
 
 iface = gr.Interface(
     fn=gradio_interface,
     inputs=[
-        gr.Image(type="numpy", tool="select"),
-        gr.Textbox(label="Coordinates (automatically filled)"),
+        gr.Image(type="numpy"),
+        gr.Number(label="X coordinate"),
+        gr.Number(label="Y coordinate"),
         gr.Textbox(label="Positive Prompt", placeholder="Enter positive prompt here"),
         gr.Textbox(label="Negative Prompt", placeholder="Enter negative prompt here")
     ],
@@ -69,7 +69,7 @@ iface = gr.Interface(
         gr.Image(label="Segmentation Mask")
     ],
     title="Interactive Image Inpainting",
-    description="Select an area on the image, provide prompts, and see the inpainted result."
+    description="Enter X and Y coordinates for the point to segment, provide prompts, and see the inpainted result."
 )
 
 iface.launch(share=True)