Flutter智能回复插件smart_reply的使用

Flutter智能回复插件smart_reply的使用

简介

smart_reply 是一个基于 Google ML Kit 的 Flutter 插件，用于生成与消息相关的智能回复建议。该插件使用设备上的模型生成回复，因此不会将消息发送到远程服务器，确保了用户隐私。

注意：这是一个非官方插件。

使用方法

下面是一个完整的示例代码，展示了如何在 Flutter 应用中使用 smart_reply 插件。

import 'dart:async';

import 'package:flutter/material.dart';
import 'package:smart_reply/smart_reply.dart';

void main() {
  runApp(MyApp());
}

class MyApp extends StatefulWidget {
  @override
  _MyAppState createState() => _MyAppState();
}

class _MyAppState extends State<MyApp> {
  final _textController = TextEditingController();
  final _smartReply = SmartReply();
  List<String> _suggestedReplies = [];
  List<TextMessage> _messages = [];
  bool _isLocalUser = true;

  @override
  void initState() {
    super.initState();
  }

  // 异步获取智能回复建议
  Future<void> getSuggestedReplies() async {
    try {
      List<String> replies = await _smartReply.suggestReplies(_messages);
      setState(() {
        _suggestedReplies = replies;
      });
    } catch (e) {
      print('Error getting suggestions: $e');
    }
  }

  // 发送消息并更新消息列表
  void _sendMessage(String message) {
    setState(() {
      _messages.add(TextMessage(
        text: message,
        timestamp: DateTime.now(),
        userId: _isLocalUser ? 'a' : 'b',
        isLocalUser: _isLocalUser,
      ));
      _isLocalUser = !_isLocalUser;
      _textController.clear();
    });

    getSuggestedReplies();
  }

  @override
  Widget build(BuildContext context) {
    return MaterialApp(
      home: GestureDetector(
        onTap: () => FocusScope.of(context).unfocus(),
        child: Scaffold(
          appBar: AppBar(
            title: const Text('Smart reply example'),
          ),
          body: SafeArea(
            child: Padding(
              padding: const EdgeInsets.all(12),
              child: Column(
                children: [
                  Expanded(
                    child: ListView.builder(
                      itemCount: _messages.length,
                      reverse: true,
                      itemBuilder: (context, i) {
                        final message = _messages[_messages.length - 1 - i];
                        return Align(
                          alignment: message.isLocalUser
                              ? Alignment.centerRight
                              : Alignment.centerLeft,
                          child: Container(
                            padding: EdgeInsets.all(10),
                            decoration: BoxDecoration(
                              borderRadius: BorderRadius.circular(10),
                              color: message.isLocalUser
                                  ? Colors.grey
                                  : Colors.blue,
                            ),
                            child: Text(
                              message.text,
                              style: TextStyle(fontSize: 14, color: Colors.white),
                            ),
                          ),
                        );
                      },
                    ),
                  ),
                  TextField(
                    controller: _textController,
                    onSubmitted: _sendMessage,
                    textInputAction: TextInputAction.send,
                    decoration: InputDecoration(
                      labelText: _isLocalUser ? 'To remote' : 'To local',
                    ),
                  ),
                  Container(
                    margin: EdgeInsets.only(top: 24),
                    child: Wrap(
                      runAlignment: WrapAlignment.center,
                      alignment: WrapAlignment.spaceEvenly,
                      children: [
                        for (var s in _suggestedReplies)
                          Padding(
                            padding: const EdgeInsets.symmetric(horizontal: 6),
                            child: OutlinedButton(
                              child: Text(s),
                              onPressed: () => _sendMessage(s),
                            ),
                          ),
                      ],
                    ),
                  )
                ],
              ),
            ),
          ),
        ),
      ),
    );
  }
}

Android 设置

在 android/app/build.gradle 文件中，禁用对 tflite 文件的压缩，以避免 getSuggestions 抛出 PlatformException：

android {
    // ...
    aaptOptions {
        noCompress "tflite"
    }
}