fix translation flow

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

Author: alperozturk96

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

@@ -24,6 +24,7 @@ import androidx.compose.foundation.lazy.items
 import androidx.compose.foundation.pager.HorizontalPager
 import androidx.compose.foundation.pager.rememberPagerState
 import androidx.compose.material3.ExperimentalMaterial3Api
+import androidx.compose.material3.FloatingActionButton
 import androidx.compose.material3.Icon
 import androidx.compose.material3.IconButton
 import androidx.compose.material3.LinearProgressIndicator
@@ -96,6 +97,7 @@ fun AssistantScreen(
     val messageId by viewModel.snackbarMessageId.collectAsState()
     val screenOverlayState by viewModel.screenOverlayState.collectAsState()
     val selectedTaskType by viewModel.selectedTaskType.collectAsState()
+    val isTranslationTask by viewModel.isTranslationTask.collectAsState()
     val filteredTaskList by viewModel.filteredTaskList.collectAsState()
     val screenState by viewModel.screenState.collectAsState()
     val taskTypes by viewModel.taskTypes.collectAsState()
@@ -192,7 +194,7 @@ fun AssistantScreen(
                         }
                     },
                     bottomBar = {
-                        if (!taskTypes.isNullOrEmpty()) {
+                        if (!taskTypes.isNullOrEmpty() && selectedTaskType?.isTranslate() != true) {
                             InputBar(
                                 sessionId,
                                 selectedTaskType,
@@ -202,6 +204,24 @@ fun AssistantScreen(
                     },
                     snackbarHost = {
                         SnackbarHost(snackbarHostState)
+                    },
+                    floatingActionButton = {
+                        if (selectedTaskType?.isTranslate() == true && !isTranslationTask) {
+                            FloatingActionButton(onClick = {
+                                viewModel.updateTranslationTaskState(true)
+                                viewModel.updateScreenState(AssistantScreenState.Translation(null))
+                            }, content = {
+                                Row(
+                                    verticalAlignment = Alignment.CenterVertically,
+                                    modifier = Modifier.padding(16.dp)
+                                ) {
+                                    Icon(
+                                        painter = painterResource(R.drawable.ic_plus),
+                                        contentDescription = "translate button"
+                                    )
+                                }
+                            })
+                        }
                     }
                 ) { paddingValues ->
                     when (screenState) {
@@ -317,12 +337,7 @@ private fun InputBar(sessionId: Long?, selectedTaskType: TaskTypeData?, viewMode
                                 viewModel.createConversation(text)
                             }
                         } else {
-                            if (taskType.isTranslate()) {
-                                // TODO:
-                                // viewModel.translate()
-                            } else {
-                                viewModel.createTask(input = text, taskType = taskType)
-                            }
+                            viewModel.createTask(input = text, taskType = taskType)
                         }
 
                         scope.launch {

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

@@ -63,6 +63,9 @@ class AssistantViewModel(
     private val _snackbarMessageId = MutableStateFlow<Int?>(null)
     val snackbarMessageId: StateFlow<Int?> = _snackbarMessageId
 
+    private val _isTranslationTask = MutableStateFlow<Boolean>(false)
+    val isTranslationTask: StateFlow<Boolean> = _isTranslationTask
+
     private val _selectedTask = MutableStateFlow<Task?>(null)
     val selectedTask: StateFlow<Task?> = _selectedTask
 
@@ -186,7 +189,13 @@ class AssistantViewModel(
                     isChat && chats.isEmpty() -> AssistantScreenState.emptyChatList()
                     isChat -> AssistantScreenState.ChatContent
                     !isChat && (tasks == null || tasks.isEmpty()) -> AssistantScreenState.emptyTaskList()
-                    else -> AssistantScreenState.TaskContent
+                    else -> {
+                        if (!_isTranslationTask.value) {
+                            AssistantScreenState.TaskContent
+                        } else {
+                            _screenState.value
+                        }
+                    }
                 }
             }.collect { newState ->
                 _screenState.value = newState
@@ -399,6 +408,12 @@ class AssistantViewModel(
         }
     }
 
+    fun updateTranslationTaskState(value: Boolean) {
+        _isTranslationTask.update {
+            value
+        }
+    }
+
     private fun removeTaskFromList(id: Long) {
         _filteredTaskList.update { currentList ->
             currentList?.filter { it.id != id }

app/src/main/java/com/nextcloud/client/assistant/task/TaskView.kt

@@ -64,6 +64,7 @@ fun TaskView(task: Task, viewModel: AssistantViewModel, capability: OCCapability
                     viewModel.selectTask(task)
 
                     if (task.type == "core:text2text:translate") {
+                        viewModel.updateTranslationTaskState(true)
                         viewModel.updateScreenState(AssistantScreenState.Translation(task))
                     } else {
                         showTaskDetailBottomSheet = true

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

@@ -68,13 +68,15 @@ fun TranslationScreen(selectedTaskType: TaskTypeData?, viewModel: AssistantViewM
     }
 
     BackHandler {
+        viewModel.updateTranslationTaskState(false)
         viewModel.selectTask(null)
         viewModel.updateScreenState(AssistantScreenState.TaskContent)
     }
 
     // task is unselected
     DisposableEffect(Unit) {
         onDispose {
+            viewModel.updateTranslationTaskState(false)
             viewModel.selectTask(null)
         }
     }
@@ -86,7 +88,6 @@ fun TranslationScreen(selectedTaskType: TaskTypeData?, viewModel: AssistantViewM
             .padding(top = 32.dp),
         floatingActionButton = {
             FloatingActionButton(onClick = {
-                // TODO:
                 val originLang = sourceState.language
                 val targetLang = targetState.language
                 if (originLang != null && targetLang != null) {