Update main.js

main.js

@@ -112,9 +112,20 @@ async function sendPrompt() {
   });
 
   const data = await response.json();
-  $('#loadingSpinner').hide();
-  editor.setValue(data.response);
+  // Check if context is empty
+  if (context.trim() === '') {
+    // If context is empty, set the entire editor value
+    editor.setValue(data.response);
+  } else {
+    // Get the current selection range in the editor
+    const selectionStart = editor.getSelectionStart(); // Get start index of selection
+    const selectionEnd = editor.getSelectionEnd(); // Get end index of selection
+    
+    // Replace only the highlighted text with the model's response
+    editor.replaceRange(data.response, selectionStart, selectionEnd);
+  }
   $('#status').text('Ready');
+  $('#loadingSpinner').hide();
 }
 
 // Add this function to handle right-click context menu