Update app.py

app.py

@@ -171,6 +171,29 @@ 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")
 
+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],
+]
+
+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)
@@ -185,17 +208,6 @@ def build_demo(embed_mode, concurrency_count=1):
                         <img src="data:image/png;base64,{base64_image}" alt="ShowUI" width="320" style="margin-bottom: 10px;"/>
                     </div>
                     <p>ShowUI is a lightweight vision-language-action model for GUI agents.</p>
-                    <div style="display: flex; justify-content: center; gap: 15px; font-size: 20px;">
-                        <a href="https://huggingface.co/showlab/ShowUI-2B" target="_blank">
-                            <img src="https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-ShowUI--2B-blue" alt="model"/>
-                        </a>
-                        <a href="https://arxiv.org/abs/2411.17465" target="_blank">
-                            <img src="https://img.shields.io/badge/arXiv%20paper-2411.17465-b31b1b.svg" alt="arXiv"/>
-                        </a>
-                        <a href="https://github.com/showlab/ShowUI" target="_blank">
-                            <img src="https://img.shields.io/badge/GitHub-ShowUI-black" alt="GitHub"/>
-                        </a>
-                    </div>
                 </div>
                 """
             )
@@ -210,25 +222,6 @@ def build_demo(embed_mode, concurrency_count=1):
                 )
                 submit_btn = gr.Button(value="Submit", variant="primary")
 
-                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],
-                ]
-
-                def check_if_example(query):
-                    """Check if the query matches an example."""
-                    for _, example_query, is_example in examples:
-                        if example_query == query:
-                            return is_example
-                    return False
-
                 gr.Examples(
                     examples=[[e[0], e[1]] for e in examples],
                     inputs=[imagebox, textbox],
@@ -239,11 +232,6 @@ def build_demo(embed_mode, concurrency_count=1):
 
             with gr.Column(scale=8):
                 output_img = gr.Image(type="pil", label="Output Image")
-                gr.HTML(
-                    """
-                    <p><strong>Note:</strong> The <span style="color: red;">red point</span> on the output image represents the predicted clickable coordinates.</p>
-                    """
-                )
                 output_coords = gr.Textbox(label="Clickable Coordinates")
 
                 with gr.Row(elem_id="action-buttons", equal_height=True):