feat: add mock writing assistant for UI testing
Temporary MockWritingAssistant that returns fake transformed text after a short delay. Enabled via useMockAi flag in ShortNotePostViewModel. Allows testing the AI writing help UI on devices without Gemini Nano. TODO: Set useMockAi = false before shipping. https://claude.ai/code/session_01RbCYGrbbapRMike8WQy41F
This commit is contained in:
@@ -0,0 +1,97 @@
|
|||||||
|
/*
|
||||||
|
* Copyright (c) 2025 Vitor Pamplona
|
||||||
|
*
|
||||||
|
* Permission is hereby granted, free of charge, to any person obtaining a copy of
|
||||||
|
* this software and associated documentation files (the "Software"), to deal in
|
||||||
|
* the Software without restriction, including without limitation the rights to use,
|
||||||
|
* copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the
|
||||||
|
* Software, and to permit persons to whom the Software is furnished to do so,
|
||||||
|
* subject to the following conditions:
|
||||||
|
*
|
||||||
|
* The above copyright notice and this permission notice shall be included in all
|
||||||
|
* copies or substantial portions of the Software.
|
||||||
|
*
|
||||||
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
|
||||||
|
* FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
|
||||||
|
* COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN
|
||||||
|
* AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
||||||
|
* WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||||||
|
*/
|
||||||
|
package com.vitorpamplona.amethyst.service.ai
|
||||||
|
|
||||||
|
import kotlinx.coroutines.delay
|
||||||
|
|
||||||
|
/**
|
||||||
|
* TODO: Remove before shipping. Debug-only mock for testing the AI writing help UI
|
||||||
|
* on devices without Gemini Nano / AICore support.
|
||||||
|
*/
|
||||||
|
class MockWritingAssistant : WritingAssistant {
|
||||||
|
override suspend fun checkAvailability(): WritingAssistantStatus = WritingAssistantStatus.Available
|
||||||
|
|
||||||
|
override suspend fun transform(
|
||||||
|
text: String,
|
||||||
|
tone: WritingTone,
|
||||||
|
): WritingResult {
|
||||||
|
delay(800)
|
||||||
|
|
||||||
|
val transformed =
|
||||||
|
when (tone) {
|
||||||
|
WritingTone.CORRECT -> {
|
||||||
|
correctMock(text)
|
||||||
|
}
|
||||||
|
|
||||||
|
WritingTone.REPHRASE -> {
|
||||||
|
"Here's another way to put it: $text"
|
||||||
|
}
|
||||||
|
|
||||||
|
WritingTone.SHORTER -> {
|
||||||
|
text
|
||||||
|
.split(".")
|
||||||
|
.firstOrNull()
|
||||||
|
?.trim()
|
||||||
|
?.plus(".") ?: text
|
||||||
|
}
|
||||||
|
|
||||||
|
WritingTone.ELABORATE -> {
|
||||||
|
"$text Furthermore, this point deserves deeper consideration and nuance."
|
||||||
|
}
|
||||||
|
|
||||||
|
WritingTone.FRIENDLY -> {
|
||||||
|
"Hey! $text Hope that makes sense! :)"
|
||||||
|
}
|
||||||
|
|
||||||
|
WritingTone.PROFESSIONAL -> {
|
||||||
|
"I would like to bring to your attention the following: $text"
|
||||||
|
}
|
||||||
|
|
||||||
|
WritingTone.MORE_DIRECT -> {
|
||||||
|
text.replace("I think ", "").replace("maybe ", "").replace("perhaps ", "")
|
||||||
|
}
|
||||||
|
|
||||||
|
WritingTone.PUNCHY -> {
|
||||||
|
text.uppercase().replace(".", "!")
|
||||||
|
}
|
||||||
|
|
||||||
|
WritingTone.EMOJIFY -> {
|
||||||
|
"$text \uD83D\uDE80\uD83D\uDD25\u2728"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return WritingResult(
|
||||||
|
originalText = text,
|
||||||
|
transformedText = transformed,
|
||||||
|
tone = tone,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
private fun correctMock(text: String): String =
|
||||||
|
text
|
||||||
|
.replace("teh ", "the ")
|
||||||
|
.replace("dont ", "don't ")
|
||||||
|
.replace("cant ", "can't ")
|
||||||
|
.replace("wont ", "won't ")
|
||||||
|
.replace("i ", "I ")
|
||||||
|
|
||||||
|
override fun close() {}
|
||||||
|
}
|
||||||
+10
-1
@@ -44,6 +44,7 @@ import com.vitorpamplona.amethyst.model.Account
|
|||||||
import com.vitorpamplona.amethyst.model.LocalCache
|
import com.vitorpamplona.amethyst.model.LocalCache
|
||||||
import com.vitorpamplona.amethyst.model.Note
|
import com.vitorpamplona.amethyst.model.Note
|
||||||
import com.vitorpamplona.amethyst.model.User
|
import com.vitorpamplona.amethyst.model.User
|
||||||
|
import com.vitorpamplona.amethyst.service.ai.MockWritingAssistant
|
||||||
import com.vitorpamplona.amethyst.service.ai.WritingAssistant
|
import com.vitorpamplona.amethyst.service.ai.WritingAssistant
|
||||||
import com.vitorpamplona.amethyst.service.ai.WritingAssistantFactory
|
import com.vitorpamplona.amethyst.service.ai.WritingAssistantFactory
|
||||||
import com.vitorpamplona.amethyst.service.ai.WritingAssistantStatus
|
import com.vitorpamplona.amethyst.service.ai.WritingAssistantStatus
|
||||||
@@ -297,6 +298,9 @@ open class ShortNotePostViewModel :
|
|||||||
var wantsAnonymousPost by mutableStateOf(false)
|
var wantsAnonymousPost by mutableStateOf(false)
|
||||||
|
|
||||||
// AI Writing Help
|
// AI Writing Help
|
||||||
|
// TODO: Remove useMockAi before shipping. Set to true to test UI without Gemini Nano.
|
||||||
|
private val useMockAi = true
|
||||||
|
|
||||||
var wantsAiHelp by mutableStateOf(false)
|
var wantsAiHelp by mutableStateOf(false)
|
||||||
var aiResult by mutableStateOf<WritingResult?>(null)
|
var aiResult by mutableStateOf<WritingResult?>(null)
|
||||||
var aiStatus by mutableStateOf<WritingAssistantStatus>(WritingAssistantStatus.Unavailable)
|
var aiStatus by mutableStateOf<WritingAssistantStatus>(WritingAssistantStatus.Unavailable)
|
||||||
@@ -305,7 +309,12 @@ open class ShortNotePostViewModel :
|
|||||||
|
|
||||||
fun initWritingAssistant(context: android.content.Context) {
|
fun initWritingAssistant(context: android.content.Context) {
|
||||||
if (writingAssistant == null) {
|
if (writingAssistant == null) {
|
||||||
writingAssistant = WritingAssistantFactory.create(context)
|
writingAssistant =
|
||||||
|
if (useMockAi) {
|
||||||
|
MockWritingAssistant()
|
||||||
|
} else {
|
||||||
|
WritingAssistantFactory.create(context)
|
||||||
|
}
|
||||||
viewModelScope.launch(Dispatchers.IO) {
|
viewModelScope.launch(Dispatchers.IO) {
|
||||||
aiStatus = writingAssistant?.checkAvailability() ?: WritingAssistantStatus.Unavailable
|
aiStatus = writingAssistant?.checkAvailability() ?: WritingAssistantStatus.Unavailable
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user