downsampled images and only overlay points at len=4

app.py

@@ -60,7 +60,6 @@ def click_callback(img_path, evt: gr.SelectData):
     # Save off a copy of the first input image
     if len(roi_coords) == 0:
         input_image = copy.copy(img)
-    display_image = copy.copy(input_image)
 
     # Either create a new coordinate, or adjust the position of an existing coordinate
     if len(roi_coords) < 4:
@@ -69,17 +68,20 @@ def click_callback(img_path, evt: gr.SelectData):
         distances = [math.dist(evt.index, coord) for coord in roi_coords]
         roi_coords[np.argmin(distances)] = evt.index
 
-    # Only overlay the bounding box if all four coordinates have been defined
     if len(roi_coords) == 4:
+        display_image = copy.copy(input_image)
+
         # Overlay the corners of the ROI
         pts = np.array(roi_coords, np.int32).reshape((-1, 1, 2))
         cv2.polylines(display_image, [pts], True, (255, 255, 255), 2)
 
-    # Always overlay the location of the coordinates
-    for coord in roi_coords:
-        cv2.circle(display_image, coord, radius=5, color=(255, 0, 0), thickness=-1)
+        # Always overlay the location of the coordinates
+        for coord in roi_coords:
+            cv2.circle(display_image, coord, radius=5, color=(255, 0, 0), thickness=-1)
 
-    return display_image, *get_warped_image()
+        return display_image, *get_warped_image()
+    else:
+        return input_image, *get_warped_image()
 
 
 def clear_variables(*kwargs):
@@ -143,7 +145,7 @@ with gr.Blocks() as demo:
             image_output = gr.Image(label="Cropped Warp", type="filepath", tool="editor", interactive=True)
             slider_image_width = gr.Slider(label="Width", minimum=10, maximum=900, step=1)
             slider_image_height = gr.Slider(label="Height", minimum=10, maximum=900, step=1)
-            json_t_matrix = gr.Numpy(label="Homography", row_count=3, col_count=3, headers=['', '', ''], interactive=False)
+            json_t_matrix = gr.Numpy(label="Transformation Matrix", row_count=3, col_count=3, headers=['', '', ''], interactive=False)
 
     image_input.select(click_callback, image_input, [image_input, image_output, json_t_matrix, slider_image_width, slider_image_height])
     image_input.clear(clear_variables)
@@ -152,5 +154,5 @@ with gr.Blocks() as demo:
     slider_image_height.release(resize_image, [image_output, gr.State(None), slider_image_height], image_output)
 
 demo.queue(concurrency_count=10, max_size=20)
-demo.launch(inbrowser=True)
-# demo.launch()
+# demo.launch(inbrowser=True)
+demo.launch()