Update app.py

app.py

@@ -176,21 +176,6 @@ 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)
@@ -232,17 +217,17 @@ def build_demo(embed_mode, concurrency_count=1):
 
                 gr.Examples(
                     examples=[
-                        ["./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],
+                        ["./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."],
                     ],
-                    inputs=[imagebox, textbox, state_is_example_image],
+                    inputs=[imagebox, textbox],
                     examples_per_page=3
                 )
 
@@ -260,9 +245,23 @@ 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, state_is_example_image],
+                [imagebox, textbox],
                 [output_img, output_coords, state_image_path, state_session_id, state_is_example_image],
             )
 
@@ -296,13 +295,4 @@ if __name__ == "__main__":
         server_port=7860,
         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,
-    )
+    )