Styled summary

master
M66B 3 weeks ago
parent 4ac3307a5b
commit 58fa1285fe

@ -165,7 +165,7 @@ public class AI {
return context.getString(R.string.title_summarize); return context.getString(R.string.title_summarize);
} }
static String getSummaryText(Context context, EntityMessage message) throws JSONException, IOException { static Spanned getSummaryText(Context context, EntityMessage message) throws JSONException, IOException {
File file = message.getFile(context); File file = message.getFile(context);
if (!file.exists()) if (!file.exists())
return null; return null;
@ -183,6 +183,7 @@ public class AI {
HtmlHelper.truncate(d, MAX_SUMMARIZE_TEXT_SIZE); HtmlHelper.truncate(d, MAX_SUMMARIZE_TEXT_SIZE);
StringBuilder sb = new StringBuilder();
if (OpenAI.isAvailable(context)) { if (OpenAI.isAvailable(context)) {
String model = prefs.getString("openai_model", OpenAI.DEFAULT_MODEL); String model = prefs.getString("openai_model", OpenAI.DEFAULT_MODEL);
float temperature = prefs.getFloat("openai_temperature", OpenAI.DEFAULT_TEMPERATURE); float temperature = prefs.getFloat("openai_temperature", OpenAI.DEFAULT_TEMPERATURE);
@ -208,7 +209,6 @@ public class AI {
OpenAI.Message[] completions = OpenAI.Message[] completions =
OpenAI.completeChat(context, model, input.toArray(new OpenAI.Message[0]), temperature, 1); OpenAI.completeChat(context, model, input.toArray(new OpenAI.Message[0]), temperature, 1);
StringBuilder sb = new StringBuilder();
for (OpenAI.Message completion : completions) for (OpenAI.Message completion : completions)
for (OpenAI.Content content : completion.getContent()) for (OpenAI.Content content : completion.getContent())
if (OpenAI.CONTENT_TEXT.equals(content.getType())) { if (OpenAI.CONTENT_TEXT.equals(content.getType())) {
@ -216,7 +216,6 @@ public class AI {
sb.append('\n'); sb.append('\n');
sb.append(content.getContent()); sb.append(content.getContent());
} }
return sb.toString();
} else if (Gemini.isAvailable(context)) { } else if (Gemini.isAvailable(context)) {
String model = prefs.getString("gemini_model", Gemini.DEFAULT_MODEL); String model = prefs.getString("gemini_model", Gemini.DEFAULT_MODEL);
float temperature = prefs.getFloat("gemini_temperature", Gemini.DEFAULT_TEMPERATURE); float temperature = prefs.getFloat("gemini_temperature", Gemini.DEFAULT_TEMPERATURE);
@ -235,15 +234,17 @@ public class AI {
Gemini.Message[] completions = Gemini.Message[] completions =
Gemini.generate(context, model, new Gemini.Message[]{content}, temperature, 1); Gemini.generate(context, model, new Gemini.Message[]{content}, temperature, 1);
StringBuilder sb = new StringBuilder();
for (Gemini.Message completion : completions) for (Gemini.Message completion : completions)
for (String result : completion.getContent()) { for (String result : completion.getContent()) {
if (sb.length() != 0) if (sb.length() != 0)
sb.append('\n'); sb.append('\n');
sb.append(result); sb.append(result);
} }
return sb.toString();
} else } else
throw new IllegalArgumentException("No AI available"); throw new IllegalArgumentException("No AI available");
String html = Markdown.toHtml(sb.toString());
Document doc = HtmlHelper.sanitizeCompose(context, html, false);
return HtmlHelper.fromDocument(context, doc, null, null);
} }
} }

@ -1572,7 +1572,7 @@ public class EntityRule {
} }
try { try {
message.preview = AI.getSummaryText(context, message); message.preview = AI.getSummaryText(context, message).toString();
} catch (Throwable ex) { } catch (Throwable ex) {
message.error = Log.formatThrowable(ex); message.error = Log.formatThrowable(ex);
db.message().setMessageError(message.id, message.error); db.message().setMessageError(message.id, message.error);

@ -26,6 +26,7 @@ import android.content.Context;
import android.content.SharedPreferences; import android.content.SharedPreferences;
import android.os.Build; import android.os.Build;
import android.os.Bundle; import android.os.Bundle;
import android.text.Spanned;
import android.util.TypedValue; import android.util.TypedValue;
import android.view.LayoutInflater; import android.view.LayoutInflater;
import android.view.View; import android.view.View;
@ -87,7 +88,7 @@ public class FragmentDialogSummarize extends FragmentDialogBase {
tvFrom.setText(args.getString("from")); tvFrom.setText(args.getString("from"));
tvSubject.setText(args.getString("subject")); tvSubject.setText(args.getString("subject"));
new SimpleTask<String>() { new SimpleTask<Spanned>() {
@Override @Override
protected void onPreExecute(Bundle args) { protected void onPreExecute(Bundle args) {
tvSummary.setVisibility(View.GONE); tvSummary.setVisibility(View.GONE);
@ -103,7 +104,7 @@ public class FragmentDialogSummarize extends FragmentDialogBase {
} }
@Override @Override
protected String onExecute(Context context, Bundle args) throws Throwable { protected Spanned onExecute(Context context, Bundle args) throws Throwable {
long id = args.getLong("id"); long id = args.getLong("id");
DB db = DB.getInstance(context); DB db = DB.getInstance(context);
@ -112,14 +113,14 @@ public class FragmentDialogSummarize extends FragmentDialogBase {
return null; return null;
long start = new Date().getTime(); long start = new Date().getTime();
String summary = AI.getSummaryText(context, message); Spanned summary = AI.getSummaryText(context, message);
args.putLong("elapsed", new Date().getTime() - start); args.putLong("elapsed", new Date().getTime() - start);
return summary; return summary;
} }
@Override @Override
protected void onExecuted(Bundle args, String summary) { protected void onExecuted(Bundle args, Spanned summary) {
tvSummary.setText(summary); tvSummary.setText(summary);
tvSummary.setVisibility(View.VISIBLE); tvSummary.setVisibility(View.VISIBLE);
ibCopy.setVisibility(View.VISIBLE); ibCopy.setVisibility(View.VISIBLE);

Loading…
Cancel
Save