Add stop generating button

Author: sollarp

app/src/main/java/com/chatgptlite/wanted/constants/Constants.kt

@@ -9,3 +9,4 @@ const val urlToGithub = "https://github.com/lambiengcode"
 
 const val matchResultString = "\"text\":"
 const val matchResultTurboString = "\"content\":"
+const val ConversationTestTag = "ConversationTestTag"

app/src/main/java/com/chatgptlite/wanted/ui/conversations/Conversation.kt

@@ -1,17 +1,27 @@
 package com.chatgptlite.wanted.ui.conversations
 
+import androidx.compose.animation.animateContentSize
 import androidx.compose.foundation.layout.*
 import androidx.compose.foundation.lazy.LazyColumn
 import androidx.compose.foundation.lazy.rememberLazyListState
+import androidx.compose.foundation.shape.RoundedCornerShape
+import androidx.compose.material.icons.Icons
+import androidx.compose.material.icons.filled.Stop
+import androidx.compose.material3.ExtendedFloatingActionButton
+import androidx.compose.material3.Icon
 import androidx.compose.material3.Surface
+import androidx.compose.material3.Text
 import androidx.compose.runtime.Composable
 import androidx.compose.runtime.collectAsState
 import androidx.compose.runtime.getValue
+import androidx.compose.ui.Alignment
 import androidx.compose.ui.Modifier
+import androidx.compose.ui.graphics.Color
 import androidx.compose.ui.platform.testTag
 import androidx.compose.ui.tooling.preview.Preview
 import androidx.compose.ui.unit.dp
 import androidx.hilt.navigation.compose.hiltViewModel
+import com.chatgptlite.wanted.constants.ConversationTestTag
 import com.chatgptlite.wanted.models.MessageModel
 import com.chatgptlite.wanted.ui.conversations.components.MessageCard
 import com.chatgptlite.wanted.ui.conversations.components.TextInput
@@ -27,19 +37,18 @@ fun Conversation() {
         ) {
             Box(Modifier.fillMaxSize()) {
                 Column(Modifier.fillMaxSize()) {
-                    MessageList(modifier = Modifier
-                        .weight(1f)
-                        .padding(horizontal = 16.dp))
+                    MessageList(
+                        modifier = Modifier
+                            .weight(1f)
+                            .padding(horizontal = 16.dp)
+                    )
                     TextInput()
                 }
             }
         }
     }
 }
 
-
-const val ConversationTestTag = "ConversationTestTag"
-
 @Composable
 fun MessageList(
     modifier: Modifier = Modifier,
@@ -49,6 +58,7 @@ fun MessageList(
 
     val conversationId by conversationViewModel.currentConversationState.collectAsState()
     val messagesMap by conversationViewModel.messagesState.collectAsState()
+    val isFabExpanded by conversationViewModel.isFabExpanded.collectAsState()
 
     val messages: List<MessageModel> =
         if (messagesMap[conversationId] == null) listOf() else messagesMap[conversationId]!!
@@ -76,11 +86,33 @@ fun MessageList(
                 }
             }
         }
+        ExtendedFloatingActionButton(
+            text = {
+                Text(text = "Stop Generating", color = Color.White)
+            },
+            icon = {
+                Icon(
+                    imageVector = Icons.Default.Stop,
+                    contentDescription = "Stop Generating",
+                    tint = Color.White,
+                    modifier = Modifier
+                        .size(35.dp)
+                )
+            },
+            onClick = {
+                conversationViewModel.stopReceivingResults()
+            },
+            shape = RoundedCornerShape(16.dp),
+            modifier = Modifier
+                .align(Alignment.BottomEnd)
+                .padding(bottom = 16.dp)
+                .animateContentSize(),
+            expanded = isFabExpanded,
+        )
     }
 }
 
 
-
 @Preview(showBackground = true)
 @Composable
 fun DefaultPreview2() {

app/src/main/java/com/chatgptlite/wanted/ui/conversations/ConversationViewModel.kt

@@ -30,12 +30,18 @@ class ConversationViewModel @Inject constructor(
     private val _messages: MutableStateFlow<HashMap<String, MutableList<MessageModel>>> =
         MutableStateFlow(HashMap())
     private val _isFetching: MutableStateFlow<Boolean> = MutableStateFlow(false)
+    private val _isFabExpanded = MutableStateFlow(false)
 
     val currentConversationState: StateFlow<String> = _currentConversation.asStateFlow()
     val conversationsState: StateFlow<MutableList<ConversationModel>> = _conversations.asStateFlow()
     val messagesState: StateFlow<HashMap<String, MutableList<MessageModel>>> =
         _messages.asStateFlow()
     val isFetching: StateFlow<Boolean> = _isFetching.asStateFlow()
+    val isFabExpanded: StateFlow<Boolean> get() = _isFabExpanded
+
+    private var stopReceivingResults = false
+
+
 
     suspend fun initialize() {
         _isFetching.value = true
@@ -59,6 +65,7 @@ class ConversationViewModel @Inject constructor(
     }
 
     suspend fun sendMessage(message: String) {
+        stopReceivingResults = false
         if (getMessagesByConversation(_currentConversation.value).isEmpty()) {
             createConversationRemote(message)
         }
@@ -83,12 +90,19 @@ class ConversationViewModel @Inject constructor(
                 messagesTurbo = getMessagesParamsTurbo(_currentConversation.value)
             )
         )
-
         var answerFromGPT: String = ""
-
-        flow.collect {
-            answerFromGPT += it
+        // When flow collecting updateLocalAnswer including FAB behavior expanded.
+        // On completion FAB == false
+        flow.onCompletion {
+            setFabExpanded(false)
+        }.collect { value ->
+            if (stopReceivingResults) {
+                setFabExpanded(false)
+                return@collect
+            }
+            answerFromGPT += value
             updateLocalAnswer(answerFromGPT.trim())
+            setFabExpanded(true)
         }
 
         // Save to Firestore
@@ -210,4 +224,10 @@ class ConversationViewModel @Inject constructor(
 
         _messages.value = messagesMap
     }
+    fun stopReceivingResults() {
+        stopReceivingResults = true
+    }
+    private fun setFabExpanded(expanded: Boolean) {
+        _isFabExpanded.value = expanded
+    }
 }