Update app.py

app.py

@@ -32,9 +32,8 @@ def process_input(text_input, labels_or_premise, mode):
         results = {label: score for label, score in zip(prediction['labels'], prediction['scores'])}
         return results, ''
     else:  # NLI mode
-        pred= nli_classifier([{"text": text_input, "text_pair": labels_or_premise}],return_all_scores=True)[0]
-        results= {pred['label']:pred['score'] for pred in pred}
-
+        pred = nli_classifier([{"text": text_input, "text_pair": labels_or_premise}], return_all_scores=True)[0]
+        results = {pred['label']: pred['score'] for pred in pred}
         return results, ''
 
 def update_interface(mode):
@@ -43,20 +42,25 @@ def update_interface(mode):
             gr.update(
                 label="🏷️ Categories", 
                 placeholder="Enter comma-separated categories...",
-                value=zero_shot_examples[0][1]
             ),
-            gr.update(value=zero_shot_examples[0][0])
+            gr.update()
         )
     else:
         return (
             gr.update(
                 label="🔎 Hypothesis", 
                 placeholder="Enter a hypothesis to compare with the premise...",
-                value=nli_examples[0][1]
             ),
-            gr.update(value=nli_examples[0][0])
+            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
@@ -78,14 +82,14 @@ with gr.Blocks() as demo:
             label="✍️ Input Text",
             placeholder="Enter your text...",
             lines=3,
-            value=zero_shot_examples[0][0]  # Initial value
+            value=zero_shot_examples[0][0]
         )
         
         labels_or_premise = gr.Textbox(
             label="🏷️ Categories",
             placeholder="Enter comma-separated categories...",
             lines=2,
-            value=zero_shot_examples[0][1]  # Initial value
+            value=zero_shot_examples[0][1]
         )
         
         submit_btn = gr.Button("Submit")
@@ -96,18 +100,18 @@ with gr.Blocks() as demo:
         ]
 
         with gr.Column(variant="panel") as zero_shot_examples_panel:
-            gr.Examples(
+            examples_zero = gr.Examples(
                 examples=zero_shot_examples,
                 inputs=[text_input, labels_or_premise],
                 label="Zero-Shot Classification Examples",
             )
     
         with gr.Column(variant="panel") as nli_examples_panel:
-            gr.Examples(
+            examples_nli = gr.Examples(
                 examples=nli_examples,
                 inputs=[text_input, labels_or_premise],
                 label="Natural Language Inference Examples",
-        )
+            )
 
     def update_visibility(mode):
         return (
@@ -115,6 +119,7 @@ with gr.Blocks() as demo:
             gr.update(visible=(mode == "Natural Language Inference"))
         )
 
+    # Update interface on mode change
     mode.change(
         fn=update_interface,
         inputs=[mode],
@@ -127,6 +132,19 @@ 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,
+        inputs=[mode],
+        outputs=[labels_or_premise]
+    )
+    
     submit_btn.click(
         fn=process_input,
         inputs=[text_input, labels_or_premise, mode],