Upload app.py with huggingface_hub

app.py

@@ -312,10 +312,14 @@ Answer:"""
 
 def run_attack(text: str, style: str, model_key: str, task: str, progress=gr.Progress()):
     """Run the Unicode attack and compare predictions."""
+    import time
+
     if not text.strip():
         return "", "", "", "Please enter some text.", ""
 
     try:
+        t_start = time.time()
+
         # Transform text
         progress(0.0, desc="Transforming text...")
         styled_text = transform_text(text, style)
@@ -337,13 +341,14 @@ def run_attack(text: str, style: str, model_key: str, task: str, progress=gr.Pro
         styled_pred = get_prediction(model, styled_text, task, model_key)
 
         progress(1.0, desc="Done!")
+        elapsed = time.time() - t_start
 
         # Determine result
         if original_pred != styled_pred:
-            status = f"ATTACK SUCCEEDED: Prediction changed from {original_pred} to {styled_pred}"
+            status = f"ATTACK SUCCEEDED: Prediction changed from {original_pred} to {styled_pred}\nTime taken: {elapsed:.1f}s"
             result_color = "green"
         else:
-            status = f"Attack failed: Prediction unchanged ({original_pred})"
+            status = f"Attack failed: Prediction unchanged ({original_pred})\nTime taken: {elapsed:.1f}s"
             result_color = "red"
 
         yield styled_text, original_pred, styled_pred, status, result_color
@@ -397,12 +402,14 @@ SAMPLE_SENTENCES = {
 
 
 def get_sample_choices(task):
-    """Return dropdown choices for the selected task."""
+    """Return dropdown choices and set text input to first sample."""
     sentences = SAMPLE_SENTENCES.get(task, SAMPLE_SENTENCES['fact_verification'])
-    return gr.update(
+    first = sentences[0]
+    dropdown_update = gr.update(
         choices=[(s[:80] + "..." if len(s) > 80 else s, s) for s in sentences],
-        value=None,
+        value=first,
     )
+    return dropdown_update, first
 
 
 def fill_from_sample(sample):
@@ -492,11 +499,11 @@ def create_demo():
                 outputs=[text_input],
             )
 
-            # Wire up task change → update sample dropdown choices
+            # Wire up task change → update sample dropdown + text input
             task_dropdown.change(
                 fn=get_sample_choices,
                 inputs=[task_dropdown],
-                outputs=[sample_dropdown],
+                outputs=[sample_dropdown, text_input],
             )
 
             run_btn.click(