better ui ux

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

Author: alperozturk96

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

@@ -62,12 +62,15 @@ class AssistantViewModel(
     private val _snackbarMessageId = MutableStateFlow<Int?>(null)
     val snackbarMessageId: StateFlow<Int?> = _snackbarMessageId
 
-    private val _isTranslationTask = MutableStateFlow<Boolean>(false)
+    private val _isTranslationTask = MutableStateFlow(false)
     val isTranslationTask: StateFlow<Boolean> = _isTranslationTask
 
-    private val _isTranslationTaskCreated = MutableStateFlow<Boolean>(false)
+    private val _isTranslationTaskCreated = MutableStateFlow(false)
     val isTranslationTaskCreated: StateFlow<Boolean> = _isTranslationTaskCreated
 
+    private val _translationTaskOutput = MutableStateFlow("")
+    val translationTaskOutput: StateFlow<String> = _translationTaskOutput
+
     private val selectedTask = MutableStateFlow<Task?>(null)
 
     private val _selectedTaskType = MutableStateFlow<TaskTypeData?>(null)
@@ -204,17 +207,18 @@ class AssistantViewModel(
         }
     }
 
+    // region translation
     fun translate(textToTranslate: String, originLanguage: TranslationLanguage, targetLanguage: TranslationLanguage) {
         viewModelScope.launch(Dispatchers.IO) {
-            val task = _selectedTaskType.value
-            if (task == null) {
+            val taskType = _selectedTaskType.value
+            if (taskType == null) {
                 _snackbarMessageId.update {
                     R.string.assistant_screen_select_task
                 }
                 return@launch
             }
 
-            val model = task.toTranslationModel()
+            val model = taskType.toTranslationModel()
 
             if (model == null) {
                 _snackbarMessageId.update {
@@ -231,15 +235,55 @@ class AssistantViewModel(
                 model = model.model
             )
 
-            val result = remoteRepository.translate(input, task)
+            val result = remoteRepository.translate(input, taskType)
             if (result.isSuccess) {
-                _isTranslationTaskCreated.update {
-                    true
-                }
+                _isTranslationTaskCreated.update { true }
+
+                val selectedTaskId = selectedTask.value?.id ?: return@launch
+
+                pollTranslationResult(
+                    taskType = taskType,
+                    selectedTaskId = selectedTaskId
+                )
+
+                _isTranslationTaskCreated.update { false }
             }
         }
     }
 
+    private suspend fun pollTranslationResult(
+        taskType: TaskTypeData,
+        selectedTaskId: Long,
+        maxRetries: Int = 3,
+    ) {
+        val taskTypeId = taskType.id ?: return
+
+        repeat(maxRetries) { attempt ->
+            val translationTasks = remoteRepository.getTaskList(taskTypeId)
+            val translationResult = translationTasks
+                ?.find { it.id == selectedTaskId }
+                ?.output
+                ?.output
+
+            if (!translationResult.isNullOrBlank()) {
+                _translationTaskOutput.update { translationResult }
+                return
+            }
+
+            Log_OC.d(TAG, "Translation not ready yet (attempt ${attempt + 1}/$maxRetries)")
+
+            if (attempt < maxRetries - 1) {
+                delay(POLLING_INTERVAL_MS)
+            }
+        }
+
+        Log_OC.w(TAG, "Translation polling finished but result is still empty")
+        updateSnackbarMessage(R.string.translation_screen_task_processing)
+        onTranslationScreenDismissed()
+    }
+    // endregion
+
+
     // region chat
     fun sendChatMessage(content: String, sessionId: Long) = viewModelScope.launch(Dispatchers.IO) {
         val request = ChatMessageRequest(

app/src/main/java/com/nextcloud/client/assistant/translate/TranslationScreen.kt

@@ -38,6 +38,7 @@ import androidx.compose.material3.TextField
 import androidx.compose.material3.TextFieldDefaults
 import androidx.compose.runtime.Composable
 import androidx.compose.runtime.DisposableEffect
+import androidx.compose.runtime.LaunchedEffect
 import androidx.compose.runtime.collectAsState
 import androidx.compose.runtime.getValue
 import androidx.compose.runtime.mutableStateOf
@@ -71,6 +72,8 @@ fun TranslationScreen(
 ) {
     val languages = remember(selectedTaskType) { selectedTaskType?.toTranslationLanguages() }
     val isTranslationTaskCreated by viewModel.isTranslationTaskCreated.collectAsState()
+    val translationTaskOutput by viewModel.translationTaskOutput.collectAsState()
+
     var sourceState by remember {
         mutableStateOf(
             TranslationSideState(
@@ -84,6 +87,10 @@ fun TranslationScreen(
         mutableStateOf(TranslationSideState(language = languages?.targetLanguages?.firstOrNull(), isTarget = true))
     }
 
+    LaunchedEffect(translationTaskOutput) {
+        targetState = targetState.copy(text = translationTaskOutput)
+    }
+
     BackHandler {
         viewModel.onTranslationScreenDismissed()
         viewModel.updateScreenState(AssistantScreenState.TaskContent)