mirror/FairEmail
1
0
Fork 0
mirror of https://github.com/M66B/FairEmail.git synced 2025-03-15 00:21:10 +00:00
Code Issues Releases Wiki Activity

Added AI template selection

Browse source
This commit is contained in:
M66B 2024-05-28 23:04:37 +02:00
parent 74dc7fa801
commit c21b0b615b
2 changed files with 63 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
app/src/main/java/eu/faircode/email/AI.java
View file

@ -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}));
}

56
app/src/main/java/eu/faircode/email/FragmentCompose.java
View file

@ -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) {