Initial commit: HeliBoard + Gemma 4 on-device AI correction

Integrates LiteRT-LM (Google) with Gemma 4 E2B-it for on-device spell/grammar correction in HeliBoard. After sentence-ending punctuation, the model checks the sentence and shows a correction in the suggestion strip. Tapping it replaces the original text — fully offline, no cloud API. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-04-10 14:57:10 +02:00
commit aa8fff2c0d
17 changed files with 1038 additions and 0 deletions
@@ -0,0 +1,74 @@
+From: AIKeyboard AI Integration <ai@aikeyboard.local>
+Date: Wed, 09 Apr 2025 00:00:00 +0000
+Subject: [PATCH 3/3] Add italic colour styling for AI correction suggestions
+
+Modifies two methods in SuggestionStripLayoutHelper:
+
+1. getSuggestionTextColor(): Returns mColorAutoCorrect for AI suggestions
+   (KIND_APP_DEFINED / KIND_AI_CORRECTION = 6), giving them the same accent
+   colour as auto-corrections.
+
+2. getStyledSuggestedWord(): Adds an italic StyleSpan for AI suggestions so
+   users can visually distinguish them from regular word predictions.
+
+Both method signatures and bodies verified against HeliBoard main (Apr 2025).
+---
+ .../suggestions/SuggestionStripLayoutHelper.java | 31 ++++++++++++++++++
+ 1 file changed, 31 insertions(+)
+
+diff --git a/app/src/main/java/helium314/keyboard/latin/suggestions/SuggestionStripLayoutHelper.java b/app/src/main/java/helium314/keyboard/latin/suggestions/SuggestionStripLayoutHelper.java
+--- a/app/src/main/java/helium314/keyboard/latin/suggestions/SuggestionStripLayoutHelper.java
+++ b/app/src/main/java/helium314/keyboard/latin/suggestions/SuggestionStripLayoutHelper.java
+@@ -1,3 +1,5 @@
+import android.graphics.Typeface;
+import android.text.style.StyleSpan;
+ /*
+  * Copyright (C) 2013 The Android Open Source Project
+  *
+@@ -1,0 +1,0 @@ private int getSuggestionTextColor(final SuggestedWords suggestedWords,
+     private int getSuggestionTextColor(final SuggestedWords suggestedWords,
+             final int indexInSuggestedWords) {
+-        final boolean isTypedWord = suggestedWords.getInfo(indexInSuggestedWords).isKindOf(SuggestedWordInfo.KIND_TYPED);
+        final SuggestedWordInfo wordInfo = suggestedWords.getInfo(indexInSuggestedWords);
+        final boolean isTypedWord = wordInfo.isKindOf(SuggestedWordInfo.KIND_TYPED);
+
+        // AI correction: use auto-correct accent colour so it stands out
+        if (wordInfo.isKindOf(SuggestedWordInfo.KIND_APP_DEFINED)) {
+            return mColorAutoCorrect;
+        }
+
+         final int color;
+         if (indexInSuggestedWords == SuggestedWords.INDEX_OF_AUTO_CORRECTION && suggestedWords.mWillAutoCorrect) {
+@@ -1,0 +1,0 @@ private CharSequence getStyledSuggestedWord(final SuggestedWords suggestedWords,
+     private CharSequence getStyledSuggestedWord(final SuggestedWords suggestedWords,
+             final int indexInSuggestedWords) {
+         if (indexInSuggestedWords >= suggestedWords.size()) {
+             return null;
+         }
+         final String word = suggestedWords.getLabel(indexInSuggestedWords);
+         final boolean isAutoCorrection = suggestedWords.mWillAutoCorrect
+                 && indexInSuggestedWords == SuggestedWords.INDEX_OF_AUTO_CORRECTION;
+         final boolean isTypedWordValid = suggestedWords.mTypedWordValid
+                 && indexInSuggestedWords == SuggestedWords.INDEX_OF_TYPED_WORD;
+-        if (!isAutoCorrection && !isTypedWordValid) {
+        final boolean isAiCorrection = suggestedWords.getInfo(indexInSuggestedWords)
+                .isKindOf(SuggestedWordInfo.KIND_APP_DEFINED);
+        if (!isAutoCorrection && !isTypedWordValid && !isAiCorrection) {
+             return word;
+         }
+
+         final Spannable spannedWord = new SpannableString(word);
+         final int options = mSuggestionStripOptions;
+         if ((isAutoCorrection && (options & AUTO_CORRECT_BOLD) != 0)
+                 || (isTypedWordValid && (options & VALID_TYPED_WORD_BOLD) != 0)) {
+             addStyleSpan(spannedWord, BOLD_SPAN);
+         }
+         if (isAutoCorrection && (options & AUTO_CORRECT_UNDERLINE) != 0) {
+             addStyleSpan(spannedWord, UNDERLINE_SPAN);
+         }
+        // AI correction: italic to signal model-generated suggestion
+        if (isAiCorrection) {
+            addStyleSpan(spannedWord, new StyleSpan(Typeface.ITALIC));
+        }
+         return spannedWord;
+     }