Update app.py

app.py

@@ -125,7 +125,7 @@ def run_showui(image, query, session_id):
 
 def save_and_upload_data(image_path, query, session_id, is_example_image, votes=None):
     """Save the data to a JSON file and upload to S3."""
-    if is_example_image:
+    if is_example_image == "True":
         return
 
     votes = votes or {"upvotes": 0, "downvotes": 0}
@@ -148,7 +148,7 @@ def save_and_upload_data(image_path, query, session_id, is_example_image, votes=
 
 def update_vote(vote_type, session_id, is_example_image):
     """Update the vote count and re-upload the JSON file."""
-    if is_example_image:
+    if is_example_image == "True":
         return "Example image used. No vote recorded."
 
     local_file_name = f"{session_id}.json"
@@ -183,22 +183,10 @@ examples = [
     ["./examples/safari_google.png", "Click on search bar.", True],
 ]
 
-def check_if_example(query):
-    """Check if the query matches an example query."""
-    print(f"Checking query: {query}")
-    for _, example_query, is_example in examples:
-        print(f"Comparing with example: {example_query}")
-        if query.strip() == example_query.strip():
-            print("Match found!")
-            return is_example
-    print("No match found.")
-    return False
-
 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)
         state_session_id = gr.State(value=None)
-        state_is_example_image = gr.State(value=False)
 
         if not embed_mode:
             gr.HTML(
@@ -222,12 +210,33 @@ def build_demo(embed_mode, concurrency_count=1):
                 )
                 submit_btn = gr.Button(value="Submit", variant="primary")
 
+                # Examples component
                 gr.Examples(
                     examples=[[e[0], e[1]] for e in examples],
                     inputs=[imagebox, textbox],
-                    outputs=[state_is_example_image],
-                    fn=lambda img, query: check_if_example(query),
-                    examples_per_page=3
+                    outputs=[textbox],  # Only update the query textbox
+                    examples_per_page=3,
+                )
+
+                # Add a hidden dropdown to pass the `is_example` flag
+                is_example_dropdown = gr.Dropdown(
+                    choices=["True", "False"],
+                    value="False",
+                    visible=False,
+                    label="Is Example Image",
+                )
+
+                def set_is_example(query):
+                    # Find the example and return its `is_example` flag
+                    for _, example_query, is_example in examples:
+                        if query.strip() == example_query.strip():
+                            return str(is_example)  # Return as string for Dropdown compatibility
+                    return "False"
+
+                textbox.change(
+                    set_is_example,
+                    inputs=[textbox],
+                    outputs=[is_example_dropdown],
                 )
 
             with gr.Column(scale=8):
@@ -249,31 +258,31 @@ def build_demo(embed_mode, concurrency_count=1):
                 
                 save_and_upload_data(image_path, query, session_id, is_example_image)
                 
-                return result_image, click_coords, image_path, session_id, is_example_image
+                return result_image, click_coords, image_path, session_id
 
             submit_btn.click(
                 on_submit,
-                [imagebox, textbox, state_is_example_image],
-                [output_img, output_coords, state_image_path, state_session_id, state_is_example_image],
+                [imagebox, textbox, is_example_dropdown],
+                [output_img, output_coords, state_image_path, state_session_id],
             )
 
             clear_btn.click(
-                lambda: (None, None, None, None, None),
+                lambda: (None, None, None, None),
                 inputs=None,
-                outputs=[imagebox, textbox, output_img, output_coords, state_image_path, state_session_id, state_is_example_image],
+                outputs=[imagebox, textbox, output_img, output_coords, state_image_path, state_session_id],
                 queue=False
             )
 
             upvote_btn.click(
                 lambda session_id, is_example_image: update_vote("upvote", session_id, is_example_image),
-                inputs=[state_session_id, state_is_example_image],
+                inputs=[state_session_id, is_example_dropdown],
                 outputs=[],
                 queue=False
             )
 
             downvote_btn.click(
                 lambda session_id, is_example_image: update_vote("downvote", session_id, is_example_image),
-                inputs=[state_session_id, state_is_example_image],
+                inputs=[state_session_id, is_example_dropdown],
                 outputs=[],
                 queue=False
             )
@@ -287,4 +296,4 @@ if __name__ == "__main__":
         server_port=7860,
         ssr_mode=False,
         debug=True,
-    )
+    )