Fix ViewModel initialization order and chat history

This commit addresses issues related to the initialization order in PhotoReasoningViewModel:

1. Automatic Initialization:
   - Added an `init` block to PhotoReasoningViewModel to automatically call `initializeViewModel()` upon creation.
   - `initializeViewModel()` loads the system message and chat history, then sets an `_isInitialized` flag.

2. Robust `reason()` Method:
   - The `reason()` method now checks if the ViewModel is initialized.
   - If not initialized, it launches a coroutine to initialize first, then proceeds with the reasoning logic.
   - If already initialized, it proceeds directly.
   - The core logic of `reason()` has been moved to a new private method `performReasoning()`.

This ensures that:
- The ViewModel attempts to initialize as soon as it's created.
- Calls to `reason()` will always wait for initialization to complete if it hasn't happened yet.
- The system message and database entries (loaded during `loadSystemMessage` which calls `loadChatHistory` which calls `rebuildChatHistory`) are consistently included in the chat history before any user interaction that triggers `reason()`.

Author: google-labs-jules[bot]

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

@@ -104,13 +104,23 @@ class PhotoReasoningViewModel(
     private var commandProcessingJob: Job? = null
     private val stopExecutionFlag = AtomicBoolean(false)
 
-    fun reason(
+    init {
+        viewModelScope.launch {
+            initializeViewModel()
+        }
+    }
+
+    private suspend fun initializeViewModel() {
+        loadSystemMessage(MainActivity.getInstance()?.applicationContext!!)
+    }
+
+    private fun performReasoning(
         userInput: String,
         selectedImages: List<Bitmap>,
         screenInfoForPrompt: String? = null,
         imageUrisForChat: List<String>? = null
     ) {
-        Log.d(TAG, "reason() called. User input: '$userInput', Image count: ${selectedImages.size}, ScreenInfo: ${screenInfoForPrompt != null}, ImageUris: ${imageUrisForChat != null}")
+        Log.d(TAG, "performReasoning() 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
@@ -197,6 +207,22 @@ class PhotoReasoningViewModel(
         }
     }
 
+    fun reason(
+        userInput: String,
+        selectedImages: List<Bitmap>,
+        screenInfoForPrompt: String? = null,
+        imageUrisForChat: List<String>? = null
+    ) {
+        if (!_isInitialized.value) {
+            viewModelScope.launch {
+                initializeViewModel()
+                performReasoning(userInput, selectedImages, screenInfoForPrompt, imageUrisForChat)
+            }
+            return
+        }
+        performReasoning(userInput, selectedImages, screenInfoForPrompt, imageUrisForChat)
+    }
+
     fun onStopClicked() {
         _showStopNotificationFlow.value = false // Hide notification immediately on stop
         stopExecutionFlag.set(true)