add translation logic

Signed-off-by: alperozturk96 <alper_ozturk@proton.me>

Author: alperozturk96

app/src/main/java/com/nextcloud/client/assistant/AssistantViewModel.kt

@@ -16,6 +16,7 @@ import com.nextcloud.client.assistant.repository.remote.AssistantRemoteRepositor
 import com.nextcloud.utils.TimeConstants.MILLIS_PER_SECOND
 import com.nextcloud.utils.extensions.isHuman
 import com.owncloud.android.R
+import com.owncloud.android.lib.common.operations.RemoteOperationResult
 import com.owncloud.android.lib.common.utils.Log_OC
 import com.owncloud.android.lib.resources.assistant.chat.model.ChatMessage
 import com.owncloud.android.lib.resources.assistant.chat.model.ChatMessageRequest
@@ -188,32 +189,35 @@ class AssistantViewModel(
     }
 
     fun translate(textToTranslate: String, originLanguage: TranslationLanguage, targetLanguage: TranslationLanguage) {
-        val task = _selectedTaskType.value
-        if (task == null) {
-            _snackbarMessageId.update {
-                R.string.assistant_screen_select_task
+        viewModelScope.launch(Dispatchers.IO) {
+            val task = _selectedTaskType.value
+            if (task == null) {
+                _snackbarMessageId.update {
+                    R.string.assistant_screen_select_task
+                }
+                return@launch
             }
-            return
-        }
 
-        val model = task.toTranslationModel()
+            val model = task.toTranslationModel()
 
-        if (model == null) {
-            _snackbarMessageId.update {
-                R.string.translation_screen_error_message
+            if (model == null) {
+                _snackbarMessageId.update {
+                    R.string.translation_screen_error_message
+                }
+                return@launch
             }
-            return
-        }
 
-        val input = TranslationRequest(
-            input = textToTranslate,
-            originLanguage = originLanguage.code,
-            targetLanguage = targetLanguage.code,
-            maxTokens = model.maxTokens,
-            model = model.model
-        ).toJson()
+            val input = TranslationRequest(
+                input = textToTranslate,
+                originLanguage = originLanguage.code,
+                targetLanguage = targetLanguage.code,
+                maxTokens = model.maxTokens,
+                model = model.model
+            )
 
-        createTask(input, task)
+            val result = remoteRepository.translate(input, task)
+            handleTaskCreation(result)
+        }
     }
 
     // region chat
@@ -261,6 +265,10 @@ class AssistantViewModel(
     // region task
     fun createTask(input: String, taskType: TaskTypeData) = viewModelScope.launch(Dispatchers.IO) {
         val result = remoteRepository.createTask(input, taskType)
+        handleTaskCreation(result)
+    }
+
+    private suspend fun handleTaskCreation(result: RemoteOperationResult<*>) {
         val message = if (result.isSuccess) {
             R.string.assistant_screen_task_create_success_message
         } else {

app/src/main/java/com/nextcloud/client/assistant/repository/remote/AssistantRemoteRepository.kt

@@ -15,6 +15,7 @@ import com.owncloud.android.lib.resources.assistant.chat.model.Session
 import com.owncloud.android.lib.resources.assistant.chat.model.SessionTask
 import com.owncloud.android.lib.resources.assistant.v2.model.Task
 import com.owncloud.android.lib.resources.assistant.v2.model.TaskTypeData
+import com.owncloud.android.lib.resources.assistant.v2.model.TranslationRequest
 
 interface AssistantRemoteRepository {
     suspend fun getTaskTypes(): List<TaskTypeData>?
@@ -36,4 +37,6 @@ interface AssistantRemoteRepository {
     suspend fun generateSession(sessionId: String): SessionTask?
 
     suspend fun checkGeneration(taskId: String, sessionId: String): ChatMessage?
+
+    suspend fun translate(input: TranslationRequest, taskType: TaskTypeData): RemoteOperationResult<Void>
 }

app/src/main/java/com/nextcloud/client/assistant/repository/remote/AssistantRemoteRepositoryImpl.kt

@@ -27,11 +27,13 @@ import com.owncloud.android.lib.resources.assistant.v1.GetTaskListRemoteOperatio
 import com.owncloud.android.lib.resources.assistant.v1.GetTaskTypesRemoteOperationV1
 import com.owncloud.android.lib.resources.assistant.v1.model.toV2
 import com.owncloud.android.lib.resources.assistant.v2.CreateTaskRemoteOperationV2
+import com.owncloud.android.lib.resources.assistant.v2.CreateTranslationTaskRemoteOperation
 import com.owncloud.android.lib.resources.assistant.v2.DeleteTaskRemoteOperationV2
 import com.owncloud.android.lib.resources.assistant.v2.GetTaskListRemoteOperationV2
 import com.owncloud.android.lib.resources.assistant.v2.GetTaskTypesRemoteOperationV2
 import com.owncloud.android.lib.resources.assistant.v2.model.Task
 import com.owncloud.android.lib.resources.assistant.v2.model.TaskTypeData
+import com.owncloud.android.lib.resources.assistant.v2.model.TranslationRequest
 import com.owncloud.android.lib.resources.status.NextcloudVersion
 import com.owncloud.android.lib.resources.status.OCCapability
 import kotlinx.coroutines.Dispatchers
@@ -124,4 +126,9 @@ class AssistantRemoteRepositoryImpl(private val client: NextcloudClient, capabil
             val result = CheckGenerationRemoteOperation(taskId, sessionId).execute(client)
             if (result.isSuccess) result.resultData else null
         }
+
+    override suspend fun translate(input: TranslationRequest, taskType: TaskTypeData): RemoteOperationResult<Void> =
+        withContext(Dispatchers.IO) {
+            CreateTranslationTaskRemoteOperation(input, taskType).execute(client)
+        }
 }

app/src/main/java/com/nextcloud/client/assistant/repository/remote/MockAssistantRemoteRepository.kt

@@ -18,6 +18,7 @@ import com.owncloud.android.lib.resources.assistant.v2.model.Task
 import com.owncloud.android.lib.resources.assistant.v2.model.TaskInput
 import com.owncloud.android.lib.resources.assistant.v2.model.TaskOutput
 import com.owncloud.android.lib.resources.assistant.v2.model.TaskTypeData
+import com.owncloud.android.lib.resources.assistant.v2.model.TranslationRequest
 
 @Suppress("MagicNumber")
 class MockAssistantRemoteRepository(private val giveEmptyTasks: Boolean = false) : AssistantRemoteRepository {
@@ -68,12 +69,14 @@ class MockAssistantRemoteRepository(private val giveEmptyTasks: Boolean = false)
     }
 
     override suspend fun deleteTask(id: Long): RemoteOperationResult<Void> =
-        RemoteOperationResult<Void>(RemoteOperationResult.ResultCode.OK)
+        RemoteOperationResult(RemoteOperationResult.ResultCode.OK)
     override suspend fun fetchChatMessages(id: Long): List<ChatMessage> = emptyList()
     override suspend fun sendChatMessage(request: ChatMessageRequest): ChatMessage? = null
     override suspend fun createConversation(title: String): CreateConversation? = null
     override suspend fun checkSession(sessionId: String): Session? = null
     override suspend fun generateSession(sessionId: String): SessionTask? = null
 
     override suspend fun checkGeneration(taskId: String, sessionId: String): ChatMessage? = null
+    override suspend fun translate(input: TranslationRequest, taskType: TaskTypeData): RemoteOperationResult<Void> =
+        RemoteOperationResult(RemoteOperationResult.ResultCode.OK)
 }