update

app.py

@@ -176,24 +176,31 @@ def run_chm(
         ax[0].text(*src, str(i), color="white", bbox=dict(facecolor="black", alpha=0.5))
         ax[1].text(*tgt, str(i), color="black", bbox=dict(facecolor="white", alpha=0.7))
 
+    # Create a colormap that will generate 49 distinct colors
+    cmap = plt.get_cmap(
+        "gist_rainbow", 49
+    )  # 'gist_rainbow' is just an example, you can choose another colormap
+
     # Drawing lines between corresponding source and target points
-    for src, tgt in zip(src_points_converted, tgt_points_converted):
-        con = ConnectionPatch(
-            xyA=tgt,
-            xyB=src,
-            coordsA="data",
-            coordsB="data",
-            axesA=ax[1],
-            axesB=ax[0],
-            color="green",
-        )
-        ax[1].add_artist(con)
+    # for i, (src, tgt) in enumerate(zip(src_points_converted, tgt_points_converted)):
+    #     con = ConnectionPatch(
+    #         xyA=tgt,
+    #         xyB=src,
+    #         coordsA="data",
+    #         coordsB="data",
+    #         axesA=ax[1],
+    #         axesB=ax[0],
+    #         color=cmap(i),
+    #         linewidth=2,
+    #     )
+        # ax[1].add_artist(con)
 
     # Adding legend
-    ax[0].legend()
-    ax[1].legend()
+    ax[0].legend(loc="lower right", bbox_to_anchor=(1, -0.075))
+    ax[1].legend(loc="lower right", bbox_to_anchor=(1, -0.075))
 
     plt.tight_layout()
+    plt.subplots_adjust(wspace=0.1, hspace=0.1)
     fig.suptitle("CHM Correspondences\nUsing $\it{pas\_psi.pt}$ Weights ", fontsize=16)
     return fig
 
@@ -216,40 +223,6 @@ def generate_correspondences(
     )
 
 
-# # Gradio App
-# main = gr.Interface(
-#     fn=generate_correspondences,
-#     inputs=[
-#         gr.Image(shape=(240, 240), type="pil"),
-#         gr.Image(shape=(240, 240), type="pil"),
-#         gr.Slider(minimum=1, maximum=240, step=1, default=15, label="Min X"),
-#         gr.Slider(minimum=1, maximum=240, step=1, default=215, label="Max X"),
-#         gr.Slider(minimum=1, maximum=240, step=1, default=15, label="Min Y"),
-#         gr.Slider(minimum=1, maximum=240, step=1, default=215, label="Max Y"),
-#     ],
-#     allow_flagging="never",
-#     outputs="plot",
-#     examples=[
-#         ["./examples/sample1.jpeg", "./examples/sample2.jpeg", 17, 223, 17, 223],
-#         [
-#             "./examples/Red_Winged_Blackbird_0012_6015.jpg",
-#             "./examples/Red_Winged_Blackbird_0025_5342.jpg",
-#             17,
-#             223,
-#             17,
-#             223,
-#         ],
-#         [
-#             "./examples/Yellow_Headed_Blackbird_0026_8545.jpg",
-#             "./examples/Yellow_Headed_Blackbird_0020_8549.jpg",
-#             17,
-#             223,
-#             17,
-#             223,
-#         ],
-#     ],
-# )
-
 with gr.Blocks() as demo:
     gr.Markdown(
         """
@@ -320,6 +293,22 @@ Performs keypoint transform from a 7x7 gird on the source image to the target im
     gr.Examples(
         [
             ["./examples/sample1.jpeg", "./examples/sample2.jpeg", 17, 223, 17, 223],
+            [
+                "./examples/Red_Winged_Blackbird_0012_6015.jpg",
+                "./examples/Red_Winged_Blackbird_0025_5342.jpg",
+                17,
+                223,
+                17,
+                223,
+            ],
+            [
+                "./examples/Yellow_Headed_Blackbird_0026_8545.jpg",
+                "./examples/Yellow_Headed_Blackbird_0020_8549.jpg",
+                17,
+                223,
+                17,
+                223,
+            ],
         ],
         inputs=[
             image1,
@@ -331,7 +320,6 @@ Performs keypoint transform from a 7x7 gird on the source image to the target im
         ],
     )
 
-    # Add a Button component to run the app.
     run_btn = gr.Button("Run")
 
     run_btn.click(