Fix up review comments

Author: bentrengrove

ai-catalog/samples/imagen/src/main/java/com/android/ai/samples/imagen/ImagenScreen.kt

@@ -80,10 +80,7 @@ fun ImagenScreen(viewModel: ImagenViewModel = hiltViewModel()) {
 
 @Composable
 @OptIn(ExperimentalMaterial3Api::class)
-private fun ImagenScreen(
-    uiState: ImagenUIState,
-    onGenerateClick: (String) -> Unit
-) {
+private fun ImagenScreen(uiState: ImagenUIState, onGenerateClick: (String) -> Unit) {
     val isGenerating = uiState is ImagenUIState.Loading
 
     Scaffold(
@@ -113,15 +110,15 @@ private fun ImagenScreen(
                 uiState = uiState,
                 modifier = Modifier
                     .fillMaxWidth()
-                    .aspectRatio(1f)
+                    .aspectRatio(1f),
             )
 
             Spacer(modifier = Modifier.height(16.dp))
 
             GenerationInput(
                 onGenerateClick = onGenerateClick,
                 enabled = !isGenerating,
-                modifier = Modifier.fillMaxSize(),
+                modifier = Modifier.fillMaxWidth(),
             )
 
             // Ensure the screen scrolls when the keyboard appears
@@ -200,7 +197,7 @@ private fun GenerationInput(onGenerateClick: (String) -> Unit, enabled: Boolean,
             },
             enabled = enabled,
             contentPadding = ButtonDefaults.ButtonWithIconContentPadding,
-            modifier = modifier.fillMaxWidth(),
+            modifier = Modifier.fillMaxWidth(),
         ) {
             Icon(
                 Icons.Default.SmartToy,

ai-catalog/samples/imagen/src/main/java/com/android/ai/samples/imagen/ImagenViewModel.kt

@@ -33,7 +33,10 @@ import kotlinx.coroutines.launch
 sealed interface ImagenUIState {
     data object Initial : ImagenUIState
     data object Loading : ImagenUIState
-    data class ImageGenerated(val bitmap: Bitmap, val contentDescription: String) : ImagenUIState
+    data class ImageGenerated(
+        val bitmap: Bitmap,
+        val contentDescription: String,
+    ) : ImagenUIState
     data class Error(val message: String) : ImagenUIState
 }