mirror of https://github.com/M66B/FairEmail.git
M66B
8 months ago
parent
912c2dc505
commit
a3b8bd98de
@ -0,0 +1,199 @@
|
|||||||
|
package eu.faircode.email;
|
||||||
|
|
||||||
|
/*
|
||||||
|
This file is part of FairEmail.
|
||||||
|
|
||||||
|
FairEmail is free software: you can redistribute it and/or modify
|
||||||
|
it under the terms of the GNU General Public License as published by
|
||||||
|
the Free Software Foundation, either version 3 of the License, or
|
||||||
|
(at your option) any later version.
|
||||||
|
|
||||||
|
FairEmail is distributed in the hope that it will be useful,
|
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
GNU General Public License for more details.
|
||||||
|
|
||||||
|
You should have received a copy of the GNU General Public License
|
||||||
|
along with FairEmail. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
Copyright 2018-2025 by Marcel Bokhorst (M66B)
|
||||||
|
*/
|
||||||
|
|
||||||
|
import static android.app.Activity.RESULT_OK;
|
||||||
|
|
||||||
|
import android.app.Dialog;
|
||||||
|
import android.content.Context;
|
||||||
|
import android.content.DialogInterface;
|
||||||
|
import android.content.SharedPreferences;
|
||||||
|
import android.os.Bundle;
|
||||||
|
import android.text.TextUtils;
|
||||||
|
import android.util.Pair;
|
||||||
|
import android.view.LayoutInflater;
|
||||||
|
import android.view.View;
|
||||||
|
import android.view.ViewGroup;
|
||||||
|
import android.widget.AdapterView;
|
||||||
|
import android.widget.ArrayAdapter;
|
||||||
|
import android.widget.CheckBox;
|
||||||
|
import android.widget.EditText;
|
||||||
|
import android.widget.Spinner;
|
||||||
|
import android.widget.TextView;
|
||||||
|
|
||||||
|
import androidx.annotation.NonNull;
|
||||||
|
import androidx.annotation.Nullable;
|
||||||
|
import androidx.appcompat.app.AlertDialog;
|
||||||
|
import androidx.preference.PreferenceManager;
|
||||||
|
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
public class FragmentDialogAI extends FragmentDialogBase {
|
||||||
|
private Spinner spPrompt;
|
||||||
|
private EditText etPrompt;
|
||||||
|
|
||||||
|
@NonNull
|
||||||
|
@Override
|
||||||
|
public Dialog onCreateDialog(@Nullable Bundle savedInstanceState) {
|
||||||
|
final Context context = getContext();
|
||||||
|
|
||||||
|
SharedPreferences prefs = PreferenceManager.getDefaultSharedPreferences(context);
|
||||||
|
String systemPrompt = prefs.getString("openai_system", null);
|
||||||
|
|
||||||
|
Bundle args = getArguments();
|
||||||
|
boolean has_body = args.getBoolean("has_body");
|
||||||
|
boolean has_reply = args.getBoolean("has_reply");
|
||||||
|
boolean has_system = (OpenAI.isAvailable(context) && !TextUtils.isEmpty(systemPrompt));
|
||||||
|
|
||||||
|
final View view = LayoutInflater.from(context).inflate(R.layout.dialog_ai, null);
|
||||||
|
spPrompt = view.findViewById(R.id.spPrompt);
|
||||||
|
etPrompt = view.findViewById(R.id.etPrompt);
|
||||||
|
final CheckBox cbInputSystem = view.findViewById(R.id.cbInputSystem);
|
||||||
|
final CheckBox cbInputBody = view.findViewById(R.id.cbInputBody);
|
||||||
|
final CheckBox cbInputReply = view.findViewById(R.id.cbInputReply);
|
||||||
|
final ContentLoadingProgressBar pbWait = view.findViewById(R.id.pbWait);
|
||||||
|
|
||||||
|
ArrayAdapter<Pair<String, String>> adapter = new ArrayAdapter<Pair<String, String>>(context, android.R.layout.simple_spinner_item, android.R.id.text1) {
|
||||||
|
@NonNull
|
||||||
|
@Override
|
||||||
|
public View getView(int position, @Nullable View convertView, @NonNull ViewGroup parent) {
|
||||||
|
return _getView(position, super.getView(position, convertView, parent));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public View getDropDownView(int position, @Nullable View convertView, @NonNull ViewGroup parent) {
|
||||||
|
return _getView(position, super.getDropDownView(position, convertView, parent));
|
||||||
|
}
|
||||||
|
|
||||||
|
private View _getView(int position, View view) {
|
||||||
|
Pair<String, String> prompt = getItem(position);
|
||||||
|
TextView tv = view.findViewById(android.R.id.text1);
|
||||||
|
tv.setText(prompt == null ? null : prompt.first);
|
||||||
|
return view;
|
||||||
|
}
|
||||||
|
|
||||||
|
};
|
||||||
|
adapter.setDropDownViewResource(android.R.layout.simple_spinner_dropdown_item);
|
||||||
|
spPrompt.setAdapter(adapter);
|
||||||
|
|
||||||
|
spPrompt.setOnItemSelectedListener(new AdapterView.OnItemSelectedListener() {
|
||||||
|
@Override
|
||||||
|
public void onItemSelected(AdapterView<?> parent, View view, int position, long id) {
|
||||||
|
Object tag = spPrompt.getTag();
|
||||||
|
if (tag != null && tag.equals(position))
|
||||||
|
return;
|
||||||
|
spPrompt.setTag(position);
|
||||||
|
Pair<String, String> selected = adapter.getItem(position);
|
||||||
|
etPrompt.setText(selected == null ? null : selected.second);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onNothingSelected(AdapterView<?> parent) {
|
||||||
|
etPrompt.setText(null);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
cbInputSystem.setChecked(has_system);
|
||||||
|
cbInputBody.setChecked(has_body);
|
||||||
|
|
||||||
|
if (BuildConfig.DEBUG) {
|
||||||
|
cbInputSystem.setEnabled(has_system);
|
||||||
|
cbInputBody.setEnabled(has_body);
|
||||||
|
cbInputReply.setEnabled(has_reply);
|
||||||
|
} else {
|
||||||
|
cbInputSystem.setVisibility(has_system ? View.VISIBLE : View.GONE);
|
||||||
|
cbInputBody.setVisibility(has_body ? View.VISIBLE : View.GONE);
|
||||||
|
cbInputReply.setVisibility(has_reply ? View.VISIBLE : View.GONE);
|
||||||
|
}
|
||||||
|
|
||||||
|
new SimpleTask<List<Pair<String, String>>>() {
|
||||||
|
@Override
|
||||||
|
protected void onPreExecute(Bundle args) {
|
||||||
|
spPrompt.setEnabled(false);
|
||||||
|
etPrompt.setEnabled(false);
|
||||||
|
pbWait.setVisibility(View.VISIBLE);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected void onPostExecute(Bundle args) {
|
||||||
|
spPrompt.setEnabled(true);
|
||||||
|
etPrompt.setEnabled(true);
|
||||||
|
pbWait.setVisibility(View.GONE);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected List<Pair<String, String>> onExecute(Context context, Bundle args) throws Throwable {
|
||||||
|
DB db = DB.getInstance(context);
|
||||||
|
|
||||||
|
List<Pair<String, String>> prompts = new ArrayList<>();
|
||||||
|
List<EntityAnswer> answers = db.answer().getAiPrompts();
|
||||||
|
if (answers != null)
|
||||||
|
for (EntityAnswer answer : answers) {
|
||||||
|
String html = answer.getData(context, null).getHtml();
|
||||||
|
prompts.add(new Pair<>(answer.name, JsoupEx.parse(html).body().text()));
|
||||||
|
}
|
||||||
|
|
||||||
|
return prompts;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected void onExecuted(Bundle args, List<Pair<String, String>> prompts) {
|
||||||
|
prompts.add(0, new Pair<>(context.getString(R.string.title_advanced_default_prompt), AI.getDefaultPrompt(getContext())));
|
||||||
|
adapter.addAll(prompts);
|
||||||
|
|
||||||
|
if (savedInstanceState != null) {
|
||||||
|
int prompt = savedInstanceState.getInt("fair:prompt");
|
||||||
|
spPrompt.setTag(prompt);
|
||||||
|
spPrompt.setSelection(prompt);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected void onException(Bundle args, Throwable ex) {
|
||||||
|
Log.unexpectedError(getParentFragmentManager(), ex);
|
||||||
|
}
|
||||||
|
|
||||||
|
}.execute(this, new Bundle(), "ai:prompts");
|
||||||
|
|
||||||
|
AlertDialog.Builder builder = new AlertDialog.Builder(context)
|
||||||
|
.setView(view)
|
||||||
|
.setNegativeButton(android.R.string.cancel, null)
|
||||||
|
.setPositiveButton(android.R.string.ok, new DialogInterface.OnClickListener() {
|
||||||
|
@Override
|
||||||
|
public void onClick(DialogInterface dialog, int which) {
|
||||||
|
Bundle args = getArguments();
|
||||||
|
args.putString("prompt", etPrompt.getText().toString());
|
||||||
|
args.putBoolean("input_system", has_system && cbInputSystem.isChecked());
|
||||||
|
args.putBoolean("input_body", has_body && cbInputBody.isChecked());
|
||||||
|
args.putBoolean("input_reply", has_reply && cbInputReply.isChecked());
|
||||||
|
sendResult(RESULT_OK);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
return builder.create();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onSaveInstanceState(@NonNull Bundle outState) {
|
||||||
|
outState.putInt("fair:prompt", spPrompt.getSelectedItemPosition());
|
||||||
|
super.onSaveInstanceState(outState);
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,90 @@
|
|||||||
|
<?xml version="1.0" encoding="utf-8"?>
|
||||||
|
<ScrollView xmlns:android="http://schemas.android.com/apk/res/android"
|
||||||
|
xmlns:app="http://schemas.android.com/apk/res-auto"
|
||||||
|
android:layout_width="match_parent"
|
||||||
|
android:layout_height="wrap_content"
|
||||||
|
android:fadeScrollbars="false"
|
||||||
|
android:padding="24dp"
|
||||||
|
android:scrollbarStyle="outsideOverlay">
|
||||||
|
|
||||||
|
<androidx.constraintlayout.widget.ConstraintLayout
|
||||||
|
android:layout_width="match_parent"
|
||||||
|
android:layout_height="wrap_content">
|
||||||
|
|
||||||
|
<TextView
|
||||||
|
android:id="@+id/tvAI"
|
||||||
|
android:layout_width="0dp"
|
||||||
|
android:layout_height="wrap_content"
|
||||||
|
android:drawableStart="@drawable/twotone_smart_toy_24"
|
||||||
|
android:drawablePadding="6dp"
|
||||||
|
android:text="@string/title_advanced_ai"
|
||||||
|
android:textAppearance="@style/TextAppearance.AppCompat.Large"
|
||||||
|
android:textColor="?android:attr/textColorPrimary"
|
||||||
|
app:layout_constraintEnd_toEndOf="parent"
|
||||||
|
app:layout_constraintStart_toStartOf="parent"
|
||||||
|
app:layout_constraintTop_toTopOf="parent" />
|
||||||
|
|
||||||
|
<eu.faircode.email.SpinnerEx
|
||||||
|
android:id="@+id/spPrompt"
|
||||||
|
android:layout_width="0dp"
|
||||||
|
android:layout_height="wrap_content"
|
||||||
|
android:layout_marginTop="12dp"
|
||||||
|
app:layout_constraintEnd_toEndOf="parent"
|
||||||
|
app:layout_constraintStart_toStartOf="parent"
|
||||||
|
app:layout_constraintTop_toBottomOf="@id/tvAI" />
|
||||||
|
|
||||||
|
<EditText
|
||||||
|
android:id="@+id/etPrompt"
|
||||||
|
android:layout_width="0dp"
|
||||||
|
android:layout_height="wrap_content"
|
||||||
|
android:layout_marginTop="12dp"
|
||||||
|
android:inputType="textCapSentences"
|
||||||
|
android:textAppearance="@style/TextAppearance.AppCompat.Medium"
|
||||||
|
app:layout_constraintEnd_toEndOf="parent"
|
||||||
|
app:layout_constraintStart_toStartOf="parent"
|
||||||
|
app:layout_constraintTop_toBottomOf="@+id/spPrompt" />
|
||||||
|
|
||||||
|
<CheckBox
|
||||||
|
android:id="@+id/cbInputSystem"
|
||||||
|
android:layout_width="wrap_content"
|
||||||
|
android:layout_height="wrap_content"
|
||||||
|
android:layout_marginTop="12dp"
|
||||||
|
android:text="@string/title_advanced_ai_input_system"
|
||||||
|
android:textAppearance="@style/TextAppearance.AppCompat.Medium"
|
||||||
|
app:layout_constraintStart_toStartOf="parent"
|
||||||
|
app:layout_constraintTop_toBottomOf="@id/etPrompt" />
|
||||||
|
|
||||||
|
<CheckBox
|
||||||
|
android:id="@+id/cbInputBody"
|
||||||
|
android:layout_width="wrap_content"
|
||||||
|
android:layout_height="wrap_content"
|
||||||
|
android:layout_marginTop="12dp"
|
||||||
|
android:text="@string/title_advanced_ai_input_body"
|
||||||
|
android:textAppearance="@style/TextAppearance.AppCompat.Medium"
|
||||||
|
app:layout_constraintStart_toStartOf="parent"
|
||||||
|
app:layout_constraintTop_toBottomOf="@id/cbInputSystem" />
|
||||||
|
|
||||||
|
<CheckBox
|
||||||
|
android:id="@+id/cbInputReply"
|
||||||
|
android:layout_width="wrap_content"
|
||||||
|
android:layout_height="wrap_content"
|
||||||
|
android:layout_marginTop="12dp"
|
||||||
|
android:text="@string/title_advanced_ai_input_reply"
|
||||||
|
android:textAppearance="@style/TextAppearance.AppCompat.Medium"
|
||||||
|
app:layout_constraintStart_toStartOf="parent"
|
||||||
|
app:layout_constraintTop_toBottomOf="@id/cbInputBody" />
|
||||||
|
|
||||||
|
<eu.faircode.email.ContentLoadingProgressBar
|
||||||
|
android:id="@+id/pbWait"
|
||||||
|
style="@style/Base.Widget.AppCompat.ProgressBar"
|
||||||
|
android:layout_width="wrap_content"
|
||||||
|
android:layout_height="wrap_content"
|
||||||
|
android:layout_marginStart="12dp"
|
||||||
|
android:indeterminate="true"
|
||||||
|
app:layout_constraintBottom_toBottomOf="parent"
|
||||||
|
app:layout_constraintEnd_toEndOf="parent"
|
||||||
|
app:layout_constraintStart_toStartOf="parent"
|
||||||
|
app:layout_constraintTop_toTopOf="parent"
|
||||||
|
app:show_delay="0" />
|
||||||
|
</androidx.constraintlayout.widget.ConstraintLayout>
|
||||||
|
</ScrollView>
|
||||||
Loading…
Reference in new issue