Method getSummaryText

app/src/main/java/eu/faircode/email/AI.java:154–245

(Context context, EntityMessage message, long template, String prompt)

    }

    static Spanned getSummaryText(Context context, EntityMessage message, long template, String prompt) throws JSONException, IOException {
        File file = message.getFile(context);
        if (!file.exists())
            return null;

        Document d = JsoupEx.parse(file);

        SharedPreferences prefs = PreferenceManager.getDefaultSharedPreferences(context);
        boolean remove_signatures = prefs.getBoolean("remove_signatures", false);
        if (remove_signatures)
            HtmlHelper.removeSignatures(d);

        d = HtmlHelper.sanitizeView(context, d, false);

        HtmlHelper.truncate(d, MAX_SUMMARIZE_TEXT_SIZE);

        String body = d.body().text().trim();
        if (TextUtils.isEmpty(body))
            return null;

        String templatePrompt = prompt;
        if (templatePrompt == null && template > 0L) {
            DB db = DB.getInstance(context);
            EntityAnswer t = db.answer().getAnswer(template);
            if (t != null) {
                String html = t.getData(context, null).getHtml();
                templatePrompt = JsoupEx.parse(html).body().text();
            }
        }

        StringBuilder sb = new StringBuilder();
        if (OpenAI.isAvailable(context)) {
            String model = prefs.getString("openai_model", OpenAI.DEFAULT_MODEL);
            float temperature = prefs.getFloat("openai_temperature", OpenAI.DEFAULT_TEMPERATURE);
            String defaultPrompt = prefs.getString("openai_summarize", OpenAI.DEFAULT_SUMMARY_PROMPT);
            boolean multimodal = prefs.getBoolean("openai_multimodal", false);

            List<OpenAI.Message> input = new ArrayList<>();

            if (multimodal) {
                input.add(new OpenAI.Message(OpenAI.USER,
                        new OpenAI.Content[]{new OpenAI.Content(OpenAI.CONTENT_TEXT,
                                templatePrompt == null ? defaultPrompt : templatePrompt)}));

                SpannableStringBuilder ssb = HtmlHelper.fromDocument(context, d, null, null);
                input.add(new OpenAI.Message(OpenAI.USER,
                        OpenAI.Content.get(ssb, message.id, context)));
            } else {
                List<String> contents = new ArrayList<>();
                contents.add(templatePrompt == null ? defaultPrompt : templatePrompt);
                contents.add(body);
                input.add(new OpenAI.Message(OpenAI.USER, new OpenAI.Content[]{
                        new OpenAI.Content(OpenAI.CONTENT_TEXT, TextUtils.join("\n", contents))}));
            }

            OpenAI.Message[] completions =
                    OpenAI.completeChat(context, model, input.toArray(new OpenAI.Message[0]), temperature, 1);