diff --git a/app/src/main/AndroidManifest.xml b/app/src/main/AndroidManifest.xml
index 7db43670..968b3708 100644
--- a/app/src/main/AndroidManifest.xml
+++ b/app/src/main/AndroidManifest.xml
@@ -1,4 +1,5 @@
-<?xml version="1.0" encoding="utf-8"?><!--
+<?xml version="1.0" encoding="utf-8"?>
+<!--
      Copyright 2025 The Android Open Source Project
 
      Licensed under the Apache License, Version 2.0 (the "License");
diff --git a/app/src/main/java/com/android/developers/androidify/navigation/MainNavigation.kt b/app/src/main/java/com/android/developers/androidify/navigation/MainNavigation.kt
index 8eab6f36..2a8fbbc5 100644
--- a/app/src/main/java/com/android/developers/androidify/navigation/MainNavigation.kt
+++ b/app/src/main/java/com/android/developers/androidify/navigation/MainNavigation.kt
@@ -32,6 +32,7 @@ import androidx.compose.runtime.remember
 import androidx.compose.runtime.setValue
 import androidx.compose.ui.platform.LocalContext
 import androidx.compose.ui.unit.IntOffset
+import androidx.hilt.navigation.compose.hiltViewModel
 import androidx.lifecycle.viewmodel.navigation3.rememberViewModelStoreNavEntryDecorator
 import androidx.navigation3.runtime.entry
 import androidx.navigation3.runtime.entryProvider
@@ -39,8 +40,13 @@ import androidx.navigation3.runtime.rememberSavedStateNavEntryDecorator
 import androidx.navigation3.ui.NavDisplay
 import com.android.developers.androidify.camera.CameraPreviewScreen
 import com.android.developers.androidify.creation.CreationScreen
+import com.android.developers.androidify.creation.CreationViewModel
+import com.android.developers.androidify.customize.CustomizeAndExportScreen
+import com.android.developers.androidify.customize.CustomizeExportViewModel
 import com.android.developers.androidify.home.AboutScreen
 import com.android.developers.androidify.home.HomeScreen
+import com.android.developers.androidify.results.ResultsScreen
+import com.android.developers.androidify.results.ResultsViewModel
 import com.android.developers.androidify.theme.transitions.ColorSplashTransitionScreen
 import com.google.android.gms.oss.licenses.OssLicensesMenuActivity
 
@@ -92,14 +98,20 @@ fun MainNavigation() {
                 CameraPreviewScreen(
                     onImageCaptured = { uri ->
                         backStack.removeAll { it is Create }
-                        backStack.add(Create(uri.toString()))
+                        backStack.add(Create(uri))
                         backStack.removeAll { it is Camera }
                     },
                 )
             }
             entry<Create> { createKey ->
+                val creationViewModel = hiltViewModel<CreationViewModel, CreationViewModel.Factory>(
+                    creationCallback = { factory ->
+                        factory.create(
+                            originalImageUrl = createKey.fileName,
+                        )
+                    },
+                )
                 CreationScreen(
-                    createKey.fileName,
                     onCameraPressed = {
                         backStack.removeAll { it is Camera }
                         backStack.add(Camera)
@@ -110,6 +122,64 @@ fun MainNavigation() {
                     onAboutPressed = {
                         backStack.add(About)
                     },
+                    onImageCreated = { resultImageUri, prompt, originalImageUri ->
+                        backStack.removeAll { it is Result }
+                        backStack.add(
+                            Result(
+                                resultImageUri = resultImageUri,
+                                prompt = prompt,
+                                originalImageUri = originalImageUri,
+                            ),
+                        )
+                    },
+                    creationViewModel = creationViewModel,
+                )
+            }
+            entry<Result> { resultKey ->
+                val resultsViewModel = hiltViewModel<ResultsViewModel, ResultsViewModel.Factory>(
+                    creationCallback = { factory ->
+                        factory.create(
+                            resultImageUrl = resultKey.resultImageUri,
+                            originalImageUrl = resultKey.originalImageUri,
+                            promptText = resultKey.prompt,
+                        )
+                    },
+                )
+                ResultsScreen(
+                    onNextPress = { resultImageUri, originalImageUri ->
+                        backStack.add(
+                            CustomizeExport(
+                                resultImageUri = resultImageUri,
+                                originalImageUri = originalImageUri,
+                            ),
+                        )
+                    },
+                    onAboutPress = {
+                        backStack.add(About)
+                    },
+                    onBackPress = {
+                        backStack.removeLastOrNull()
+                    },
+                    viewModel = resultsViewModel,
+                )
+            }
+            entry<CustomizeExport> { shareKey ->
+                val customizeExportViewModel = hiltViewModel<CustomizeExportViewModel, CustomizeExportViewModel.Factory>(
+                    creationCallback = { factory ->
+                        factory.create(
+                            resultImageUrl = shareKey.resultImageUri,
+                            originalImageUrl = shareKey.originalImageUri,
+                        )
+                    },
+                )
+                CustomizeAndExportScreen(
+                    onBackPress = {
+                        backStack.removeLastOrNull()
+                    },
+                    onInfoPress = {
+                        backStack.add(About)
+                    },
+                    viewModel = customizeExportViewModel,
                 )
             }
             entry<About> {
diff --git a/app/src/main/java/com/android/developers/androidify/navigation/NavigationRoutes.kt b/app/src/main/java/com/android/developers/androidify/navigation/NavigationRoutes.kt
index 81c8338f..8322c46f 100644
--- a/app/src/main/java/com/android/developers/androidify/navigation/NavigationRoutes.kt
+++ b/app/src/main/java/com/android/developers/androidify/navigation/NavigationRoutes.kt
@@ -17,6 +17,7 @@
 
 package com.android.developers.androidify.navigation
 
+import android.net.Uri
 import kotlinx.serialization.ExperimentalSerializationApi
 import kotlinx.serialization.Serializable
 
@@ -26,10 +27,39 @@ sealed interface NavigationRoute
 data object Home : NavigationRoute
 
 @Serializable
-data class Create(val fileName: String? = null, val prompt: String? = null) : NavigationRoute
+data class Create(
+    @Serializable(with = UriSerializer::class) val fileName: Uri? = null,
+    val prompt: String? = null,
+) : NavigationRoute
 
 @Serializable
 object Camera : NavigationRoute
 
 @Serializable
 object About : NavigationRoute
+
+/**
+ * Represents the result of an image generation process, used for navigation.
+ *
+ * @param resultImageUri The URI of the generated image.
+ * @param originalImageUri The URI of the original image used as a base for generation, if any.
+ * @param prompt The text prompt used to generate the image, if any.
+ */
+@Serializable
+data class Result(
+    @Serializable(with = UriSerializer::class) val resultImageUri: Uri,
+    @Serializable(with = UriSerializer::class) val originalImageUri: Uri? = null,
+    val prompt: String? = null,
+) : NavigationRoute
+
+/**
+ * Represents the navigation route to the screen for customizing and exporting a generated image.
+ *
+ * @param resultImageUri The URI of the generated image to be customized.
+ * @param originalImageUri The URI of the original image, passed along for context.
+ */
+@Serializable
+data class CustomizeExport(
+    @Serializable(with = UriSerializer::class) val resultImageUri: Uri,
+    @Serializable(with = UriSerializer::class) val originalImageUri: Uri?,
+) : NavigationRoute
diff --git a/app/src/main/java/com/android/developers/androidify/navigation/UriSerializer.kt b/app/src/main/java/com/android/developers/androidify/navigation/UriSerializer.kt
new file mode 100644
index 00000000..69950d33
--- /dev/null
+++ b/app/src/main/java/com/android/developers/androidify/navigation/UriSerializer.kt
@@ -0,0 +1,36 @@
+/*
+ * Copyright 2025 The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.android.developers.androidify.navigation
+
+import android.net.Uri
+import androidx.core.net.toUri
+import kotlinx.serialization.KSerializer
+import kotlinx.serialization.descriptors.PrimitiveKind
+import kotlinx.serialization.descriptors.PrimitiveSerialDescriptor
+import kotlinx.serialization.descriptors.SerialDescriptor
+import kotlinx.serialization.encoding.Decoder
+import kotlinx.serialization.encoding.Encoder
+
+object UriSerializer : KSerializer<Uri> {
+    override val descriptor: SerialDescriptor =
+        PrimitiveSerialDescriptor("Uri", PrimitiveKind.STRING)
+
+    override fun serialize(encoder: Encoder, value: Uri) {
+        encoder.encodeString(value.toString())
+    }
+
+    override fun deserialize(decoder: Decoder): Uri = decoder.decodeString().toUri()
+}
diff --git a/core/network/src/main/java/com/android/developers/androidify/ondevice/LocalSegmentationDataSource.kt b/core/network/src/main/java/com/android/developers/androidify/ondevice/LocalSegmentationDataSource.kt
index d9e96128..e9118cf4 100644
--- a/core/network/src/main/java/com/android/developers/androidify/ondevice/LocalSegmentationDataSource.kt
+++ b/core/network/src/main/java/com/android/developers/androidify/ondevice/LocalSegmentationDataSource.kt
@@ -37,7 +37,7 @@ interface LocalSegmentationDataSource {
 }
 
 class LocalSegmentationDataSourceImpl @Inject constructor(
-    private val moduleInstallClient: ModuleInstallClient
+    private val moduleInstallClient: ModuleInstallClient,
 ) : LocalSegmentationDataSource {
     private val segmenter by lazy {
         val options = SubjectSegmenterOptions.Builder()
@@ -60,7 +60,7 @@ class LocalSegmentationDataSourceImpl @Inject constructor(
         return areModulesAvailable
     }
     private class CustomInstallStatusListener(
-        val continuation: CancellableContinuation<Boolean>
+        val continuation: CancellableContinuation<Boolean>,
     ) : InstallStatusListener {
 
         override fun onInstallStatusUpdated(update: ModuleInstallStatusUpdate) {
@@ -70,7 +70,7 @@ class LocalSegmentationDataSourceImpl @Inject constructor(
                 continuation.resume(true)
             } else if (update.installState == STATE_FAILED || update.installState == STATE_CANCELED) {
                 continuation.resumeWithException(
-                    ImageSegmentationException("Module download failed or was canceled. State: ${update.installState}")
+                    ImageSegmentationException("Module download failed or was canceled. State: ${update.installState}"),
                 )
             } else {
                 Log.d("LocalSegmentationDataSource", "State update: ${update.installState}")
@@ -128,4 +128,4 @@ class LocalSegmentationDataSourceImpl @Inject constructor(
     }
 }
 
-class ImageSegmentationException(message: String? = null): Exception(message)
\ No newline at end of file
+class ImageSegmentationException(message: String? = null) : Exception(message)
diff --git a/core/network/src/main/java/com/android/developers/androidify/vertexai/FirebaseAiDataSource.kt b/core/network/src/main/java/com/android/developers/androidify/vertexai/FirebaseAiDataSource.kt
index 20103ff6..fef0f431 100644
--- a/core/network/src/main/java/com/android/developers/androidify/vertexai/FirebaseAiDataSource.kt
+++ b/core/network/src/main/java/com/android/developers/androidify/vertexai/FirebaseAiDataSource.kt
@@ -85,12 +85,12 @@ class FirebaseAiDataSourceImpl @Inject constructor(
         return Firebase.ai(backend = GenerativeBackend.vertexAI()).imagenModel(
             remoteConfigDataSource.imageModelName(),
             safetySettings =
-                ImagenSafetySettings(
-                    safetyFilterLevel = ImagenSafetyFilterLevel.BLOCK_LOW_AND_ABOVE,
-                    // Uses `ALLOW_ADULT` filter since `ALLOW_ALL` requires a special approval
-                    // See https://cloud.google.com/vertex-ai/generative-ai/docs/image/responsible-ai-imagen#person-face-gen
-                    personFilterLevel = ImagenPersonFilterLevel.ALLOW_ADULT,
-                ),
+            ImagenSafetySettings(
+                safetyFilterLevel = ImagenSafetyFilterLevel.BLOCK_LOW_AND_ABOVE,
+                // Uses `ALLOW_ADULT` filter since `ALLOW_ALL` requires a special approval
+                // See https://cloud.google.com/vertex-ai/generative-ai/docs/image/responsible-ai-imagen#person-face-gen
+                personFilterLevel = ImagenPersonFilterLevel.ALLOW_ADULT,
+            ),
         )
     }
 
diff --git a/core/testing/src/main/java/com/android/developers/testing/data/BitmapSample.kt b/core/testing/src/main/java/com/android/developers/testing/data/BitmapSample.kt
new file mode 100644
index 00000000..ceec4d74
--- /dev/null
+++ b/core/testing/src/main/java/com/android/developers/testing/data/BitmapSample.kt
@@ -0,0 +1,20 @@
+/*
+ * Copyright 2025 The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.android.developers.testing.data
+
+import android.graphics.Bitmap
+
+val bitmapSample = Bitmap.createBitmap(1, 1, Bitmap.Config.ARGB_8888)
diff --git a/core/testing/src/main/java/com/android/developers/testing/data/TestFileProvider.kt b/core/testing/src/main/java/com/android/developers/testing/data/TestFileProvider.kt
index bc10d9e7..75761a89 100644
--- a/core/testing/src/main/java/com/android/developers/testing/data/TestFileProvider.kt
+++ b/core/testing/src/main/java/com/android/developers/testing/data/TestFileProvider.kt
@@ -64,4 +64,8 @@ class TestFileProvider : LocalFileProvider {
     ): Uri {
         TODO("Not yet implemented")
     }
+
+    override suspend fun loadBitmapFromUri(uri: Uri): Bitmap? {
+        return bitmapSample
+    }
 }
diff --git a/core/util/src/main/java/com/android/developers/androidify/util/LocalFileProvider.kt b/core/util/src/main/java/com/android/developers/androidify/util/LocalFileProvider.kt
index 150fe769..4c2812d7 100644
--- a/core/util/src/main/java/com/android/developers/androidify/util/LocalFileProvider.kt
+++ b/core/util/src/main/java/com/android/developers/androidify/util/LocalFileProvider.kt
@@ -18,6 +18,7 @@ package com.android.developers.androidify.util
 import android.app.Application
 import android.content.ContentValues
 import android.graphics.Bitmap
+import android.graphics.BitmapFactory
 import android.net.Uri
 import android.os.Build
 import android.os.Environment
@@ -53,6 +54,9 @@ interface LocalFileProvider {
 
     @WorkerThread
     suspend fun saveUriToSharedStorage(inputUri: Uri, fileName: String, mimeType: String): Uri
+
+    @WorkerThread
+    suspend fun loadBitmapFromUri(uri: Uri): Bitmap?
 }
 
 @Singleton
@@ -120,6 +124,20 @@ class LocalFileProviderImpl @Inject constructor(
         return@withContext newUri
     }
 
+    override suspend fun loadBitmapFromUri(uri: Uri): Bitmap? {
+        return withContext(ioDispatcher) {
+            try {
+                application.contentResolver.openInputStream(uri)?.use {
+                    return@withContext BitmapFactory.decodeStream(it)
+                }
+                null
+            } catch (e: Exception) {
+                e.printStackTrace()
+                null
+            }
+        }
+    }
+
     @Throws(IOException::class)
     @WorkerThread
     private fun saveFileToUri(file: File, uri: Uri) {
diff --git a/data/src/main/java/com/android/developers/androidify/data/ImageGenerationRepository.kt b/data/src/main/java/com/android/developers/androidify/data/ImageGenerationRepository.kt
index 7016ee46..9e335077 100644
--- a/data/src/main/java/com/android/developers/androidify/data/ImageGenerationRepository.kt
+++ b/data/src/main/java/com/android/developers/androidify/data/ImageGenerationRepository.kt
@@ -38,7 +38,7 @@ interface ImageGenerationRepository {
     suspend fun saveImageToExternalStorage(imageBitmap: Bitmap): Uri
     suspend fun saveImageToExternalStorage(imageUri: Uri): Uri
 
-    suspend fun addBackgroundToBot(image: Bitmap, backgroundPrompt: String) : Bitmap
+    suspend fun addBackgroundToBot(image: Bitmap, backgroundPrompt: String): Bitmap
     suspend fun removeBackground(image: Bitmap): Bitmap
 }
 
@@ -134,7 +134,7 @@ internal class ImageGenerationRepositoryImpl @Inject constructor(
 
     override suspend fun addBackgroundToBot(image: Bitmap, backgroundPrompt: String): Bitmap {
         val backgroundBotInstructions = remoteConfigDataSource.getBotBackgroundInstructionPrompt() +
-               "\"" +  backgroundPrompt + "\""
+            "\"" + backgroundPrompt + "\""
         return firebaseAiDataSource.generateImageWithEdit(image, backgroundBotInstructions)
     }
 
diff --git a/feature/creation/src/main/java/com/android/developers/androidify/creation/CreationScreen.kt b/feature/creation/src/main/java/com/android/developers/androidify/creation/CreationScreen.kt
index 40767c56..0ed3ec46 100644
--- a/feature/creation/src/main/java/com/android/developers/androidify/creation/CreationScreen.kt
+++ b/feature/creation/src/main/java/com/android/developers/androidify/creation/CreationScreen.kt
@@ -119,18 +119,13 @@ import androidx.compose.ui.text.style.TextAlign
 import androidx.compose.ui.tooling.preview.Preview
 import androidx.compose.ui.unit.dp
 import androidx.compose.ui.unit.sp
-import androidx.core.net.toUri
 import androidx.graphics.shapes.RoundedPolygon
 import androidx.graphics.shapes.rectangle
-import androidx.hilt.navigation.compose.hiltViewModel
 import androidx.lifecycle.compose.collectAsStateWithLifecycle
 import coil3.compose.AsyncImage
 import coil3.request.ImageRequest
 import coil3.request.crossfade
-import com.android.developers.androidify.customize.CustomizeAndExportScreen
-import com.android.developers.androidify.customize.CustomizeExportViewModel
 import com.android.developers.androidify.data.DropBehaviourFactory
-import com.android.developers.androidify.results.ResultsScreen
 import com.android.developers.androidify.theme.AndroidifyTheme
 import com.android.developers.androidify.theme.LimeGreen
 import com.android.developers.androidify.theme.LocalSharedTransitionScope
@@ -158,12 +153,12 @@ import com.android.developers.androidify.creation.R as CreationR
 
 @Composable
 fun CreationScreen(
-    fileName: String? = null,
-    creationViewModel: CreationViewModel = hiltViewModel(),
+    creationViewModel: CreationViewModel,
     isMedium: Boolean = isAtLeastMedium(),
     onCameraPressed: () -> Unit = {},
     onBackPressed: () -> Unit,
     onAboutPressed: () -> Unit,
+    onImageCreated: (resultImageUri: Uri, prompt: String?, originalImageUri: Uri?) -> Unit,
 ) {
     val uiState by creationViewModel.uiState.collectAsStateWithLifecycle()
     BackHandler(
@@ -171,19 +166,28 @@ fun CreationScreen(
     ) {
         creationViewModel.onBackPress()
     }
-    LaunchedEffect(Unit) {
-        if (fileName != null) {
-            creationViewModel.onImageSelected(fileName.toUri())
-        } else {
-            creationViewModel.onImageSelected(null)
-        }
-    }
     val pickMedia = rememberLauncherForActivityResult(PickVisualMedia()) { uri ->
         if (uri != null) {
             creationViewModel.onImageSelected(uri)
         }
     }
     val snackbarHostState by creationViewModel.snackbarHostState.collectAsStateWithLifecycle()
+
+    LaunchedEffect(uiState.resultBitmapUri) {
+        uiState.resultBitmapUri?.let { resultBitmapUri ->
+            onImageCreated(
+                resultBitmapUri,
+                uiState.descriptionText.text.toString(),
+                if (uiState.selectedPromptOption == PromptType.PHOTO) {
+                    uiState.imageUri
+                } else {
+                    null
+                },
+            )
+            creationViewModel.onResultDisplayed()
+        }
+    }
+
     when (uiState.screenState) {
         ScreenState.EDIT -> {
             EditScreen(
@@ -211,46 +215,6 @@ fun CreationScreen(
                 },
             )
         }
-
-        ScreenState.RESULT -> {
-            val prompt = uiState.descriptionText.text.toString()
-            val key = if (uiState.descriptionText.text.isBlank()) {
-                uiState.imageUri.toString()
-            } else {
-                prompt
-            }
-            ResultsScreen(
-                uiState.resultBitmap!!,
-                if (uiState.selectedPromptOption == PromptType.PHOTO) {
-                    uiState.imageUri
-                } else {
-                    null
-                },
-                promptText = prompt,
-                viewModel = hiltViewModel(key = key),
-                onAboutPress = onAboutPressed,
-                onBackPress = onBackPressed,
-                onNextPress = creationViewModel::customizeExportClicked,
-            )
-        }
-
-        ScreenState.CUSTOMIZE -> {
-            val prompt = uiState.descriptionText.text.toString()
-            val key = if (uiState.descriptionText.text.isBlank()) {
-                uiState.imageUri.toString()
-            } else {
-                prompt
-            }
-            uiState.resultBitmap?.let { bitmap ->
-                CustomizeAndExportScreen(
-                    resultImage = bitmap,
-                    originalImageUri = uiState.imageUri,
-                    onBackPress = onBackPressed,
-                    onInfoPress = onAboutPressed,
-                    viewModel = hiltViewModel<CustomizeExportViewModel>(key = key),
-                )
-            }
-        }
     }
 }
 
diff --git a/feature/creation/src/main/java/com/android/developers/androidify/creation/CreationViewModel.kt b/feature/creation/src/main/java/com/android/developers/androidify/creation/CreationViewModel.kt
index b9e34558..632300f2 100644
--- a/feature/creation/src/main/java/com/android/developers/androidify/creation/CreationViewModel.kt
+++ b/feature/creation/src/main/java/com/android/developers/androidify/creation/CreationViewModel.kt
@@ -16,7 +16,6 @@
 package com.android.developers.androidify.creation
 
 import android.content.Context
-import android.graphics.Bitmap
 import android.net.Uri
 import android.util.Log
 import androidx.compose.foundation.text.input.TextFieldState
@@ -34,6 +33,9 @@ import com.android.developers.androidify.data.InternetConnectivityManager
 import com.android.developers.androidify.data.NoInternetException
 import com.android.developers.androidify.data.TextGenerationRepository
 import com.android.developers.androidify.util.LocalFileProvider
+import dagger.assisted.Assisted
+import dagger.assisted.AssistedFactory
+import dagger.assisted.AssistedInject
 import dagger.hilt.android.lifecycle.HiltViewModel
 import dagger.hilt.android.qualifiers.ApplicationContext
 import kotlinx.coroutines.Job
@@ -41,10 +43,10 @@ import kotlinx.coroutines.flow.MutableStateFlow
 import kotlinx.coroutines.flow.StateFlow
 import kotlinx.coroutines.flow.update
 import kotlinx.coroutines.launch
-import javax.inject.Inject
 
-@HiltViewModel
-class CreationViewModel @Inject constructor(
+@HiltViewModel(assistedFactory = CreationViewModel.Factory::class)
+class CreationViewModel @AssistedInject constructor(
+    @Assisted("originalImageUrl") originalImageUrl: Uri?,
     val internetConnectivityManager: InternetConnectivityManager,
     val imageGenerationRepository: ImageGenerationRepository,
     val textGenerationRepository: TextGenerationRepository,
@@ -54,11 +56,9 @@ class CreationViewModel @Inject constructor(
     val context: Context,
 ) : ViewModel() {
 
-    init {
-        viewModelScope.launch {
-            imageGenerationRepository.initialize()
-            textGenerationRepository.initialize()
-        }
+    @AssistedFactory
+    interface Factory {
+        fun create(@Assisted("originalImageUrl") originalImageUrl: Uri?): CreationViewModel
     }
 
     private var _uiState = MutableStateFlow(CreationState())
@@ -74,6 +74,14 @@ class CreationViewModel @Inject constructor(
     private var promptGenerationJob: Job? = null
     private var imageGenerationJob: Job? = null
 
+    init {
+        onImageSelected(originalImageUrl)
+        viewModelScope.launch {
+            imageGenerationRepository.initialize()
+            textGenerationRepository.initialize()
+        }
+    }
+
     fun onImageSelected(uri: Uri?) {
         _uiState.update {
             it.copy(
@@ -153,7 +161,10 @@ class CreationViewModel @Inject constructor(
                         )
                     }
                     _uiState.update {
-                        it.copy(resultBitmap = bitmap, screenState = ScreenState.RESULT)
+                        it.copy(
+                            resultBitmapUri = imageGenerationRepository.saveImage(bitmap),
+                            screenState = ScreenState.EDIT,
+                        )
                     }
                 } catch (e: Exception) {
                     handleImageGenerationError(e)
@@ -218,27 +229,15 @@ class CreationViewModel @Inject constructor(
                 cancelInProgressTask()
             }
 
-            ScreenState.RESULT -> {
-                _uiState.update {
-                    it.copy(screenState = ScreenState.EDIT, resultBitmap = null)
-                }
-            }
-
             ScreenState.EDIT -> {
                 // do nothing, back press handled outside
             }
-
-            ScreenState.CUSTOMIZE -> {
-                _uiState.update {
-                    it.copy(screenState = ScreenState.RESULT)
-                }
-            }
         }
     }
 
-    fun customizeExportClicked() {
+    fun onResultDisplayed() {
         _uiState.update {
-            it.copy(screenState = ScreenState.CUSTOMIZE)
+            it.copy(resultBitmapUri = null)
         }
     }
 }
@@ -252,14 +251,12 @@ data class CreationState(
     val generatedPrompt: String? = null,
     val promptGenerationInProgress: Boolean = false,
     val screenState: ScreenState = ScreenState.EDIT,
-    val resultBitmap: Bitmap? = null,
+    val resultBitmapUri: Uri? = null,
 )
 
 enum class ScreenState {
     EDIT,
     LOADING,
-    RESULT,
-    CUSTOMIZE,
 }
 
 data class BotColor(
diff --git a/feature/creation/src/test/kotlin/com/android/developers/androidify/creation/CreationViewModelTest.kt b/feature/creation/src/test/kotlin/com/android/developers/androidify/creation/CreationViewModelTest.kt
index 11105ce9..af360858 100644
--- a/feature/creation/src/test/kotlin/com/android/developers/androidify/creation/CreationViewModelTest.kt
+++ b/feature/creation/src/test/kotlin/com/android/developers/androidify/creation/CreationViewModelTest.kt
@@ -50,10 +50,12 @@ class CreationViewModelTest {
 
     private val internetConnectivityManager = TestInternetConnectivityManager(true)
     private val imageGenerationRepository = FakeImageGenerationRepository()
+    private val fakeUri = Uri.parse("content://test/image.jpg")
 
     @Before
     fun setup() {
         viewModel = CreationViewModel(
+            originalImageUrl = fakeUri,
             internetConnectivityManager,
             imageGenerationRepository,
             TestTextGenerationRepository(),
@@ -64,18 +66,36 @@ class CreationViewModelTest {
     }
 
     @Test
-    fun stateInitialEdit() = runTest {
+    fun stateInitialEdit_WithImage() = runTest {
         assertEquals(
             ScreenState.EDIT,
             viewModel.uiState.value.screenState,
         )
         assertEquals(false, viewModel.uiState.value.promptGenerationInProgress)
-        assertEquals(null, viewModel.uiState.value.imageUri)
+        assertEquals( fakeUri, viewModel.uiState.value.imageUri)
+    }
+
+    @Test
+    fun stateInitialEdit_WithOutImage() = runTest {
+        viewModel = CreationViewModel(
+            originalImageUrl = null,
+            internetConnectivityManager,
+            imageGenerationRepository,
+            TestTextGenerationRepository(),
+            TestFileProvider(),
+            FakeDropImageFactory(),
+            context = RuntimeEnvironment.getApplication(),
+        )
+        assertEquals(
+            ScreenState.EDIT,
+            viewModel.uiState.value.screenState,
+        )
+        assertEquals(false, viewModel.uiState.value.promptGenerationInProgress)
+        assertEquals( null, viewModel.uiState.value.imageUri)
     }
 
     @Test
     fun onImageSelected_updatesUiState() = runTest {
-        val fakeUri = Uri.parse("content://test/image.jpg")
         viewModel.onImageSelected(fakeUri)
 
         assertEquals(fakeUri, viewModel.uiState.value.imageUri)
@@ -116,8 +136,8 @@ class CreationViewModelTest {
         viewModel.onImageSelected(Uri.parse("content://test/image.jpg"))
         viewModel.onSelectedPromptOptionChanged(PromptType.PHOTO)
         viewModel.startClicked()
-        assertEquals(ScreenState.RESULT, viewModel.uiState.value.screenState)
-        assertNotNull(viewModel.uiState.value.resultBitmap)
+        assertEquals(ScreenState.EDIT, viewModel.uiState.value.screenState)
+        assertNotNull(viewModel.uiState.value.resultBitmapUri)
     }
 
     @Test
@@ -177,8 +197,8 @@ class CreationViewModelTest {
             "testing input description"
         }
         viewModel.startClicked()
-        assertEquals(ScreenState.RESULT, viewModel.uiState.value.screenState)
-        assertNotNull(viewModel.uiState.value.resultBitmap)
+        assertEquals(ScreenState.EDIT, viewModel.uiState.value.screenState)
+        assertNotNull(viewModel.uiState.value.resultBitmapUri)
     }
 
     @Test
diff --git a/feature/results/src/androidTest/java/com/android/developers/androidify/results/ResultsScreenTest.kt b/feature/results/src/androidTest/java/com/android/developers/androidify/results/ResultsScreenTest.kt
index cad23c8c..f478f44b 100644
--- a/feature/results/src/androidTest/java/com/android/developers/androidify/results/ResultsScreenTest.kt
+++ b/feature/results/src/androidTest/java/com/android/developers/androidify/results/ResultsScreenTest.kt
@@ -15,7 +15,6 @@
  */
 package com.android.developers.androidify.results
 
-import android.graphics.Bitmap
 import androidx.activity.ComponentActivity
 import androidx.compose.foundation.layout.PaddingValues
 import androidx.compose.runtime.CompositionLocalProvider
@@ -41,15 +40,15 @@ class ResultsScreenTest {
     @get:Rule
     val composeTestRule = createAndroidComposeRule<ComponentActivity>()
 
-    // Create a dummy bitmap for testing
-    private val testBitmap: Bitmap = Bitmap.createBitmap(100, 100, Bitmap.Config.ARGB_8888)
+    // Create a test bitmap for testing
+    val testUri = android.net.Uri.parse("placeholder://image")
 
     @Test
     fun resultsScreenContents_displaysActionButtons() {
         val shareButtonText = composeTestRule.activity.getString(R.string.customize_and_share)
         // Note: Download button is identified by icon, harder to test reliably without tags/desc
 
-        val initialState = ResultState(resultImageBitmap = testBitmap, promptText = "test")
+        val initialState = ResultState(resultImageUri = testUri, promptText = "test")
         val state = mutableStateOf(initialState)
 
         composeTestRule.setContent {
@@ -78,7 +77,7 @@ class ResultsScreenTest {
         val frontCardDesc = composeTestRule.activity.getString(R.string.resultant_android_bot)
 
         // Ensure promptText is non-null when bitmap is present
-        val initialState = ResultState(resultImageBitmap = testBitmap, promptText = "test")
+        val initialState = ResultState(resultImageUri = testUri, promptText = "test")
         val state = mutableStateOf(initialState)
 
         composeTestRule.setContent {
@@ -106,9 +105,9 @@ class ResultsScreenTest {
         val photoOptionText = composeTestRule.activity.getString(R.string.photo)
         val frontCardDesc = composeTestRule.activity.getString(R.string.resultant_android_bot)
         val backCardDesc = composeTestRule.activity.getString(R.string.original_image)
-        val dummyUri = android.net.Uri.parse("dummy://image")
+        val testUri = android.net.Uri.parse("placeholder://image")
 
-        val initialState = ResultState(resultImageBitmap = testBitmap, originalImageUrl = dummyUri)
+        val initialState = ResultState(resultImageUri = testUri, originalImageUrl = testUri)
         val state = mutableStateOf(initialState)
 
         composeTestRule.setContent {
@@ -143,7 +142,7 @@ class ResultsScreenTest {
         val promptText = "test prompt"
         val promptPrefix = composeTestRule.activity.getString(R.string.my_bot_is_wearing)
 
-        val initialState = ResultState(resultImageBitmap = testBitmap, promptText = promptText) // No original image URI
+        val initialState = ResultState(resultImageUri = testUri, promptText = promptText) // No original image URI
         val state = mutableStateOf(initialState)
 
         composeTestRule.setContent {
@@ -173,9 +172,9 @@ class ResultsScreenTest {
         val botOptionText = composeTestRule.activity.getString(R.string.bot)
         val photoOptionText = composeTestRule.activity.getString(R.string.photo)
         val frontCardDesc = composeTestRule.activity.getString(R.string.resultant_android_bot)
-        val dummyUri = android.net.Uri.parse("dummy://image")
+        val testUri = android.net.Uri.parse("placeholder://image")
 
-        val initialState = ResultState(resultImageBitmap = testBitmap, originalImageUrl = dummyUri)
+        val initialState = ResultState(resultImageUri = testUri, originalImageUrl = testUri)
         val state = mutableStateOf(initialState)
 
         composeTestRule.setContent {
@@ -210,7 +209,7 @@ class ResultsScreenTest {
         var shareClicked = false
 
         // Ensure promptText is non-null when bitmap is present
-        val initialState = ResultState(resultImageBitmap = testBitmap, promptText = "test")
+        val initialState = ResultState(resultImageUri = testUri, promptText = "test")
         val state = mutableStateOf(initialState)
 
         composeTestRule.setContent {
diff --git a/feature/results/src/main/java/com/android/developers/androidify/customize/CustomizeExportScreen.kt b/feature/results/src/main/java/com/android/developers/androidify/customize/CustomizeExportScreen.kt
index 0778e228..c7d213bb 100644
--- a/feature/results/src/main/java/com/android/developers/androidify/customize/CustomizeExportScreen.kt
+++ b/feature/results/src/main/java/com/android/developers/androidify/customize/CustomizeExportScreen.kt
@@ -18,9 +18,6 @@
 package com.android.developers.androidify.customize
 
 import android.Manifest
-import android.R.attr.visible
-import android.graphics.Bitmap
-import android.net.Uri
 import android.os.Build
 import androidx.compose.animation.AnimatedContent
 import androidx.compose.animation.ExperimentalSharedTransitionApi
@@ -61,18 +58,14 @@ import androidx.compose.ui.Alignment
 import androidx.compose.ui.Modifier
 import androidx.compose.ui.draw.clip
 import androidx.compose.ui.draw.dropShadow
-import androidx.compose.ui.graphics.ImageBitmap
-import androidx.compose.ui.graphics.asAndroidBitmap
 import androidx.compose.ui.graphics.shadow.Shadow
 import androidx.compose.ui.graphics.vector.ImageVector
 import androidx.compose.ui.layout.LookaheadScope
 import androidx.compose.ui.platform.LocalContext
-import androidx.compose.ui.res.imageResource
 import androidx.compose.ui.res.stringResource
 import androidx.compose.ui.res.vectorResource
 import androidx.compose.ui.tooling.preview.Preview
 import androidx.compose.ui.unit.dp
-import androidx.hilt.navigation.compose.hiltViewModel
 import androidx.lifecycle.compose.collectAsStateWithLifecycle
 import androidx.navigation3.ui.LocalNavAnimatedContentScope
 import com.android.developers.androidify.results.PermissionRationaleDialog
@@ -97,16 +90,11 @@ import com.android.developers.androidify.theme.R as ThemeR
 @OptIn(ExperimentalMaterial3Api::class)
 @Composable
 fun CustomizeAndExportScreen(
-    resultImage: Bitmap,
-    originalImageUri: Uri?,
     onBackPress: () -> Unit,
     onInfoPress: () -> Unit,
     isMediumWindowSize: Boolean = isAtLeastMedium(),
-    viewModel: CustomizeExportViewModel = hiltViewModel<CustomizeExportViewModel>(),
+    viewModel: CustomizeExportViewModel,
 ) {
-    LaunchedEffect(resultImage, originalImageUri) {
-        viewModel.setArguments(resultImage, originalImageUri)
-    }
     val state = viewModel.state.collectAsStateWithLifecycle()
     val context = LocalContext.current
     LaunchedEffect(state.value.savedUri) {
@@ -430,9 +418,9 @@ fun CustomizeExportPreview() {
         AnimatedContent(true) { targetState ->
             targetState
             CompositionLocalProvider(LocalNavAnimatedContentScope provides this@AnimatedContent) {
-                val bitmap = ImageBitmap.imageResource(R.drawable.placeholderbot)
+                val imageUri = getPlaceholderBotUri()
                 val state = CustomizeExportState(
-                    exportImageCanvas = ExportImageCanvas(imageBitmap = bitmap.asAndroidBitmap()),
+                    exportImageCanvas = ExportImageCanvas(imageUri = imageUri),
                 )
                 CustomizeExportContents(
                     state = state,
@@ -457,10 +445,10 @@ fun CustomizeExportPreviewLarge() {
         AnimatedContent(true) { targetState ->
             targetState
             CompositionLocalProvider(LocalNavAnimatedContentScope provides this@AnimatedContent) {
-                val bitmap = ImageBitmap.imageResource(R.drawable.placeholderbot)
+                val imageUri = getPlaceholderBotUri()
                 val state = CustomizeExportState(
                     exportImageCanvas = ExportImageCanvas(
-                        imageBitmap = bitmap.asAndroidBitmap(),
+                        imageUri = imageUri,
                         aspectRatioOption = SizeOption.Square,
                     ),
                     selectedTool = CustomizeTool.Background,
diff --git a/feature/results/src/main/java/com/android/developers/androidify/customize/CustomizeExportViewModel.kt b/feature/results/src/main/java/com/android/developers/androidify/customize/CustomizeExportViewModel.kt
index 416993da..4b9bd992 100644
--- a/feature/results/src/main/java/com/android/developers/androidify/customize/CustomizeExportViewModel.kt
+++ b/feature/results/src/main/java/com/android/developers/androidify/customize/CustomizeExportViewModel.kt
@@ -25,22 +25,35 @@ import androidx.compose.ui.Modifier
 import androidx.lifecycle.AndroidViewModel
 import androidx.lifecycle.viewModelScope
 import com.android.developers.androidify.data.ImageGenerationRepository
+import com.android.developers.androidify.util.LocalFileProvider
+import dagger.assisted.Assisted
+import dagger.assisted.AssistedFactory
+import dagger.assisted.AssistedInject
 import dagger.hilt.android.lifecycle.HiltViewModel
-import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.flow.MutableStateFlow
 import kotlinx.coroutines.flow.StateFlow
 import kotlinx.coroutines.flow.asStateFlow
 import kotlinx.coroutines.flow.update
 import kotlinx.coroutines.launch
-import javax.inject.Inject
 
-@HiltViewModel
-class CustomizeExportViewModel @Inject constructor(
+@HiltViewModel(assistedFactory = CustomizeExportViewModel.Factory::class)
+class CustomizeExportViewModel @AssistedInject constructor(
+    @Assisted("resultImageUrl") val resultImageUrl: Uri,
+    @Assisted("originalImageUrl") val originalImageUrl: Uri?,
     val imageGenerationRepository: ImageGenerationRepository,
     val composableBitmapRenderer: ComposableBitmapRenderer,
+    val localFileProvider: LocalFileProvider,
     application: Application,
 ) : AndroidViewModel(application) {
 
+    @AssistedFactory
+    interface Factory {
+        fun create(
+            @Assisted("resultImageUrl") resultImageUrl: Uri,
+            @Assisted("originalImageUrl")originalImageUrl: Uri?,
+        ): CustomizeExportViewModel
+    }
+
     private val _state = MutableStateFlow(CustomizeExportState())
     val state = _state.asStateFlow()
 
@@ -49,20 +62,18 @@ class CustomizeExportViewModel @Inject constructor(
     val snackbarHostState: StateFlow<SnackbarHostState>
         get() = _snackbarHostState
 
-    override fun onCleared() {
-        super.onCleared()
-    }
-
-    fun setArguments(
-        resultImageUrl: Bitmap,
-        originalImageUrl: Uri?,
-    ) {
+    init {
         _state.update {
-            CustomizeExportState(
-                originalImageUrl,
-                exportImageCanvas = it.exportImageCanvas.copy(imageBitmap = resultImageUrl),
+            it.copy(
+                originalImageUrl = originalImageUrl,
+                exportImageCanvas = it.exportImageCanvas.copy(imageUri = resultImageUrl),
             )
         }
+        loadInitialBitmap(resultImageUrl)
+    }
+
+    override fun onCleared() {
+        super.onCleared()
     }
 
     fun shareClicked() {
@@ -112,7 +123,7 @@ class CustomizeExportViewModel @Inject constructor(
                 snackbarHostState.value.showSnackbar("Background removal failed")
                 _state.update {
                     val aspectRatioToolState = (it.toolState[CustomizeTool.Size] as AspectRatioToolState)
-                        .copy(selectedToolOption =  previousSizeOption)
+                        .copy(selectedToolOption = previousSizeOption)
                     it.copy(
                         toolState = it.toolState + (CustomizeTool.Size to aspectRatioToolState),
                         showImageEditProgress = false,
@@ -153,7 +164,7 @@ class CustomizeExportViewModel @Inject constructor(
             is SizeOption -> {
                 val selectedSizeOption = toolState.selectedToolOption as SizeOption
                 val needsBackgroundRemoval = selectedSizeOption == SizeOption.Sticker &&
-                        state.value.exportImageCanvas.imageBitmapRemovedBackground == null
+                    state.value.exportImageCanvas.imageBitmapRemovedBackground == null
 
                 val imageBitmap = state.value.exportImageCanvas.imageBitmap
                 if (needsBackgroundRemoval && imageBitmap != null) {
@@ -197,7 +208,6 @@ class CustomizeExportViewModel @Inject constructor(
                 }
                 return@launch
             }
-
             val image = state.value.exportImageCanvas.imageBitmap
             if (image == null) {
                 return@launch
@@ -206,7 +216,8 @@ class CustomizeExportViewModel @Inject constructor(
             _state.update { it.copy(showImageEditProgress = true) }
             try {
                 val bitmap = imageGenerationRepository.addBackgroundToBot(
-                    image, backgroundOption.prompt,
+                    image,
+                    backgroundOption.prompt,
                 )
                 _state.update {
                     it.copy(
@@ -254,4 +265,19 @@ class CustomizeExportViewModel @Inject constructor(
             it.copy(selectedTool = tool)
         }
     }
+
+    private fun loadInitialBitmap(uri: Uri) {
+        viewModelScope.launch {
+            try {
+                val bitmap = localFileProvider.loadBitmapFromUri(uri)
+                _state.update {
+                    it.copy(
+                        exportImageCanvas = it.exportImageCanvas.copy(imageBitmap = bitmap),
+                    )
+                }
+            } catch (e: Exception) {
+                _snackbarHostState.value.showSnackbar("Could not load image.")
+            }
+        }
+    }
 }
diff --git a/feature/results/src/main/java/com/android/developers/androidify/customize/CustomizeState.kt b/feature/results/src/main/java/com/android/developers/androidify/customize/CustomizeState.kt
index 07319489..588c33e6 100644
--- a/feature/results/src/main/java/com/android/developers/androidify/customize/CustomizeState.kt
+++ b/feature/results/src/main/java/com/android/developers/androidify/customize/CustomizeState.kt
@@ -75,6 +75,7 @@ data class BackgroundToolState(
 ) : ToolState
 
 data class ExportImageCanvas(
+    val imageUri: Uri? = null,
     val imageBitmap: Bitmap? = null,
     val imageBitmapRemovedBackground: Bitmap? = null,
     val aspectRatioOption: SizeOption = SizeOption.Square,
diff --git a/feature/results/src/main/java/com/android/developers/androidify/customize/ImageRenderer.kt b/feature/results/src/main/java/com/android/developers/androidify/customize/ImageRenderer.kt
index b65673a3..dc26cac7 100644
--- a/feature/results/src/main/java/com/android/developers/androidify/customize/ImageRenderer.kt
+++ b/feature/results/src/main/java/com/android/developers/androidify/customize/ImageRenderer.kt
@@ -38,14 +38,13 @@ import androidx.compose.ui.draw.rotate
 import androidx.compose.ui.geometry.Rect
 import androidx.compose.ui.geometry.Size
 import androidx.compose.ui.graphics.ImageBitmap
-import androidx.compose.ui.graphics.asAndroidBitmap
 import androidx.compose.ui.graphics.asImageBitmap
 import androidx.compose.ui.layout.ContentScale
 import androidx.compose.ui.layout.layout
 import androidx.compose.ui.res.imageResource
 import androidx.compose.ui.tooling.preview.Preview
 import androidx.compose.ui.util.fastRoundToInt
-import com.android.developers.androidify.results.R
+import coil3.compose.AsyncImage
 import com.android.developers.androidify.theme.AndroidifyTheme
 import com.android.developers.androidify.theme.LocalAnimateBoundsScope
 @OptIn(ExperimentalMaterial3ExpressiveApi::class, ExperimentalSharedTransitionApi::class)
@@ -91,9 +90,9 @@ fun ImageResult(
                         contentScale = ContentScale.Crop,
                         contentDescription = null,
                     )
-                } else if (exportImageCanvas.imageBitmap != null) {
-                    Image(
-                        bitmap = exportImageCanvas.imageBitmap.asImageBitmap(),
+                } else if (exportImageCanvas.imageUri != null) {
+                    AsyncImage(
+                        model = exportImageCanvas.imageUri,
                         modifier = Modifier
                             .fillMaxSize(),
                         contentScale = ContentScale.Crop,
@@ -172,8 +171,9 @@ fun BackgroundLayout(
                 .then(safeAnimateBounds)
                 .rotate(rotationAnimation),
         ) {
-            val clip = if (exportImageCanvas.selectedBackgroundOption == BackgroundOption.None
-                || exportImageCanvas.selectedBackgroundOption.aiBackground) {
+            val clip = if (exportImageCanvas.selectedBackgroundOption == BackgroundOption.None ||
+                exportImageCanvas.selectedBackgroundOption.aiBackground
+            ) {
                 Modifier
             } else {
                 Modifier.clip(RoundedCornerShape(6))
@@ -208,12 +208,12 @@ private fun Modifier.safeAnimateBounds(): Modifier {
 @Preview
 @Composable
 private fun ImageRendererPreviewSquare() {
-    val bitmap = ImageBitmap.imageResource(R.drawable.placeholderbot)
+    val imageUri = getPlaceholderBotUri()
 
     AndroidifyTheme {
         ImageResult(
             ExportImageCanvas(
-                imageBitmap = bitmap.asAndroidBitmap(),
+                imageUri = imageUri,
                 canvasSize = Size(1000f, 1000f),
                 aspectRatioOption = SizeOption.Square,
                 selectedBackgroundOption = BackgroundOption.IO,
@@ -231,11 +231,11 @@ private fun ImageRendererPreviewSquare() {
 @Preview
 @Composable
 private fun ImageRendererPreviewBanner() {
-    val bitmap = ImageBitmap.imageResource(R.drawable.placeholderbot)
+    val imageUri = getPlaceholderBotUri()
     AndroidifyTheme {
         ImageResult(
             ExportImageCanvas(
-                imageBitmap = bitmap.asAndroidBitmap(),
+                imageUri = imageUri,
                 canvasSize = Size(1000f, 1000f),
                 aspectRatioOption = SizeOption.Banner,
                 selectedBackgroundOption = BackgroundOption.Lightspeed,
@@ -253,11 +253,11 @@ private fun ImageRendererPreviewBanner() {
 @Preview
 @Composable
 private fun ImageRendererPreviewWallpaper() {
-    val bitmap = ImageBitmap.imageResource(R.drawable.placeholderbot)
+    val imageUri = getPlaceholderBotUri()
     AndroidifyTheme {
         ImageResult(
             ExportImageCanvas(
-                imageBitmap = bitmap.asAndroidBitmap(),
+                imageUri = imageUri,
                 canvasSize = Size(1000f, 1000f),
                 aspectRatioOption = SizeOption.Wallpaper,
                 selectedBackgroundOption = BackgroundOption.Lightspeed,
@@ -275,11 +275,11 @@ private fun ImageRendererPreviewWallpaper() {
 @Preview(widthDp = 1280, heightDp = 800)
 @Composable
 private fun ImageRendererPreviewWallpaperTablet() {
-    val bitmap = ImageBitmap.imageResource(R.drawable.placeholderbot)
+    val imageUri = getPlaceholderBotUri()
     AndroidifyTheme {
         ImageResult(
             ExportImageCanvas(
-                imageBitmap = bitmap.asAndroidBitmap(),
+                imageUri = imageUri,
                 canvasSize = Size(1280f, 800f),
                 aspectRatioOption = SizeOption.WallpaperTablet,
                 selectedBackgroundOption = BackgroundOption.Lightspeed,
@@ -297,11 +297,11 @@ private fun ImageRendererPreviewWallpaperTablet() {
 @Preview
 @Composable
 private fun ImageRendererPreviewWallpaperSocial() {
-    val bitmap = ImageBitmap.imageResource(R.drawable.placeholderbot)
+    val imageUri = getPlaceholderBotUri()
     AndroidifyTheme {
         ImageResult(
             ExportImageCanvas(
-                imageBitmap = bitmap.asAndroidBitmap(),
+                imageUri = imageUri,
                 canvasSize = Size(1600f, 900f),
                 aspectRatioOption = SizeOption.SocialHeader,
                 selectedBackgroundOption = BackgroundOption.Lightspeed,
@@ -319,11 +319,11 @@ private fun ImageRendererPreviewWallpaperSocial() {
 @Preview
 @Composable
 fun ImageRendererPreviewWallpaperIO() {
-    val bitmap = ImageBitmap.imageResource(R.drawable.placeholderbot)
+    val imageUri = getPlaceholderBotUri()
     AndroidifyTheme {
         ImageResult(
             ExportImageCanvas(
-                imageBitmap = bitmap.asAndroidBitmap(),
+                imageUri = imageUri,
                 canvasSize = Size(1600f, 900f),
                 aspectRatioOption = SizeOption.SocialHeader,
                 selectedBackgroundOption = BackgroundOption.IO,
diff --git a/feature/results/src/main/java/com/android/developers/androidify/customize/Utils.kt b/feature/results/src/main/java/com/android/developers/androidify/customize/Utils.kt
new file mode 100644
index 00000000..40886c77
--- /dev/null
+++ b/feature/results/src/main/java/com/android/developers/androidify/customize/Utils.kt
@@ -0,0 +1,27 @@
+/*
+ * Copyright 2025 The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.android.developers.androidify.customize
+
+import android.content.ContentResolver
+import android.net.Uri
+import androidx.compose.runtime.Composable
+import androidx.compose.ui.platform.LocalContext
+import androidx.core.net.toUri
+import com.android.developers.androidify.results.R
+
+@Composable
+fun getPlaceholderBotUri(): Uri =
+    ("${ContentResolver.SCHEME_ANDROID_RESOURCE}://${LocalContext.current.packageName}/${R.drawable.placeholderbot}").toUri()
diff --git a/feature/results/src/main/java/com/android/developers/androidify/results/BotResultCard.kt b/feature/results/src/main/java/com/android/developers/androidify/results/BotResultCard.kt
index 0934b485..3f4139be 100644
--- a/feature/results/src/main/java/com/android/developers/androidify/results/BotResultCard.kt
+++ b/feature/results/src/main/java/com/android/developers/androidify/results/BotResultCard.kt
@@ -15,7 +15,6 @@
  */
 package com.android.developers.androidify.results
 
-import android.graphics.Bitmap
 import android.net.Uri
 import androidx.compose.foundation.background
 import androidx.compose.foundation.layout.Arrangement
@@ -50,7 +49,7 @@ import coil3.compose.AsyncImage
 
 @Composable
 fun BotResultCard(
-    resultImage: Bitmap,
+    resultImageUri: Uri,
     originalImageUrl: Uri?,
     promptText: String?,
     flippableState: FlippableState,
@@ -66,7 +65,7 @@ fun BotResultCard(
         flippableState = flippableState,
         onFlipStateChanged = onFlipStateChanged,
         front = {
-            FrontCard(resultImage)
+            FrontCard(resultImageUri)
         },
         back = {
             if (originalImageUrl != null) {
@@ -79,9 +78,9 @@ fun BotResultCard(
 }
 
 @Composable
-private fun FrontCard(bitmap: Bitmap) {
+private fun FrontCard(resultImageUri: Uri) {
     AsyncImage(
-        model = bitmap,
+        model = resultImageUri,
         contentDescription = stringResource(R.string.resultant_android_bot),
         contentScale = ContentScale.Crop,
         modifier = Modifier
diff --git a/feature/results/src/main/java/com/android/developers/androidify/results/ResultsScreen.kt b/feature/results/src/main/java/com/android/developers/androidify/results/ResultsScreen.kt
index a6c2036c..589ce799 100644
--- a/feature/results/src/main/java/com/android/developers/androidify/results/ResultsScreen.kt
+++ b/feature/results/src/main/java/com/android/developers/androidify/results/ResultsScreen.kt
@@ -17,7 +17,6 @@
 
 package com.android.developers.androidify.results
 
-import android.graphics.Bitmap
 import android.net.Uri
 import androidx.compose.animation.AnimatedVisibility
 import androidx.compose.animation.core.EaseOutBack
@@ -44,7 +43,6 @@ import androidx.compose.material3.SnackbarDefaults
 import androidx.compose.material3.SnackbarHost
 import androidx.compose.material3.Text
 import androidx.compose.runtime.Composable
-import androidx.compose.runtime.LaunchedEffect
 import androidx.compose.runtime.State
 import androidx.compose.runtime.getValue
 import androidx.compose.runtime.mutableStateOf
@@ -52,11 +50,8 @@ import androidx.compose.runtime.remember
 import androidx.compose.runtime.setValue
 import androidx.compose.ui.Alignment
 import androidx.compose.ui.Modifier
-import androidx.compose.ui.graphics.ImageBitmap
-import androidx.compose.ui.graphics.asAndroidBitmap
 import androidx.compose.ui.graphics.vector.ImageVector
 import androidx.compose.ui.platform.LocalInspectionMode
-import androidx.compose.ui.res.imageResource
 import androidx.compose.ui.res.stringArrayResource
 import androidx.compose.ui.res.stringResource
 import androidx.compose.ui.res.vectorResource
@@ -64,8 +59,8 @@ import androidx.compose.ui.text.font.FontWeight.Companion.Bold
 import androidx.compose.ui.tooling.preview.Preview
 import androidx.compose.ui.unit.dp
 import androidx.compose.ui.unit.sp
-import androidx.hilt.navigation.compose.hiltViewModel
 import androidx.lifecycle.compose.collectAsStateWithLifecycle
+import com.android.developers.androidify.customize.getPlaceholderBotUri
 import com.android.developers.androidify.theme.AndroidifyTheme
 import com.android.developers.androidify.theme.components.AndroidifyTopAppBar
 import com.android.developers.androidify.theme.components.PrimaryButton
@@ -78,20 +73,14 @@ import com.google.accompanist.permissions.ExperimentalPermissionsApi
 
 @Composable
 fun ResultsScreen(
-    resultImage: Bitmap,
-    originalImageUri: Uri?,
-    promptText: String?,
     modifier: Modifier = Modifier,
     verboseLayout: Boolean = allowsFullContent(),
     onBackPress: () -> Unit,
     onAboutPress: () -> Unit,
-    onNextPress: () -> Unit,
-    viewModel: ResultsViewModel = hiltViewModel<ResultsViewModel>(),
+    onNextPress: (resultImageUri: Uri, originalImageUri: Uri?) -> Unit,
+    viewModel: ResultsViewModel,
 ) {
     val state = viewModel.state.collectAsStateWithLifecycle()
-    LaunchedEffect(resultImage, originalImageUri, promptText) {
-        viewModel.setArguments(resultImage, originalImageUri, promptText)
-    }
     val snackbarHostState by viewModel.snackbarHostState.collectAsStateWithLifecycle()
     Scaffold(
         snackbarHost = {
@@ -121,7 +110,14 @@ fun ResultsScreen(
             contentPadding,
             state,
             verboseLayout = verboseLayout,
-            onCustomizeShareClicked = onNextPress,
+            onCustomizeShareClicked = {
+                viewModel.state.value.resultImageUri?.let {
+                    onNextPress(
+                        it,
+                        viewModel.state.value.originalImageUrl,
+                    )
+                }
+            },
         )
     }
 }
@@ -132,11 +128,11 @@ fun ResultsScreen(
 @Composable
 private fun ResultsScreenPreview() {
     AndroidifyTheme {
-        val bitmap = ImageBitmap.imageResource(R.drawable.placeholderbot)
+        val imageUri = getPlaceholderBotUri()
         val state = remember {
             mutableStateOf(
                 ResultState(
-                    resultImageBitmap = bitmap.asAndroidBitmap(),
+                    resultImageUri = imageUri,
                     promptText = "wearing a hat with straw hair",
                 ),
             )
@@ -154,11 +150,11 @@ private fun ResultsScreenPreview() {
 @Composable
 private fun ResultsScreenPreviewSmall() {
     AndroidifyTheme {
-        val bitmap = ImageBitmap.imageResource(R.drawable.placeholderbot)
+        val imageUri = getPlaceholderBotUri()
         val state = remember {
             mutableStateOf(
                 ResultState(
-                    resultImageBitmap = bitmap.asAndroidBitmap(),
+                    resultImageUri = imageUri,
                     promptText = "wearing a hat with straw hair",
                 ),
             )
@@ -182,7 +178,7 @@ fun ResultsScreenContents(
     defaultSelectedResult: ResultOption = ResultOption.ResultImage,
 ) {
     ResultsBackground()
-    val showResult = state.value.resultImageBitmap != null
+    val showResult = state.value.resultImageUri != null
     var selectedResultOption by remember {
         mutableStateOf(defaultSelectedResult)
     }
@@ -210,7 +206,7 @@ fun ResultsScreenContents(
                     .fillMaxSize(),
             ) {
                 BotResultCard(
-                    state.value.resultImageBitmap!!,
+                    state.value.resultImageUri!!,
                     state.value.originalImageUrl,
                     state.value.promptText,
                     modifier = Modifier.align(Alignment.Center),
diff --git a/feature/results/src/main/java/com/android/developers/androidify/results/ResultsViewModel.kt b/feature/results/src/main/java/com/android/developers/androidify/results/ResultsViewModel.kt
index a54af8ea..5005ecc4 100644
--- a/feature/results/src/main/java/com/android/developers/androidify/results/ResultsViewModel.kt
+++ b/feature/results/src/main/java/com/android/developers/androidify/results/ResultsViewModel.kt
@@ -15,20 +15,33 @@
  */
 package com.android.developers.androidify.results
 
-import android.graphics.Bitmap
 import android.net.Uri
 import androidx.compose.material3.SnackbarHostState
 import androidx.lifecycle.ViewModel
+import dagger.assisted.Assisted
+import dagger.assisted.AssistedFactory
+import dagger.assisted.AssistedInject
 import dagger.hilt.android.lifecycle.HiltViewModel
 import kotlinx.coroutines.flow.MutableStateFlow
 import kotlinx.coroutines.flow.StateFlow
 import kotlinx.coroutines.flow.asStateFlow
 import kotlinx.coroutines.flow.update
-import javax.inject.Inject
 
-@HiltViewModel
-class ResultsViewModel @Inject constructor() : ViewModel() {
+@HiltViewModel(assistedFactory = ResultsViewModel.Factory::class)
+class ResultsViewModel @AssistedInject constructor(
+    @Assisted("resultImageUrl") val resultImageUrl: Uri?,
+    @Assisted("originalImageUrl") val originalImageUrl: Uri?,
+    @Assisted("promptText") val promptText: String?,
+) : ViewModel() {
 
+    @AssistedFactory
+    interface Factory {
+        fun create(
+            @Assisted("resultImageUrl") resultImageUrl: Uri?,
+            @Assisted("originalImageUrl") originalImageUrl: Uri?,
+            @Assisted("promptText") promptText: String?,
+        ): ResultsViewModel
+    }
     private val _state = MutableStateFlow(ResultState())
     val state = _state.asStateFlow()
 
@@ -37,11 +50,7 @@ class ResultsViewModel @Inject constructor() : ViewModel() {
     val snackbarHostState: StateFlow<SnackbarHostState>
         get() = _snackbarHostState
 
-    fun setArguments(
-        resultImageUrl: Bitmap,
-        originalImageUrl: Uri?,
-        promptText: String?,
-    ) {
+    init {
         _state.update {
             ResultState(resultImageUrl, originalImageUrl, promptText = promptText)
         }
@@ -49,7 +58,7 @@ class ResultsViewModel @Inject constructor() : ViewModel() {
 }
 
 data class ResultState(
-    val resultImageBitmap: Bitmap? = null,
+    val resultImageUri: Uri? = null,
     val originalImageUrl: Uri? = null,
     val promptText: String? = null,
 )
diff --git a/feature/results/src/screenshotTest/java/com/android/developers/androidify/results/ResultsScreenScreenshotTest.kt b/feature/results/src/screenshotTest/java/com/android/developers/androidify/results/ResultsScreenScreenshotTest.kt
index f03cfbd3..56e6b586 100644
--- a/feature/results/src/screenshotTest/java/com/android/developers/androidify/results/ResultsScreenScreenshotTest.kt
+++ b/feature/results/src/screenshotTest/java/com/android/developers/androidify/results/ResultsScreenScreenshotTest.kt
@@ -29,6 +29,7 @@ import androidx.compose.runtime.remember
 import androidx.compose.ui.platform.LocalInspectionMode
 import androidx.compose.ui.tooling.preview.Preview
 import androidx.compose.ui.unit.dp
+import androidx.core.net.toUri
 import com.android.developers.androidify.theme.AndroidifyTheme
 import com.android.developers.androidify.util.AdaptivePreview
 import com.android.developers.androidify.util.SmallPhonePreview
@@ -43,7 +44,7 @@ class ResultsScreenScreenshotTest {
         val state = remember {
             mutableStateOf(
                 ResultState(
-                    resultImageBitmap = mockBitmap,
+                    resultImageUri = "test://mockbitmap/${mockBitmap.hashCode()}".toUri(),
                     promptText = "wearing a hat with straw hair",
                 ),
             )
@@ -68,7 +69,7 @@ class ResultsScreenScreenshotTest {
         val state = remember {
             mutableStateOf(
                 ResultState(
-                    resultImageBitmap = mockBitmap,
+                    resultImageUri = "test://mockbitmap/${mockBitmap.hashCode()}".toUri(),
                     promptText = "wearing a hat with straw hair",
                 ),
             )
@@ -92,7 +93,7 @@ class ResultsScreenScreenshotTest {
         val state = remember {
             mutableStateOf(
                 ResultState(
-                    resultImageBitmap = mockBitmap,
+                    resultImageUri = "test://mockbitmap/${mockBitmap.hashCode()}".toUri(),
                     promptText = "wearing a hat with straw hair",
                 ),
             )
diff --git a/feature/results/src/test/kotlin/com/android/developers/androidify/customize/CustomizeStateTest.kt b/feature/results/src/test/kotlin/com/android/developers/androidify/customize/CustomizeStateTest.kt
index ecaf65d0..87e7481e 100644
--- a/feature/results/src/test/kotlin/com/android/developers/androidify/customize/CustomizeStateTest.kt
+++ b/feature/results/src/test/kotlin/com/android/developers/androidify/customize/CustomizeStateTest.kt
@@ -53,7 +53,7 @@ class CustomizeStateTest {
                 SizeOption.WallpaperTablet,
                 SizeOption.Banner,
                 SizeOption.SocialHeader,
-                SizeOption.Sticker
+                SizeOption.Sticker,
             ),
             state.options,
         )
@@ -87,7 +87,7 @@ class CustomizeStateTest {
     @Test
     fun exportImageCanvas_defaultValues() {
         val canvas = ExportImageCanvas()
-        Assert.assertNull(canvas.imageBitmap)
+        Assert.assertNull(canvas.imageUri)
         Assert.assertEquals(SizeOption.Square, canvas.aspectRatioOption)
         Assert.assertEquals(Size(1000f, 1000f), canvas.canvasSize)
         Assert.assertNull(canvas.mainImageUri)
diff --git a/feature/results/src/test/kotlin/com/android/developers/androidify/customize/CustomizeViewModelTest.kt b/feature/results/src/test/kotlin/com/android/developers/androidify/customize/CustomizeViewModelTest.kt
index c08d4f77..db0a1fb4 100644
--- a/feature/results/src/test/kotlin/com/android/developers/androidify/customize/CustomizeViewModelTest.kt
+++ b/feature/results/src/test/kotlin/com/android/developers/androidify/customize/CustomizeViewModelTest.kt
@@ -17,9 +17,10 @@
 
 package com.android.developers.androidify.customize
 
-import android.graphics.Bitmap
 import android.net.Uri
 import androidx.test.core.app.ApplicationProvider
+import com.android.developers.testing.data.TestFileProvider
+import com.android.developers.testing.data.bitmapSample
 import com.android.developers.testing.repository.FakeImageGenerationRepository
 import com.android.developers.testing.util.FakeComposableBitmapRenderer
 import com.android.developers.testing.util.MainDispatcherRule
@@ -45,36 +46,34 @@ class CustomizeViewModelTest {
     val mainDispatcherRule = MainDispatcherRule()
 
     private lateinit var viewModel: CustomizeExportViewModel
-
-    private val fakeBitmap = Bitmap.createBitmap(1, 1, Bitmap.Config.ARGB_8888)
     private val originalFakeUri = Uri.parse("content://com.example.app/images/original.jpg")
 
+    private val fakeUri = Uri.parse("content://com.example.app/images/original.jpg")
+
     @Before
     fun setup() {
         viewModel = CustomizeExportViewModel(
+            fakeUri,
+            originalFakeUri,
             FakeImageGenerationRepository(),
             composableBitmapRenderer = FakeComposableBitmapRenderer(),
             application = ApplicationProvider.getApplicationContext(),
+            localFileProvider = TestFileProvider(),
         )
     }
 
     @Test
-    fun stateInitialEmpty() = runTest {
-        assertEquals(
-            CustomizeExportState(),
-            viewModel.state.value,
+    fun stateResultUri_NotNull() = runTest {
+        assertNotNull(
+            viewModel.state.value.exportImageCanvas.imageUri,
         )
     }
 
     @Test
     fun setArgumentsWithOriginalImage() = runTest {
-        viewModel.setArguments(
-            fakeBitmap,
-            originalFakeUri,
-        )
         assertEquals(
             CustomizeExportState(
-                exportImageCanvas = ExportImageCanvas(imageBitmap = fakeBitmap),
+                exportImageCanvas = ExportImageCanvas(imageUri = fakeUri, imageBitmap = bitmapSample),
                 originalImageUrl = originalFakeUri,
             ),
             viewModel.state.value,
@@ -83,13 +82,17 @@ class CustomizeViewModelTest {
 
     @Test
     fun setArgumentsWithPrompt() = runTest {
-        viewModel.setArguments(
-            fakeBitmap,
+        val viewModel = CustomizeExportViewModel(
+            fakeUri,
             null,
+            FakeImageGenerationRepository(),
+            composableBitmapRenderer = FakeComposableBitmapRenderer(),
+            application = ApplicationProvider.getApplicationContext(),
+            localFileProvider = TestFileProvider(),
         )
         assertEquals(
             CustomizeExportState(
-                exportImageCanvas = ExportImageCanvas(imageBitmap = fakeBitmap),
+                exportImageCanvas = ExportImageCanvas(imageUri = fakeUri, imageBitmap = bitmapSample),
                 originalImageUrl = null,
             ),
             viewModel.state.value,
@@ -105,11 +108,6 @@ class CustomizeViewModelTest {
             }
         }
 
-        viewModel.setArguments(
-            fakeBitmap,
-            originalFakeUri,
-        )
-
         viewModel.downloadClicked()
         assertNotNull(values.last().externalOriginalSavedUri)
         assertEquals(
@@ -127,10 +125,6 @@ class CustomizeViewModelTest {
                 values.add(it)
             }
         }
-        viewModel.setArguments(
-            fakeBitmap,
-            originalFakeUri,
-        )
         advanceUntilIdle()
         viewModel.shareClicked()
         // Ensure all coroutines on the test scheduler complete
@@ -141,9 +135,12 @@ class CustomizeViewModelTest {
     @Test
     fun changeBackground_NotNull() = runTest {
         val viewModel = CustomizeExportViewModel(
+            fakeUri,
+            null,
             FakeImageGenerationRepository(),
             composableBitmapRenderer = FakeComposableBitmapRenderer(),
             application = ApplicationProvider.getApplicationContext(),
+            localFileProvider = TestFileProvider(),
         )
         val values = mutableListOf<CustomizeExportState>()
         // Launch collector on the backgroundScope directly to use runTest's scheduler
@@ -152,10 +149,6 @@ class CustomizeViewModelTest {
                 values.add(it)
             }
         }
-        viewModel.setArguments(
-            fakeBitmap,
-            originalFakeUri,
-        )
         advanceUntilIdle()
         viewModel.selectedToolStateChanged(
             BackgroundToolState(
@@ -182,10 +175,6 @@ class CustomizeViewModelTest {
                 values.add(it)
             }
         }
-        viewModel.setArguments(
-            fakeBitmap,
-            originalFakeUri,
-        )
         advanceUntilIdle()
         viewModel.selectedToolStateChanged(
             BackgroundToolState(
diff --git a/feature/results/src/test/kotlin/com/android/developers/androidify/results/ResultsViewModelTest.kt b/feature/results/src/test/kotlin/com/android/developers/androidify/results/ResultsViewModelTest.kt
index 5f425658..43bf653c 100644
--- a/feature/results/src/test/kotlin/com/android/developers/androidify/results/ResultsViewModelTest.kt
+++ b/feature/results/src/test/kotlin/com/android/developers/androidify/results/ResultsViewModelTest.kt
@@ -17,13 +17,11 @@
 
 package com.android.developers.androidify.results
 
-import android.graphics.Bitmap
 import android.net.Uri
 import com.android.developers.testing.util.MainDispatcherRule
 import kotlinx.coroutines.ExperimentalCoroutinesApi
 import kotlinx.coroutines.test.runTest
 import org.junit.Assert.assertEquals
-import org.junit.Before
 import org.junit.Rule
 import org.junit.Test
 import org.junit.runner.RunWith
@@ -34,19 +32,14 @@ class ResultsViewModelTest {
     @get:Rule
     val mainDispatcherRule = MainDispatcherRule()
 
-    private lateinit var viewModel: ResultsViewModel
-
-    private val fakeBitmap = Bitmap.createBitmap(1, 1, Bitmap.Config.ARGB_8888)
     private val fakePromptText = "Pink Hair, plaid shirt, jeans"
     private val originalFakeUri = Uri.parse("content://com.example.app/images/original.jpg")
 
-    @Before
-    fun setup() {
-        viewModel = ResultsViewModel()
-    }
+    private val fakeUri = Uri.parse("content://test/image.jpg")
 
     @Test
     fun stateInitialEmpty() = runTest {
+        val viewModel = ResultsViewModel(null, null, null)
         assertEquals(
             ResultState(),
             viewModel.state.value,
@@ -54,15 +47,11 @@ class ResultsViewModelTest {
     }
 
     @Test
-    fun setArgumentsWithOriginalImage() = runTest {
-        viewModel.setArguments(
-            fakeBitmap,
-            originalFakeUri,
-            promptText = null,
-        )
+    fun setArgumentsWithOriginalImage_isCorrect() = runTest {
+        val viewModel = ResultsViewModel(fakeUri, originalFakeUri, null)
         assertEquals(
             ResultState(
-                resultImageBitmap = fakeBitmap,
+                resultImageUri = fakeUri,
                 originalImageUrl = originalFakeUri,
             ),
             viewModel.state.value,
@@ -70,15 +59,11 @@ class ResultsViewModelTest {
     }
 
     @Test
-    fun setArgumentsWithPrompt() = runTest {
-        viewModel.setArguments(
-            fakeBitmap,
-            null,
-            promptText = fakePromptText,
-        )
+    fun initialState_withPrompt_isCorrect() = runTest {
+        val viewModel = ResultsViewModel(fakeUri, null, fakePromptText)
         assertEquals(
             ResultState(
-                resultImageBitmap = fakeBitmap,
+                resultImageUri = fakeUri,
                 originalImageUrl = null,
                 promptText = fakePromptText,
             ),