Added AI template selection

app/src/main/java/eu/faircode/email/AI.java

@@ -43,7 +43,7 @@ public class AI {
         return (OpenAI.isAvailable(context) || Gemini.isAvailable(context));
     }
 
-    static String completeChat(Context context, long id, CharSequence body) throws JSONException, IOException {
+    static String completeChat(Context context, long id, CharSequence body, long template) throws JSONException, IOException {
         String reply = null;
         if (body == null || TextUtils.isEmpty(body.toString().trim())) {
             body = "?";
@@ -66,6 +66,14 @@ public class AI {
             }
         }
 
+        String prompt = null;
+        DB db = DB.getInstance(context);
+        EntityAnswer t = db.answer().getAnswer(template);
+        if (t != null) {
+            String html = t.getHtml(context, null);
+            prompt = JsoupEx.parse(html).body().text();
+        }
+
         SharedPreferences prefs = PreferenceManager.getDefaultSharedPreferences(context);
         if (OpenAI.isAvailable(context)) {
             String model = prefs.getString("openai_model", OpenAI.DEFAULT_MODEL);
@@ -84,7 +92,7 @@ public class AI {
                             new OpenAI.Content(OpenAI.CONTENT_TEXT, body.toString())}));
             } else {
                 messages.add(new OpenAI.Message(OpenAI.USER, new OpenAI.Content[]{
-                        new OpenAI.Content(OpenAI.CONTENT_TEXT, answer)}));
+                        new OpenAI.Content(OpenAI.CONTENT_TEXT, prompt == null ? answer : prompt)}));
                 messages.add(new OpenAI.Message(OpenAI.USER, new OpenAI.Content[]{
                         new OpenAI.Content(OpenAI.CONTENT_TEXT, reply)}));
             }
@@ -114,7 +122,7 @@ public class AI {
                 messages.add(new Gemini.Message(Gemini.USER,
                         new String[]{Gemini.truncateParagraphs(body.toString())}));
             else {
-                messages.add(new Gemini.Message(Gemini.USER, new String[]{answer}));
+                messages.add(new Gemini.Message(Gemini.USER, new String[]{prompt == null ? answer : prompt}));
                 messages.add(new Gemini.Message(Gemini.USER, new String[]{reply}));
             }
 

app/src/main/java/eu/faircode/email/FragmentCompose.java

@@ -1884,7 +1884,7 @@ public class FragmentCompose extends FragmentBase {
             @Override
             public void onClick(View view) {
                 if (AI.isAvailable(context))
-                    onAI();
+                    onAI(view);
             }
         });
         menu.findItem(R.id.menu_ai).setActionView(ibAI);
@@ -2639,7 +2639,53 @@ public class FragmentCompose extends FragmentBase {
         }.serial().execute(this, args, "compose:print");
     }
 
-    private void onAI() {
+    private void onAI(View anchor) {
+        new SimpleTask<List<EntityAnswer>>() {
+            @Override
+            protected List<EntityAnswer> onExecute(Context context, Bundle args) throws Throwable {
+                DB db = DB.getInstance(context);
+                return db.answer().getAiPrompts();
+            }
+
+            @Override
+            protected void onExecuted(Bundle args, List<EntityAnswer> prompts) {
+                if (prompts == null || prompts.isEmpty())
+                    _onAi(null);
+                else {
+                    PopupMenuLifecycle popupMenu = new PopupMenuLifecycle(getContext(), getViewLifecycleOwner(), anchor);
+
+                    popupMenu.getMenu()
+                            .add(Menu.NONE, 1, 1, R.string.title_answer_standard)
+                            .setIntent(new Intent().putExtra("id", -1L));
+
+                    for (int i = 0; i < prompts.size(); i++) {
+                        EntityAnswer prompt = prompts.get(i);
+                        popupMenu.getMenu()
+                                .add(Menu.NONE, i + 2, i + 2, prompt.name)
+                                .setIntent(new Intent().putExtra("id", prompt.id));
+                    }
+
+                    popupMenu.setOnMenuItemClickListener(new PopupMenu.OnMenuItemClickListener() {
+                        @Override
+                        public boolean onMenuItemClick(MenuItem item) {
+                            long id = item.getIntent().getLongExtra("id", -1L);
+                            _onAi(id);
+                            return true;
+                        }
+                    });
+
+                    popupMenu.show();
+                }
+            }
+
+            @Override
+            protected void onException(Bundle args, Throwable ex) {
+                Log.unexpectedError(getParentFragmentManager(), ex);
+            }
+        }.execute(this, new Bundle(), "AI:template");
+    }
+
+    private void _onAi(Long template) {
         int start = etBody.getSelectionStart();
         int end = etBody.getSelectionEnd();
         boolean selection = (start >= 0 && end > start);
@@ -2649,6 +2695,7 @@ public class FragmentCompose extends FragmentBase {
         Bundle args = new Bundle();
         args.putLong("id", working);
         args.putCharSequence("body", body);
+        args.putLong("template", template == null ? -1L : template);
 
         new SimpleTask<String>() {
             @Override
@@ -2667,8 +2714,9 @@ public class FragmentCompose extends FragmentBase {
             protected String onExecute(Context context, Bundle args) throws Throwable {
                 long id = args.getLong("id");
                 CharSequence body = args.getCharSequence("body");
+                long template = args.getLong("template");
 
-                return AI.completeChat(context, id, body);
+                return AI.completeChat(context, id, body, template);
             }
 
             @Override
@@ -2708,7 +2756,7 @@ public class FragmentCompose extends FragmentBase {
             protected void onException(Bundle args, Throwable ex) {
                 Log.unexpectedError(getParentFragmentManager(), ex, !(ex instanceof IOException));
             }
-        }.serial().execute(this, args, "AI");
+        }.serial().execute(this, args, "AI:run");
     }
 
     private void onTranslate(View anchor) {