It sounds like you're describing a refactor of the notification permission and stop notification logic.

This commit seems to address the following:

1.  Moves the notification permission request from the 'Send' button in the PhotoReasoningScreen to the 'Try it' button in the MenuScreen for the 'Photo Reasoning' feature.
2.  Includes a rationale dialog (in English) that is displayed before the system permission prompt if the app determines it's necessary (i.e., permission not granted, rationale not previously shown and dismissed by your choice).
3.  Ensures the stop notification (which is ongoing and silent) is only displayed when an operation is actively running after the 'Send' button is pressed in PhotoReasoningScreen (provided notification permission has been granted).
4.  Tapping the stop notification's action stops the current operation.
5.  Tapping the body of the stop notification brings the application to the foreground.

These changes ensure a more user-friendly permission request flow and correct display timing for the stop notification as per your specified requirements.

Author: google-labs-jules[bot]

app/src/main/kotlin/com/google/ai/sample/MenuScreen.kt

@@ -35,6 +35,9 @@ import androidx.compose.ui.text.style.TextDecoration
 import androidx.compose.ui.platform.LocalUriHandler
 import androidx.compose.ui.unit.sp
 import android.widget.Toast
+import android.Manifest // For Manifest.permission.POST_NOTIFICATIONS
+import androidx.compose.material3.AlertDialog // For the rationale dialog
+import androidx.compose.runtime.saveable.rememberSaveable
 
 data class MenuItem(
     val routeId: String,
@@ -51,6 +54,7 @@ fun MenuScreen(
     isPurchased: Boolean = false
 ) {
     val context = LocalContext.current
+    var showRationaleDialogForPhotoReasoning by rememberSaveable { mutableStateOf(false) }
     val menuItems = listOf(
         MenuItem("photo_reasoning", R.string.menu_reason_title, R.string.menu_reason_description)
     )
@@ -186,7 +190,26 @@ fun MenuScreen(
                             if (isTrialExpired) {
                                 Toast.makeText(context, "Please subscribe to the app to continue.", Toast.LENGTH_LONG).show()
                             } else {
-                                onItemClicked(menuItem.routeId)
+                                if (menuItem.routeId == "photo_reasoning") {
+                                    val mainActivity = context as? MainActivity
+                                    if (mainActivity != null && !mainActivity.isNotificationPermissionGranted()) {
+                                        // Check if rationale should be shown
+                                        if (mainActivity.shouldShowRequestPermissionRationale(Manifest.permission.POST_NOTIFICATIONS) && !mainActivity.hasShownNotificationRationale()) {
+                                            showRationaleDialogForPhotoReasoning = true
+                                            // onItemClicked will be called from the dialog's OK button or if rationale is not shown
+                                        } else {
+                                            // Rationale not needed or already shown and dismissed, directly request permission
+                                            mainActivity.requestNotificationPermission()
+                                            onItemClicked(menuItem.routeId) // Proceed to navigate
+                                        }
+                                    } else {
+                                        // Permission already granted or not on Android 13+ or mainActivity is null
+                                        onItemClicked(menuItem.routeId) // Proceed to navigate
+                                    }
+                                } else {
+                                    // For other menu items, navigate directly
+                                    onItemClicked(menuItem.routeId)
+                                }
                             }
                         },
                         enabled = !isTrialExpired, // Disable button if trial is expired
@@ -273,6 +296,42 @@ fun MenuScreen(
             }
         }
     }
+
+    if (showRationaleDialogForPhotoReasoning) {
+        val mainActivity = LocalContext.current as? MainActivity
+        AlertDialog(
+            onDismissRequest = {
+                showRationaleDialogForPhotoReasoning = false
+                // If dismissed, still proceed to the item, permission will be asked by OS if needed later by other flows, or user can try again.
+                // Or, we can choose not to navigate if they dismiss. For now, let's navigate.
+                 mainActivity?.let { onItemClicked("photo_reasoning") }
+            },
+            title = { Text("Notification Permission") },
+            text = { Text("You can grant notification permission if you want to be able to stop Screen Operator via notifications.") },
+            confirmButton = {
+                TextButton(
+                    onClick = {
+                        showRationaleDialogForPhotoReasoning = false
+                        mainActivity?.setNotificationRationaleShown(true)
+                        mainActivity?.requestNotificationPermission()
+                        mainActivity?.let { onItemClicked("photo_reasoning") } // Proceed to navigate
+                    }
+                ) {
+                    Text("OK")
+                }
+            },
+            dismissButton = {
+                TextButton(
+                    onClick = {
+                        showRationaleDialogForPhotoReasoning = false
+                        mainActivity?.let { onItemClicked("photo_reasoning") } // Proceed to navigate even on cancel
+                    }
+                ) {
+                    Text("Cancel")
+                }
+            }
+        )
+    }
 }
 
 @Preview(showSystemUi = true)

app/src/main/kotlin/com/google/ai/sample/feature/multimodal/PhotoReasoningScreen.kt

@@ -146,7 +146,7 @@ internal fun PhotoReasoningRoute(
 
     // Hoisted: var showNotificationRationaleDialog by rememberSaveable { mutableStateOf(false) }
     // This state will now be managed in PhotoReasoningRoute and passed down.
-    var showNotificationRationaleDialogStateInRoute by rememberSaveable { mutableStateOf(false) }
+    // var showNotificationRationaleDialogStateInRoute by rememberSaveable { mutableStateOf(false) } // Removed
 
 
     val coroutineScope = rememberCoroutineScope()
@@ -208,8 +208,8 @@ internal fun PhotoReasoningRoute(
         },
         isKeyboardOpen = isKeyboardOpen,
         onStopClicked = { viewModel.onStopClicked() },
-        showNotificationRationaleDialog = showNotificationRationaleDialogStateInRoute,
-        onShowNotificationRationaleDialogChange = { showNotificationRationaleDialogStateInRoute = it },
+        // showNotificationRationaleDialog = showNotificationRationaleDialogStateInRoute, // Removed
+        // onShowNotificationRationaleDialogChange = { showNotificationRationaleDialogStateInRoute = it }, // Removed
         isInitialized = isInitialized // Pass the collected state
     )
 }
@@ -228,8 +228,8 @@ fun PhotoReasoningScreen(
     onClearChatHistory: () -> Unit = {},
     isKeyboardOpen: Boolean,
     onStopClicked: () -> Unit = {},
-    showNotificationRationaleDialog: Boolean, // New parameter
-    onShowNotificationRationaleDialogChange: (Boolean) -> Unit, // New parameter
+    // showNotificationRationaleDialog: Boolean, // Removed
+    // onShowNotificationRationaleDialogChange: (Boolean) -> Unit, // Removed
     isInitialized: Boolean = true // Added parameter with default for preview
 ) {
     var userQuestion by rememberSaveable { mutableStateOf("") }
@@ -363,37 +363,14 @@ fun PhotoReasoningScreen(
                         modifier = Modifier.weight(1f).padding(end = 8.dp)
                     )
                     IconButton(onClick = {
-                        val activity = context as? MainActivity
-                        if (activity != null && !activity.isNotificationPermissionGranted()) {
-                            if (!activity.hasShownNotificationRationale()) {
-                                onShowNotificationRationaleDialogChange(true) // Use new callback
-                            } else {
-                                // Rationale already shown, or user previously denied.
-                                // Directly request permission or decide if reasoning should be blocked.
-                                // For now, let's request again. System handles "don't ask again".
-                                activity.requestNotificationPermission()
-                                // Optionally, delay onReasonClicked until permission result,
-                                // but for now, let it proceed as notification is a value-add.
-                                 if (isAccessibilityServiceEnabled) {
-                                    if (userQuestion.isNotBlank()) {
-                                        onReasonClicked(userQuestion, imageUris.toList())
-                                        userQuestion = ""
-                                    }
-                                } else {
-                                    onEnableAccessibilityService()
-                                    Toast.makeText(context, "Enable the Accessibility service for Screen Operator", Toast.LENGTH_LONG).show()
-                                }
-                            }
-                        } else { // Permission granted or not needed (older OS)
-                            if (isAccessibilityServiceEnabled) {
-                                if (userQuestion.isNotBlank()) {
-                                    onReasonClicked(userQuestion, imageUris.toList())
-                                    userQuestion = ""
-                                }
-                            } else {
-                                onEnableAccessibilityService()
-                                Toast.makeText(context, "Enable the Accessibility service for Screen Operator", Toast.LENGTH_LONG).show()
+                        if (isAccessibilityServiceEnabled) {
+                            if (userQuestion.isNotBlank()) {
+                                onReasonClicked(userQuestion, imageUris.toList())
+                                userQuestion = ""
                             }
+                        } else {
+                            onEnableAccessibilityService()
+                            Toast.makeText(context, "Enable the Accessibility service for Screen Operator", Toast.LENGTH_LONG).show()
                         }
                     },
                     enabled = isInitialized && isAccessibilityServiceEnabled && userQuestion.isNotBlank(), // Modified enabled state
@@ -493,40 +470,7 @@ fun PhotoReasoningScreen(
         }
     }
 
-    if (showNotificationRationaleDialog) { // Use new parameter
-        AlertDialog(
-            onDismissRequest = { onShowNotificationRationaleDialogChange(false) }, // Use new callback
-            title = { Text("Notification Permission") },
-            text = { Text("You can grant notification permission if you want to be able to stop Screen Operator via notifications.") },
-            confirmButton = {
-                TextButton(
-                    onClick = {
-                        onShowNotificationRationaleDialogChange(false) // Use new callback
-                        val activity = context as? MainActivity
-                        activity?.setNotificationRationaleShown(true)
-                        activity?.requestNotificationPermission()
-                        // Proceed with reasonClicked after user acknowledges rationale and permission is requested
-                        // This is a choice: either proceed or wait for permission result.
-                        // For simplicity and as it's a value-add, let's allow proceeding.
-                        if (isAccessibilityServiceEnabled) {
-                            if (userQuestion.isNotBlank()) { // Check again, user might have cleared it
-                                onReasonClicked(userQuestion, imageUris.toList())
-                                userQuestion = ""
-                            }
-                        } else {
-                           onEnableAccessibilityService()
-                           Toast.makeText(context, "Enable the Accessibility service for Screen Operator", Toast.LENGTH_LONG).show()
-                        }
-                    }
-                ) {
-                    Text("OK")
-                }
-            },
-            dismissButton = {
-                 TextButton(onClick = { onShowNotificationRationaleDialogChange(false) }) { Text("Cancel") } // Use new callback
-            }
-        )
-    }
+    // Removed AlertDialog for notification rationale
 }
 
 @Composable
@@ -1069,8 +1013,6 @@ fun PhotoReasoningScreenPreviewWithContent() {
             ),
             isKeyboardOpen = false,
             onStopClicked = {},
-            showNotificationRationaleDialog = false,
-            onShowNotificationRationaleDialogChange = {},
             isInitialized = true
         )
     }
@@ -1173,8 +1115,6 @@ fun PhotoReasoningScreenPreviewEmpty() {
         PhotoReasoningScreen(
             isKeyboardOpen = false,
             onStopClicked = {},
-            showNotificationRationaleDialog = false,
-            onShowNotificationRationaleDialogChange = {},
             isInitialized = true
         )
     }