Fix: Restore screenshot display in chat and consolidate AI prompts.

This commit addresses your feedback following previous loop fixes:
1.  **Restored Screenshot Display in Chat:**
    - Modified `PhotoReasoningViewModel.reason()` to accept an `imageUrisForChat`
      parameter. The `PhotoReasoningMessage` for your turn created
      within `reason()` now uses this parameter to populate its `imageUris`
      field, enabling the UI to display associated images.
    - Updated `PhotoReasoningViewModel.addScreenshotToConversation()` to pass the
      captured screenshot's URI to `reason()` via `imageUrisForChat`.
    - Updated the UI call site in `PhotoReasoningRoute.kt` to pass URIs of
      your selected gallery images to `reason()` via `imageUrisForChat`.
    - This ensures both programmatic and your selected images are displayed
      in the chat again.

2.  **Consolidated `screenInfo` for AI and Chat:**
    - Removed direct chat message creation (e.g., "Screenshot captured...") from
      `addScreenshotToConversation()`.
    - `addScreenshotToConversation()` now calls `reason()` with a generic
      prompt (e.g., "Analyze...") and passes `screenInfo` (from Accessibility
      Service) to the new `screenInfoForPrompt` parameter in `reason()`.
    - `reason()` now combines its `userInput` parameter with `screenInfoForPrompt`
      to form a single `aiPromptText`. This `aiPromptText` is used for your
      message in chat (along with image URIs) and for the textual
      part of the prompt sent to the AI.
    - This prevents `screenInfo` from being duplicated in chat or sent
      redundantly to the AI.

3.  **Improved Screenshot Responsiveness:**
    - In `ScreenOperatorAccessibilityService.kt`, the processing delay for
      `Command.TakeScreenshot` in `scheduleNextCommandProcessing()` was
      reduced from 850ms to 50ms.

These changes ensure screenshots are displayed correctly, AI prompts are
cleaner, and programmatic screenshot actions are more responsive, while
maintaining previous fixes for stability and loop prevention.

Author: google-labs-jules[bot]

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

@@ -189,7 +189,12 @@ internal fun PhotoReasoningRoute(
                         if (result is SuccessResult) (result.drawable as BitmapDrawable).bitmap else null
                     } catch (e: Exception) { null }
                 }
-                viewModel.reason(inputText, bitmaps)
+                viewModel.reason(
+                    userInput = inputText,
+                    selectedImages = bitmaps,
+                    screenInfoForPrompt = null, // User-initiated messages don't have prior screen context here
+                    imageUrisForChat = selectedItems.map { it.toString() }
+                )
             }
         },
         isAccessibilityServiceEnabled = isAccessibilityServiceEffectivelyEnabled,

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

@@ -107,9 +107,10 @@ class PhotoReasoningViewModel(
     fun reason(
         userInput: String,
         selectedImages: List<Bitmap>,
-        screenInfoForPrompt: String? = null
+        screenInfoForPrompt: String? = null,
+        imageUrisForChat: List<String>? = null
     ) {
-        Log.d(TAG, "reason() called. User input: '$userInput', Image count: ${selectedImages.size}, ScreenInfo: ${screenInfoForPrompt != null}")
+        Log.d(TAG, "reason() called. User input: '$userInput', Image count: ${selectedImages.size}, ScreenInfo: ${screenInfoForPrompt != null}, ImageUris: ${imageUrisForChat != null}")
         _uiState.value = PhotoReasoningUiState.Loading
         Log.d(TAG, "Setting _showStopNotificationFlow to true")
         _showStopNotificationFlow.value = true
@@ -136,6 +137,7 @@ class PhotoReasoningViewModel(
         val userMessage = PhotoReasoningMessage(
             text = aiPromptText, // Use the combined text
             participant = PhotoParticipant.USER,
+            imageUris = imageUrisForChat, // Use the new parameter here
             isPending = false
         )
         _chatState.addMessage(userMessage)
@@ -1002,7 +1004,8 @@ class PhotoReasoningViewModel(
                         reason(
                             userInput = genericAnalysisPrompt,
                             selectedImages = listOf(bitmap),
-                            screenInfoForPrompt = screenInfo
+                            screenInfoForPrompt = screenInfo,
+                            imageUrisForChat = listOf(screenshotUri.toString()) // Add this argument
                         )
 
                         // Show a toast to indicate the screenshot was added