Remove parentheses if weight == 1

1 files changed, 8 insertions, 2 deletions

diff --git a/javascript/edit-attention.js b/javascript/edit-attention.js
index 5fe1117c..c02d292f 100644
--- a/javascript/edit-attention.js
+++ b/javascript/edit-attention.js
@@ -99,7 +99,13 @@ function keyupEditAttention(event){
 	weight = parseFloat(weight.toPrecision(12));

 	if(String(weight).length == 1) weight += ".0"

 

-	text = text.slice(0, selectionEnd + 1) + weight + text.slice(selectionEnd + 1 + end - 1);

+    if (closeCharacter == ')' && weight == 1) {

+        text = text.slice(0, selectionStart - 1) + text.slice(selectionStart, selectionEnd) + text.slice(selectionEnd + 5);

+        selectionStart--;

+        selectionEnd--;

+    } else {

+        text = text.slice(0, selectionEnd + 1) + weight + text.slice(selectionEnd + 1 + end - 1);

+    }

 

 	target.focus();

 	target.value = text;

@@ -111,4 +117,4 @@ function keyupEditAttention(event){
 

 addEventListener('keydown', (event) => {

     keyupEditAttention(event);

-});
\ No newline at end of file
+});