Update app.py

app.py

@@ -215,19 +215,27 @@ def build_demo(embed_mode, concurrency_count=1):
                 )
                 submit_btn = gr.Button(value="Submit", variant="primary")
 
+                examples = [
+                    ["./examples/app_store.png", "Download Kindle."],
+                    ["./examples/ios_setting.png", "Turn off Do not disturb."],
+                    ["./examples/apple_music.png", "Star to favorite."],
+                    ["./examples/map.png", "Boston."],
+                    ["./examples/wallet.png", "Scan a QR code."],
+                    ["./examples/word.png", "More shapes."],
+                    ["./examples/web_shopping.png", "Proceed to checkout."],
+                    ["./examples/web_forum.png", "Post my comment."],
+                    ["./examples/safari_google.png", "Click on search bar."],
+                ]
+
+                def set_example_flag(image_path, query):
+                    is_example = any(image_path == example[0] for example in examples)
+                    return image_path, query, is_example
+
                 gr.Examples(
-                    examples=[
-                        ["./examples/app_store.png", "Download Kindle."],
-                        ["./examples/ios_setting.png", "Turn off Do not disturb."],
-                        ["./examples/apple_music.png", "Star to favorite."],
-                        ["./examples/map.png", "Boston."],
-                        ["./examples/wallet.png", "Scan a QR code."],
-                        ["./examples/word.png", "More shapes."],
-                        ["./examples/web_shopping.png", "Proceed to checkout."],
-                        ["./examples/web_forum.png", "Post my comment."],
-                        ["./examples/safari_google.png", "Click on search bar."],
-                    ],
+                    examples=examples,
                     inputs=[imagebox, textbox],
+                    outputs=[state_is_example_image],
+                    fn=set_example_flag,
                     examples_per_page=3
                 )
 
@@ -245,13 +253,11 @@ def build_demo(embed_mode, concurrency_count=1):
                     downvote_btn = gr.Button(value="Too bad!", variant="secondary")
                     clear_btn = gr.Button(value="🗑️ Clear", interactive=True)
 
-            def on_submit(image, query):
+            def on_submit(image, query, is_example_image):
                 if image is None:
                     raise ValueError("No image provided. Please upload an image before submitting.")
                 
                 session_id = datetime.now().strftime("%Y%m%d_%H%M%S")
-                print("image_path:", image)
-                is_example_image = isinstance(image, str) and image.startswith("./examples/")
                 
                 result_image, click_coords, image_path = run_showui(image, query, session_id)
                 
@@ -261,7 +267,7 @@ def build_demo(embed_mode, concurrency_count=1):
 
             submit_btn.click(
                 on_submit,
-                [imagebox, textbox],
+                [imagebox, textbox, state_is_example_image],
                 [output_img, output_coords, state_image_path, state_session_id, state_is_example_image],
             )
 
@@ -295,4 +301,4 @@ if __name__ == "__main__":
         server_port=7860,
         ssr_mode=False,
         debug=True,
-    )
+    )