Update app.py

app.py

@@ -8,7 +8,7 @@ import gradio as gr
 import matplotlib
 
 from gradio_utils.utils import (process_img, get_select_coords, select_skeleton,
-                                reset_skeleton, reset_kp, process)
+                                reset_skeleton, reset_kp, process, update_examples)
 
 LENGTH = 480  # Length of the square area displaying/editing images
 
@@ -45,6 +45,7 @@ with gr.Blocks() as demo:
         "skeleton": [],
         "prev_point": None,
         "curr_type_point": "start",
+        "load_example": False,
     })
     with gr.Row():
         # Upload & Preprocess Image Column
@@ -57,7 +58,7 @@ with gr.Blocks() as demo:
                 width=LENGTH,
                 type="pil",
                 image_mode="RGB",
-                label="Preprocess Image",
+                label="Support Image",
                 show_label=True,
                 interactive=True,
             )
@@ -116,19 +117,61 @@ with gr.Blocks() as demo:
             gr.Markdown(
                 """<p style="text-align: center; font-size: 20px">Output</p>"""
             )
-            output_img = gr.Plot(label="Output Image",)
+            output_img = gr.Plot(label="Output Image", )
     with gr.Row():
         eval_btn = gr.Button(value="Evaluate")
     with gr.Row():
         gr.Markdown("## Examples")
+    with gr.Row():
+        example_null = gr.Textbox(type='text',
+                                  visible=False
+                                  )
+    with gr.Row():
+        examples = gr.Examples([
+            ['examples/dog2.png',
+             'examples/dog1.png',
+             json.dumps({
+                 'points': [(232, 200), (312, 204), (228, 264), (316, 472), (316, 616), (296, 868), (412, 872),
+                            (416, 624), (604, 608), (648, 860), (764, 852), (696, 608), (684, 432)],
+                 'skeleton': [(0, 1), (1, 2), (0, 2), (3, 4), (4, 5),
+                              (3, 7), (7, 6), (3, 12), (12, 8), (8, 9),
+                              (12, 11), (11, 10)],
+             })
+             ],
+            ['examples/sofa1.jpg',
+             'examples/sofa2.jpg',
+             json.dumps({
+                 'points': [(112, 328), (120, 260), (104, 208), (200, 260), (208, 336), (216, 212), (208, 172),
+                            (284, 180), (276, 324), (156, 308), (172, 228), (256, 232), (168, 184), (260, 196)]
+
+,
+                 'skeleton': [(0, 1), (3, 1), (3, 4), (10, 9), (11, 8),
+                              (1, 10), (10, 11), (11, 3), (1, 2), (7, 6),
+                              (5, 13), (5, 3), (13, 11), (12, 10), (12, 2),
+                              (6, 10), (7, 11)],
+             })],
+            ['examples/person1.jpeg',
+             'examples/person2.jpeg',
+             json.dumps({
+                 'points': [(380, 484), (640, 488), (520, 616), (424, 736), (612, 724)],
+                 'skeleton': [(0, 1), (1, 2), (0, 2), (2, 3), (2, 4),
+                              (4, 3)],
+             })]
+        ],
+            inputs=[support_image, query_image, example_null],
+            outputs=[support_image, kp_support_image, skel_support_image, query_image, global_state],
+            fn=update_examples,
+            run_on_click=True,
+            examples_per_page=5,
+        )
 
-    support_image.change(process_img,
+    support_image.upload(process_img,
                          inputs=[support_image, global_state],
                          outputs=[kp_support_image, global_state])
     kp_support_image.select(get_select_coords,
-                       [global_state],
-                       [global_state, kp_support_image],
-                       queue=False,)
+                            [global_state],
+                            [global_state, kp_support_image],
+                            queue=False, )
     confirm_kp_button.click(reset_skeleton,
                             inputs=global_state,
                             outputs=skel_support_image)
@@ -136,16 +179,17 @@ with gr.Blocks() as demo:
                          inputs=global_state,
                          outputs=[kp_support_image, skel_support_image])
     undo_skel_button.click(reset_skeleton,
-                            inputs=global_state,
-                            outputs=skel_support_image)
+                           inputs=global_state,
+                           outputs=skel_support_image)
     skel_support_image.select(select_skeleton,
                               inputs=[global_state],
                               outputs=[global_state, skel_support_image])
     eval_btn.click(fn=process,
                    inputs=[query_image, global_state],
-                   outputs=[output_img, global_state])
+                   outputs=[output_img])
 
 if __name__ == "__main__":
     print("Start app", parser.parse_args())
     gr.close_all()
     demo.launch(show_api=False)
+