Реализация AI-ассистента для написания текстов (Writing Assistant) в мобильном приложении

Реализация AI-ассистента для написания текстов (Writing Assistant) в мобильном приложении

Writing Assistant в мобильном приложении — это не просто поле ввода с кнопкой «улучшить». Это редактор, который работает в контексте: понимает тип документа (письмо, пост, отчёт), поддерживает стриминг, не сбрасывает курсор при вставке, и переживает фоновый режим без потери состояния.

Именно на этих деталях ломаются 80% реализаций.

Архитектура редактора с AI

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

// iOS: вставка AI-текста через NSTextStorage без сброса курсора
func insertAIText(_ text: String, at range: NSRange) {
    guard let textView = self.textView else { return }
    let storage = textView.textStorage

    // Сохраняем позицию курсора
    let cursorOffset = textView.selectedRange.location

    storage.beginEditing()
    storage.replaceCharacters(
        in: range,
        with: NSAttributedString(string: text, attributes: defaultTypingAttributes)
    )
    storage.endEditing()

    // Восстанавливаем курсор после вставки
    let newOffset = cursorOffset + (text.count - range.length)
    textView.selectedRange = NSRange(location: max(0, newOffset), length: 0)
}

На Android с EditText аналог через Editable.replace() + сохранение SelectionStart/SelectionEnd через android.text.Selection.

Стриминг и прогрессивная вставка

Writing Assistant обязан стримить текст — пользователь видит, как AI печатает. Технически это AsyncStream<String> (iOS) или Flow<String> (Android), каждый чанк добавляется в конец активного параграфа.

Типичная проблема: при быстром стриминге (> 20 символов/сек) UITextView начинает лагать на длинных текстах. Причина — textStorage триггерит layout pass на каждое изменение. Решение — батчинг апдейтов:

private var streamBuffer = ""
private var streamTimer: Timer?

func appendStreamChunk(_ chunk: String) {
    streamBuffer += chunk

    if streamTimer == nil {
        streamTimer = Timer.scheduledTimer(withTimeInterval: 0.05, repeats: false) { [weak self] _ in
            guard let self else { return }
            self.textView.textStorage.beginEditing()
            self.textView.textStorage.append(NSAttributedString(string: self.streamBuffer))
            self.textView.textStorage.endEditing()
            self.streamBuffer = ""
            self.streamTimer = nil
        }
    }
}

Каждые 50 мс — один layout pass вместо 20. На iPhone SE 2nd gen разница видна невооружённым глазом.

Режимы ассистента: как не захламить UI

Writing Assistant обычно предлагает несколько действий: продолжить текст, переписать выделенное, изменить тон, сократить, расширить. Если показывать все кнопки сразу — UI превращается в хаос.

Правильная схема: контекстное меню появляется только при наличии выделения (для «переписать», «изменить тон»), floating action button появляется в конце параграфа (для «продолжить»). Два разных триггера — два разных UX-паттерна.

// Android Compose - floating assistant button
@Composable
fun WritingAssistantOverlay(
    textFieldState: TextFieldState,
    onContinue: () -> Unit,
    onRewrite: (String) -> Unit
) {
    val hasSelection = textFieldState.selection.length > 0

    AnimatedVisibility(visible = !hasSelection) {
        FloatingActionButton(
            onClick = onContinue,
            modifier = Modifier.align(Alignment.BottomEnd)
        ) {
            Icon(Icons.Default.AutoAwesome, "Продолжить")
        }
    }

    AnimatedVisibility(visible = hasSelection) {
        ContextualMenu(
            items = listOf("Переписать", "Изменить тон", "Сократить"),
            onSelect = { action ->
                val selected = textFieldState.text.substring(textFieldState.selection)
                onRewrite("$action: $selected")
            }
        )
    }
}

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

Каждое действие ассистента — отдельный промпт. Их нельзя обобщить в один универсальный. Несколько рабочих шаблонов:

Продолжение текста:

Continue the following text naturally, maintaining the same style, language, and tone.
Write 1-3 sentences only. Do not repeat what was already written.
Text: {last_500_chars}

Переписать выделенное:

Rewrite the following text. Keep the core meaning but improve clarity and flow.
Language: {detected_language}. Style: {business|casual|formal}.
Text: {selected_text}

Изменить тон:

Rewrite this text in a {formal|casual|empathetic|assertive} tone.
Preserve all key information. Output only the rewritten text.
Text: {selected_text}

Определение языка — через NLLanguageRecognizer (iOS) или TextClassifier из ML Kit (Android). Не полагайтесь на Locale.current — пользователь может писать на другом языке.

Состояние при фоновом режиме

Если пользователь свернул приложение в процессе генерации, iOS отправит задачу в URLSession с background конфигурацией или просто отменит запрос. Нужно сохранить промпт и статус в UserDefaults/SharedPreferences и восстановить при возврате.

При долгих генерациях (> 15 секунд для больших текстов) переходим на Background Tasks API на iOS или WorkManager на Android — стриминг в фоне невозможен, но можно получить итоговый результат через push notification.

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

Базовый ассистент с кнопкой «улучшить» через OpenAI — 3–5 дней. Полноценный редактор со стримингом, контекстным меню, режимами и сохранением состояния — 3–4 недели. Поддержка офлайн-режима с on-device моделью (через CoreML/TFLite) — отдельно от 2 недель.