diff --git a/public/app.js b/public/app.js
index 0305be3..1df069d 100644
--- a/public/app.js
+++ b/public/app.js
@@ -1113,10 +1113,11 @@ aiImproveBtn.addEventListener("click", async function () {
   }
 
   // Показываем индикатор загрузки
+  const originalHTML = aiImproveBtn.innerHTML;
   const originalTitle = aiImproveBtn.title;
   aiImproveBtn.disabled = true;
   aiImproveBtn.innerHTML =
-    '<span class="iconify" data-icon="mdi:loading" style="animation: spin 1s linear infinite;"></span>';
+    '<span class="iconify" data-icon="mdi:loading" style="animation: spin 1s linear infinite;"></span> Обработка...';
   aiImproveBtn.title = "Обработка...";
 
   try {
@@ -1147,8 +1148,7 @@ aiImproveBtn.addEventListener("click", async function () {
   } finally {
     // Восстанавливаем кнопку
     aiImproveBtn.disabled = false;
-    aiImproveBtn.innerHTML =
-      '<span class="iconify" data-icon="mdi:robot"></span>';
+    aiImproveBtn.innerHTML = originalHTML;
     aiImproveBtn.title = originalTitle;
   }
 });
@@ -2597,7 +2597,7 @@ function addNoteEventListeners() {
 
       // Обработчик кнопки ИИ для редактирования
       aiImproveEditBtn.addEventListener("click", async function () {
-        const textarea = noteElement.querySelector(".edit-note-textarea");
+        const textarea = noteContainer.querySelector(".textInput");
         const content = textarea.value.trim();
 
         if (!content) {