Update app.py

app.py

@@ -176,6 +176,21 @@ def update_vote(vote_type, session_id, is_example_image):
 with open("./assets/showui.png", "rb") as image_file:
     base64_image = base64.b64encode(image_file.read()).decode("utf-8")
 
+def on_submit(image, query, example_flag):
+    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")
+    
+    # Use the example_flag to determine if the image is an example
+    is_example_image = example_flag
+    
+    result_image, click_coords, image_path = run_showui(image, query, session_id)
+    
+    save_and_upload_data(image_path, query, session_id, is_example_image)
+    
+    return result_image, click_coords, image_path, session_id, is_example_image
+
 def build_demo(embed_mode, concurrency_count=1):
     with gr.Blocks(title="ShowUI Demo", theme=gr.themes.Default()) as demo:
         state_image_path = gr.State(value=None)
@@ -217,17 +232,17 @@ def build_demo(embed_mode, concurrency_count=1):
 
                 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/app_store.png", "Download Kindle.", True],
+                        ["./examples/ios_setting.png", "Turn off Do not disturb.", True],
+                        ["./examples/apple_music.png", "Star to favorite.", True],
+                        ["./examples/map.png", "Boston.", True],
+                        ["./examples/wallet.png", "Scan a QR code.", True],
+                        ["./examples/word.png", "More shapes.", True],
+                        ["./examples/web_shopping.png", "Proceed to checkout.", True],
+                        ["./examples/web_forum.png", "Post my comment.", True],
+                        ["./examples/safari_google.png", "Click on search bar.", True],
                     ],
-                    inputs=[imagebox, textbox],
+                    inputs=[imagebox, textbox, state_is_example_image],
                     examples_per_page=3
                 )
 
@@ -245,23 +260,9 @@ 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):
-                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)
-                
-                save_and_upload_data(image_path, query, session_id, is_example_image)
-                
-                return result_image, click_coords, image_path, session_id, is_example_image
-
             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],
             )
 
@@ -296,3 +297,12 @@ if __name__ == "__main__":
         ssr_mode=False,
         debug=True,
     )
+
+if __name__ == "__main__":
+    demo = build_demo(embed_mode=False)
+    demo.queue(api_open=False).launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        ssr_mode=False,
+        debug=True,
+    )