Реализация AI-рерайтинга текста в мобильном приложении

Реализация AI-рерайтинга текста в мобильном приложении

Рерайтинг в мобильном приложении — узкая задача: пользователь выделил фрагмент, нажал кнопку, получил переписанную версию. Простая идея, но реализаций-граблей здесь больше, чем кажется.

Работа с выделением текста

Самое трудное — не AI-часть, а корректная работа с selectedRange при замене текста. Если заменить NSRange неправильно, курсор прыгает в начало, выделение слетает, история undo ломается.

// iOS: безопасная замена выделенного текста с сохранением undo
func replaceSelection(with newText: String) {
    guard let textView = self.textView,
          let selectedRange = Range(textView.selectedRange, in: textView.text) else { return }

    // Регистрируем undo перед изменением
    textView.undoManager?.registerUndo(withTarget: self) { [oldText = textView.text, oldRange = textView.selectedRange] target in
        target.restoreText(oldText, cursorAt: oldRange)
    }

    textView.textStorage.beginEditing()
    textView.textStorage.replaceCharacters(
        in: textView.selectedRange,
        with: NSAttributedString(string: newText, attributes: textView.typingAttributes)
    )
    textView.textStorage.endEditing()

    // Устанавливаем курсор в конец вставленного текста
    let newCursorPos = textView.selectedRange.location + newText.utf16.count
    textView.selectedRange = NSRange(location: newCursorPos, length: 0)
}

На Android с EditText аналог через Editable.replace() + Selection.setSelection(). В Compose — через TextFieldState в новом API (доступен с Compose BOM 2024.06).

Промпты для разных сценариев рерайтинга

Универсального промпта нет. У каждого режима свой:

enum RewriteMode {
    case simplify, formalize, casual, shorten, expand, fix

    var systemPrompt: String {
        switch self {
        case .simplify:
            return "Rewrite the text using simpler words and shorter sentences. Preserve all meaning. Same language as input."
        case .formalize:
            return "Rewrite in formal business style. Remove colloquialisms. Preserve all key information."
        case .casual:
            return "Rewrite in a friendly, conversational tone. Natural language, not stiff."
        case .shorten:
            return "Shorten by 40-60%. Keep only essential information. No filler."
        case .expand:
            return "Expand with relevant details and examples. Add 50-100% more content. Stay on topic."
        case .fix:
            return "Fix grammar, spelling, and awkward phrasing. Minimal changes to preserve the original voice."
        }
    }
}

Ключевая строка во всех промптах: «Same language as input». Без неё GPT иногда переключается на английский, особенно если в тексте есть технические термины.

UI паттерн: «до/после»

Пользователь должен видеть оригинал рядом с рерайтом и легко откатиться. Не прячьте исходник.

@Composable
fun RewriteResultView(
    original: String,
    rewritten: String,
    onAccept: () -> Unit,
    onDiscard: () -> Unit,
    onRetry: () -> Unit
) {
    Column(modifier = Modifier.fillMaxWidth()) {
        Text("Оригинал", style = MaterialTheme.typography.labelSmall, color = MaterialTheme.colorScheme.onSurfaceVariant)
        Text(
            text = original,
            modifier = Modifier
                .fillMaxWidth()
                .background(MaterialTheme.colorScheme.surfaceVariant, RoundedCornerShape(8.dp))
                .padding(12.dp),
            style = MaterialTheme.typography.bodyMedium.copy(
                color = MaterialTheme.colorScheme.onSurfaceVariant
            )
        )

        Spacer(Modifier.height(8.dp))

        Text("Результат", style = MaterialTheme.typography.labelSmall)
        Text(
            text = rewritten,
            modifier = Modifier
                .fillMaxWidth()
                .background(MaterialTheme.colorScheme.primaryContainer, RoundedCornerShape(8.dp))
                .padding(12.dp)
        )

        Row(modifier = Modifier.fillMaxWidth(), horizontalArrangement = Arrangement.SpaceBetween) {
            TextButton(onClick = onDiscard) { Text("Отмена") }
            TextButton(onClick = onRetry) { Text("Ещё вариант") }
            Button(onClick = onAccept) { Text("Принять") }
        }
    }
}

Кнопка «Ещё вариант» важна — первый рерайт не всегда подходит, но и возиться с промптом пользователь не хочет.

Diff-подсветка изменений

Для режима fix (правка грамматики) полезно показать, что именно изменилось. Простой diff на клиенте без сервера:

// Упрощённый word-level diff
func computeDiff(original: String, rewritten: String) -> [DiffChunk] {
    let origWords = original.split(separator: " ").map(String.init)
    let newWords = rewritten.split(separator: " ").map(String.init)
    // LCS-based diff, реализация через стандартный алгоритм
    return lcs(origWords, newWords)
}

На Android — DiffUtil из androidx.recyclerview работает для списков, для текста нужна собственная реализация LCS или библиотека java-diff-utils.

Ориентиры по срокам

Базовый рерайтинг (один режим, без diff) — 2–4 дня. Полноценная реализация с множеством режимов, diff-подсветкой, корректным undo и историей вариантов — 1.5–2 недели.