feat: support prompt and suffix

LucasXu0 · LucasXu0 · commit fc1efeb70bde · 2023-01-10T10:14:52.000+08:00
diff --git a/frontend/app_flowy/packages/appflowy_editor/example/lib/home_page.dart b/frontend/app_flowy/packages/appflowy_editor/example/lib/home_page.dart
@@ -4,7 +4,7 @@ import 'dart:io';
 
 import 'package:appflowy_editor/appflowy_editor.dart';
 import 'package:example/pages/simple_editor.dart';
-import 'package:example/plugin/text_robot.dart';
+import 'package:example/plugin/AI/text_robot.dart';
 import 'package:file_picker/file_picker.dart';
 import 'package:flutter/foundation.dart';
 import 'package:flutter/material.dart';
diff --git a/frontend/app_flowy/packages/appflowy_editor/example/lib/pages/simple_editor.dart b/frontend/app_flowy/packages/appflowy_editor/example/lib/pages/simple_editor.dart
@@ -2,7 +2,9 @@ import 'dart:convert';
 
 import 'package:appflowy_editor/appflowy_editor.dart';
 import 'package:appflowy_editor_plugins/appflowy_editor_plugins.dart';
-import 'package:example/plugin/text_robot.dart';
+import 'package:example/plugin/AI/continue_to_write.dart';
+import 'package:example/plugin/AI/auto_completion.dart';
+import 'package:example/plugin/AI/getgpt3completions.dart';
 import 'package:flutter/material.dart';
 
 class SimpleEditor extends StatelessWidget {
@@ -65,8 +67,11 @@ class SimpleEditor extends StatelessWidget {
               codeBlockMenuItem,
               // Emoji
               emojiMenuItem,
-              // Text Robot
-              textRobotMenuItem,
+              // Open AI
+              if (apiKey.isNotEmpty) ...[
+                autoCompletionMenuItem,
+                continueToWriteMenuItem,
+              ]
             ],
           );
         } else {
diff --git a/frontend/app_flowy/packages/appflowy_editor/example/lib/plugin/AI/auto_completion.dart b/frontend/app_flowy/packages/appflowy_editor/example/lib/plugin/AI/auto_completion.dart
@@ -1,18 +1,19 @@
 import 'package:appflowy_editor/appflowy_editor.dart';
 import 'package:example/plugin/AI/getgpt3completions.dart';
+import 'package:example/plugin/AI/text_robot.dart';
 import 'package:flutter/material.dart';
 import 'package:flutter/services.dart';
 
-SelectionMenuItem textRobotMenuItem = SelectionMenuItem(
-  name: () => 'Open AI',
+SelectionMenuItem autoCompletionMenuItem = SelectionMenuItem(
+  name: () => 'Auto generate content',
   icon: (editorState, onSelected) => Icon(
     Icons.rocket,
     size: 18.0,
     color: onSelected
         ? editorState.editorStyle.selectionMenuItemSelectedIconColor
         : editorState.editorStyle.selectionMenuItemIconColor,
   ),
-  keywords: ['open ai', 'gpt3', 'ai'],
+  keywords: ['auto generate content', 'open ai', 'gpt3', 'ai'],
   handler: ((editorState, menuService, context) async {
     showDialog(
       context: context,
@@ -35,11 +36,11 @@ SelectionMenuItem textRobotMenuItem = SelectionMenuItem(
               if (key.logicalKey == LogicalKeyboardKey.enter) {
                 Navigator.of(context).pop();
                 // fetch the result and insert it
-                // Please fill in your own API key
-                getGPT3Completion('', controller.text, '', 200, .3,
-                    (result) async {
-                  await editorState.insertTextAtCurrentSelection(
+                final textRobot = TextRobot(editorState: editorState);
+                getGPT3Completion(apiKey, controller.text, '', (result) async {
+                  await textRobot.insertText(
                     result,
+                    inputType: TextRobotInputType.character,
                   );
                 });
               } else if (key.logicalKey == LogicalKeyboardKey.escape) {
@@ -52,44 +53,3 @@ SelectionMenuItem textRobotMenuItem = SelectionMenuItem(
     );
   }),
 );
-
-enum TextRobotInputType {
-  character,
-  word,
-}
-
-class TextRobot {
-  const TextRobot({
-    required this.editorState,
-    this.delay = const Duration(milliseconds: 30),
-  });
-
-  final EditorState editorState;
-  final Duration delay;
-
-  Future<void> insertText(
-    String text, {
-    TextRobotInputType inputType = TextRobotInputType.character,
-  }) async {
-    final lines = text.split('\n');
-    for (final line in lines) {
-      switch (inputType) {
-        case TextRobotInputType.character:
-          final iterator = line.runes.iterator;
-          while (iterator.moveNext()) {
-            await editorState.insertTextAtCurrentSelection(
-              iterator.currentAsString,
-            );
-            await Future.delayed(delay);
-          }
-          break;
-        default:
-      }
-
-      // insert new line
-      if (lines.length > 1) {
-        await editorState.insertNewLine(editorState);
-      }
-    }
-  }
-}
diff --git a/frontend/app_flowy/packages/appflowy_editor/example/lib/plugin/AI/continue_to_write.dart b/frontend/app_flowy/packages/appflowy_editor/example/lib/plugin/AI/continue_to_write.dart
@@ -0,0 +1,50 @@
+import 'package:appflowy_editor/appflowy_editor.dart';
+import 'package:example/plugin/AI/getgpt3completions.dart';
+import 'package:example/plugin/AI/text_robot.dart';
+import 'package:flutter/material.dart';
+
+SelectionMenuItem continueToWriteMenuItem = SelectionMenuItem(
+  name: () => 'Continue To Write',
+  icon: (editorState, onSelected) => Icon(
+    Icons.print,
+    size: 18.0,
+    color: onSelected
+        ? editorState.editorStyle.selectionMenuItemSelectedIconColor
+        : editorState.editorStyle.selectionMenuItemIconColor,
+  ),
+  keywords: ['continue to write'],
+  handler: ((editorState, menuService, context) async {
+    // get the current text
+    final selection =
+        editorState.service.selectionService.currentSelection.value;
+    final textNodes = editorState.service.selectionService.currentSelectedNodes;
+    if (selection == null || !selection.isCollapsed || textNodes.length != 1) {
+      return;
+    }
+    final textNode = textNodes.first as TextNode;
+    final prompt = textNode.delta.slice(0, selection.startIndex).toPlainText();
+    final suffix = textNode.delta
+        .slice(
+          selection.endIndex,
+          textNode.toPlainText().length,
+        )
+        .toPlainText();
+    debugPrint('AI: prompt = $prompt, suffix = $suffix');
+    final textRobot = TextRobot(editorState: editorState);
+    getGPT3Completion(
+      apiKey,
+      prompt,
+      suffix,
+      (result) async {
+        if (result == '\\n') {
+          await editorState.insertNewLineAtCurrentSelection();
+        } else {
+          await textRobot.insertText(
+            result,
+            inputType: TextRobotInputType.word,
+          );
+        }
+      },
+    );
+  }),
+);
diff --git a/frontend/app_flowy/packages/appflowy_editor/example/lib/plugin/AI/getgpt3completions.dart b/frontend/app_flowy/packages/appflowy_editor/example/lib/plugin/AI/getgpt3completions.dart
@@ -2,14 +2,19 @@ import 'package:http/http.dart' as http;
 import 'dart:async';
 import 'dart:convert';
 
+// Please fill in your own API key
+const apiKey = '';
+
 Future<void> getGPT3Completion(
   String apiKey,
   String prompt,
   String suffix,
-  int maxTokens,
-  double temperature,
-  Function(String) onData, // callback function to handle streaming data
-) async {
+  Future<void> Function(String)
+      onData, // callback function to handle streaming data
+  {
+  int maxTokens = 200,
+  double temperature = .3,
+}) async {
   final data = {
     'prompt': prompt,
     'suffix': suffix,
@@ -43,7 +48,6 @@ Future<void> getGPT3Completion(
       }
 
       final processedText = text
-          .replaceAll('\\n', '\n')
           .replaceAll('\\r', '\r')
           .replaceAll('\\t', '\t')
           .replaceAll('\\b', '\b')
@@ -62,7 +66,7 @@ Future<void> getGPT3Completion(
           .replaceAll('\\8', '8')
           .replaceAll('\\9', '9');
 
-      onData(processedText);
+      await onData(processedText);
     }
   }
 }
diff --git a/frontend/app_flowy/packages/appflowy_editor/example/lib/plugin/AI/text_robot.dart b/frontend/app_flowy/packages/appflowy_editor/example/lib/plugin/AI/text_robot.dart
@@ -0,0 +1,48 @@
+import 'package:appflowy_editor/appflowy_editor.dart';
+
+enum TextRobotInputType {
+  character,
+  word,
+}
+
+class TextRobot {
+  const TextRobot({
+    required this.editorState,
+    this.delay = const Duration(milliseconds: 30),
+  });
+
+  final EditorState editorState;
+  final Duration delay;
+
+  Future<void> insertText(
+    String text, {
+    TextRobotInputType inputType = TextRobotInputType.character,
+  }) async {
+    final lines = text.split('\\n');
+    for (final line in lines) {
+      if (line.isEmpty) continue;
+      switch (inputType) {
+        case TextRobotInputType.character:
+          final iterator = line.runes.iterator;
+          while (iterator.moveNext()) {
+            await editorState.insertTextAtCurrentSelection(
+              iterator.currentAsString,
+            );
+            await Future.delayed(delay, () {});
+          }
+          break;
+        case TextRobotInputType.word:
+          await editorState.insertTextAtCurrentSelection(
+            line,
+          );
+          await Future.delayed(delay, () {});
+          break;
+      }
+
+      // insert new line
+      if (lines.length > 1) {
+        await editorState.insertNewLineAtCurrentSelection();
+      }
+    }
+  }
+}
diff --git a/frontend/app_flowy/packages/appflowy_editor/lib/src/commands/text/text_commands.dart b/frontend/app_flowy/packages/appflowy_editor/lib/src/commands/text/text_commands.dart
diff --git a/frontend/app_flowy/packages/appflowy_editor/lib/src/core/transform/transaction.dart b/frontend/app_flowy/packages/appflowy_editor/lib/src/core/transform/transaction.dart
diff --git a/frontend/app_flowy/packages/appflowy_editor/lib/src/editor_state.dart b/frontend/app_flowy/packages/appflowy_editor/lib/src/editor_state.dart
