Update app.py

app.py

@@ -524,6 +524,11 @@ def capture_selection(text, seqlen, current_ids, evt: gr.SelectData | None = Non
     indices_csv = to_ranges(token_idxs)
     return indices_csv, f"Selected chars [{start}:{end}) → tokens {indices_csv}"
 
+def noise_selection(src, state_ids, seqlen, indices_csv, seed):
+    # Reuse apply_noise but force prepend/append noise to zero
+    return apply_noise(src, state_ids, seqlen, indices_csv, 0, 0, seed)
+
+
 def apply_noise(src, state_ids, seqlen, indices_csv, add_left, add_right, seed):
     ensure_model(src)
     tok = model_cache["tokenizer"]
@@ -623,11 +628,12 @@ with gr.Blocks(title="CNA — Interactive Denoising") as demo:
 
     # “Noise Selection” just applies whatever is in indices_csv
     btn_noise_selection.click(
-        apply_noise,
-        [src, ids_state, seqlen, indices_csv, 0, 0, seed],
+        noise_selection,
+        [src, ids_state, seqlen, indices_csv, seed],
         [ids_state, current_text, status]
     )
 
+
     # Manual indices + prepend/append noise
     btn_apply_noise.click(
         apply_noise,