mirror of https://github.com/M66B/FairEmail.git
parent
d77057a7a7
commit
ab072056be
@ -0,0 +1,165 @@
|
||||
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-2024 by Marcel Bokhorst (M66B)
|
||||
*/
|
||||
|
||||
|
||||
import android.content.Context;
|
||||
import android.content.SharedPreferences;
|
||||
import android.net.Uri;
|
||||
import android.text.TextUtils;
|
||||
|
||||
import androidx.annotation.NonNull;
|
||||
import androidx.preference.PreferenceManager;
|
||||
|
||||
import org.json.JSONArray;
|
||||
import org.json.JSONException;
|
||||
import org.json.JSONObject;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.io.InputStream;
|
||||
import java.net.URL;
|
||||
import java.util.Date;
|
||||
import java.util.Objects;
|
||||
|
||||
import javax.net.ssl.HttpsURLConnection;
|
||||
|
||||
public class Gemini {
|
||||
private static final int MAX_GEMINI_LEN = 1000; // characters
|
||||
private static final int TIMEOUT = 20; // seconds
|
||||
|
||||
static boolean isAvailable(Context context) {
|
||||
if (TextUtils.isEmpty(BuildConfig.GEMINI_ENDPOINT))
|
||||
return false;
|
||||
|
||||
SharedPreferences prefs = PreferenceManager.getDefaultSharedPreferences(context);
|
||||
boolean enabled = prefs.getBoolean("gemini_enabled", false);
|
||||
String apikey = prefs.getString("gemini_apikey", null);
|
||||
|
||||
return (enabled &&
|
||||
(!TextUtils.isEmpty(apikey) || !Objects.equals(getUri(context), BuildConfig.GEMINI_ENDPOINT)));
|
||||
}
|
||||
|
||||
static String[] generate(Context context, String model, String[] texts) throws JSONException, IOException {
|
||||
JSONArray jpart = new JSONArray();
|
||||
for (String text : texts) {
|
||||
JSONObject jtext = new JSONObject();
|
||||
jtext.put("text", text);
|
||||
jpart.put(jtext);
|
||||
}
|
||||
|
||||
JSONObject jcontent = new JSONObject();
|
||||
jcontent.put("parts", jpart);
|
||||
JSONArray jcontents = new JSONArray();
|
||||
jcontents.put(jcontent);
|
||||
JSONObject jrequest = new JSONObject();
|
||||
jrequest.put("contents", jcontents);
|
||||
|
||||
String path = "models/" + model + ":generateContent";
|
||||
|
||||
JSONObject jresponse = call(context, "POST", path, jrequest);
|
||||
|
||||
return new String[]{jresponse.getJSONArray("candidates")
|
||||
.getJSONObject(0)
|
||||
.getJSONObject("content")
|
||||
.getJSONArray("parts")
|
||||
.getJSONObject(0)
|
||||
.getString("text")};
|
||||
}
|
||||
|
||||
private static String getUri(Context context) {
|
||||
SharedPreferences prefs = PreferenceManager.getDefaultSharedPreferences(context);
|
||||
return prefs.getString("gemini_uri", BuildConfig.GEMINI_ENDPOINT);
|
||||
}
|
||||
|
||||
private static JSONObject call(Context context, String method, String path, JSONObject args) throws JSONException, IOException {
|
||||
SharedPreferences prefs = PreferenceManager.getDefaultSharedPreferences(context);
|
||||
String apikey = prefs.getString("gemini_apikey", null);
|
||||
|
||||
// https://ai.google.dev/tutorials/rest_quickstart
|
||||
// https://ai.google.dev/api/rest
|
||||
Uri uri = Uri.parse(getUri(context)).buildUpon()
|
||||
.appendEncodedPath(path)
|
||||
.appendQueryParameter("key", apikey)
|
||||
.build();
|
||||
Log.i("Gemini uri=" + uri);
|
||||
|
||||
long start = new Date().getTime();
|
||||
|
||||
URL url = new URL(uri.toString());
|
||||
HttpsURLConnection connection = (HttpsURLConnection) url.openConnection();
|
||||
|
||||
connection.setRequestMethod(method);
|
||||
connection.setDoOutput(args != null);
|
||||
connection.setDoInput(true);
|
||||
connection.setReadTimeout(TIMEOUT * 1000);
|
||||
connection.setConnectTimeout(TIMEOUT * 1000);
|
||||
ConnectionHelper.setUserAgent(context, connection);
|
||||
connection.setRequestProperty("Accept", "application/json");
|
||||
connection.setRequestProperty("Content-Type", "application/json");
|
||||
connection.connect();
|
||||
|
||||
try {
|
||||
if (args != null) {
|
||||
String json = args.toString();
|
||||
Log.i("Gemini request=" + json);
|
||||
connection.getOutputStream().write(json.getBytes());
|
||||
}
|
||||
|
||||
int status = connection.getResponseCode();
|
||||
if (status != HttpsURLConnection.HTTP_OK) {
|
||||
String error = "Error " + status + ": " + connection.getResponseMessage();
|
||||
try {
|
||||
InputStream is = connection.getErrorStream();
|
||||
if (is != null)
|
||||
error += "\n" + Helper.readStream(is);
|
||||
} catch (Throwable ex) {
|
||||
Log.w(ex);
|
||||
}
|
||||
Log.w("Gemini error=" + error);
|
||||
throw new IOException(error);
|
||||
}
|
||||
|
||||
String response = Helper.readStream(connection.getInputStream());
|
||||
Log.i("Gemini response=" + response);
|
||||
|
||||
return new JSONObject(response);
|
||||
} finally {
|
||||
connection.disconnect();
|
||||
long elapsed = new Date().getTime() - start;
|
||||
Log.i("Gemini elapsed=" + (elapsed / 1000f));
|
||||
}
|
||||
}
|
||||
|
||||
static String truncateParagraphs(@NonNull String text) {
|
||||
return truncateParagraphs(text, MAX_GEMINI_LEN);
|
||||
}
|
||||
|
||||
static String truncateParagraphs(@NonNull String text, int maxlen) {
|
||||
String[] paragraphs = text.split("[\\r\\n]+");
|
||||
|
||||
int i = 0;
|
||||
StringBuilder sb = new StringBuilder();
|
||||
while (i < paragraphs.length &&
|
||||
sb.length() + paragraphs[i].length() + 1 < maxlen)
|
||||
sb.append(paragraphs[i++]).append('\n');
|
||||
|
||||
return sb.toString();
|
||||
}
|
||||
}
|
Loading…
Reference in new issue