Update app.py

app.py

@@ -41,7 +41,7 @@ def update_interface(mode):
         return (
             gr.update(
                 label="🏷️ Categories", 
-                placeholder="Enter comma-separated categories...",
+                placeholder="Enter comma-separated categories..."
             ),
             gr.update()
         )
@@ -49,18 +49,11 @@ def update_interface(mode):
         return (
             gr.update(
                 label="🔎 Hypothesis", 
-                placeholder="Enter a hypothesis to compare with the premise...",
+                placeholder="Enter a hypothesis to compare with the premise..."
             ),
             gr.update()
         )
 
-def handle_example_selection(evt: gr.SelectData, mode):
-    # Return the appropriate label based on the current mode
-    if mode == "Zero-Shot Classification":
-        return gr.update(label="🏷️ Categories")
-    else:
-        return gr.update(label="🔎 Hypothesis")
-
 with gr.Blocks() as demo:
     gr.Markdown("""
     # tasksource/ModernBERT-nli demonstration
@@ -99,18 +92,23 @@ with gr.Blocks() as demo:
             gr.Markdown(label="📈 Analysis", visible=False)
         ]
 
+        # Create a dummy textbox to trigger updates when examples are selected
+        example_trigger = gr.Textbox(visible=False)
+
         with gr.Column(variant="panel") as zero_shot_examples_panel:
-            examples_zero = gr.Examples(
+            gr.Examples(
                 examples=zero_shot_examples,
                 inputs=[text_input, labels_or_premise],
                 label="Zero-Shot Classification Examples",
+                outputs=example_trigger
             )
     
         with gr.Column(variant="panel") as nli_examples_panel:
-            examples_nli = gr.Examples(
+            gr.Examples(
                 examples=nli_examples,
                 inputs=[text_input, labels_or_premise],
                 label="Natural Language Inference Examples",
+                outputs=example_trigger
             )
 
     def update_visibility(mode):
@@ -132,17 +130,11 @@ with gr.Blocks() as demo:
         outputs=[zero_shot_examples_panel, nli_examples_panel]
     )
     
-    # Handle example selection events
-    examples_zero.select(
-        fn=handle_example_selection,
-        inputs=[mode],
-        outputs=[labels_or_premise]
-    )
-    
-    examples_nli.select(
-        fn=handle_example_selection,
+    # Update the label whenever an example is selected
+    example_trigger.change(
+        fn=update_interface,
         inputs=[mode],
-        outputs=[labels_or_premise]
+        outputs=[labels_or_premise, text_input]
     )
     
     submit_btn.click(