missing files

Author: paulb777

firebaseai/ImagenScreen/ImagenFromTemplateScreen.swift

@@ -0,0 +1,99 @@
+// Copyright 2025 Google LLC
+//
+// 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
+//
+//      http://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.
+
+import SwiftUI
+import GenerativeAIUIComponents
+#if canImport(FirebaseAILogic)
+  import FirebaseAILogic
+#else
+  import FirebaseAI
+#endif
+
+struct ImagenFromTemplateScreen: View {
+  let firebaseService: FirebaseAI
+  @StateObject var viewModel: ImagenFromTemplateViewModel
+
+  init(firebaseService: FirebaseAI) {
+    self.firebaseService = firebaseService
+    _viewModel = StateObject(wrappedValue: ImagenFromTemplateViewModel(firebaseService: firebaseService))
+  }
+
+  enum FocusedField: Hashable {
+    case message
+  }
+
+  @FocusState
+  var focusedField: FocusedField?
+
+  var body: some View {
+    ZStack {
+      ScrollView {
+        VStack {
+          InputField("Enter a prompt to generate an image from template", text: $viewModel.userInput) {
+            Image(
+              systemName: viewModel.inProgress ? "stop.circle.fill" : "paperplane.circle.fill"
+            )
+            .font(.title)
+          }
+          .focused($focusedField, equals: .message)
+          .onSubmit { sendOrStop() }
+
+          let spacing: CGFloat = 10
+          LazyVGrid(columns: [
+            GridItem(.flexible(), spacing: spacing),
+            GridItem(.flexible(), spacing: spacing),
+          ], spacing: spacing) {
+            ForEach(viewModel.images, id: \.self) { image in
+              Image(uiImage: image)
+                .resizable()
+                .aspectRatio(1, contentMode: .fill)
+                .cornerRadius(12)
+                .clipped()
+            }
+          }
+          .padding(.horizontal, spacing)
+        }
+      }
+      if viewModel.inProgress {
+        ProgressOverlay()
+      }
+    }
+    .onTapGesture {
+      focusedField = nil
+    }
+    .navigationTitle("Imagen Template")
+    .onAppear {
+      focusedField = .message
+    }
+  }
+
+  private func sendMessage() {
+    Task {
+      await viewModel.generateImageFromTemplate(prompt: viewModel.userInput)
+      focusedField = .message
+    }
+  }
+
+  private func sendOrStop() {
+    if viewModel.inProgress {
+      viewModel.stop()
+    } else {
+      sendMessage()
+    }
+  }
+}
+
+#Preview {
+  ImagenFromTemplateScreen(firebaseService: FirebaseAI.firebaseAI())
+}

firebaseai/ImagenScreen/ImagenFromTemplateViewModel.swift

@@ -0,0 +1,80 @@
+// Copyright 2025 Google LLC
+//
+// 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
+//
+//      http://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.
+
+#if canImport(FirebaseAILogic)
+  import FirebaseAILogic
+#else
+  import FirebaseAI
+#endif
+import Foundation
+import OSLog
+import SwiftUI
+
+@MainActor
+class ImagenFromTemplateViewModel: ObservableObject {
+  private var logger = Logger(subsystem: Bundle.main.bundleIdentifier!, category: "generative-ai")
+
+  @Published
+  var userInput: String = ""
+
+  @Published
+  var images = [UIImage]()
+
+  @Published
+  var errorMessage: String?
+
+  @Published
+  var inProgress = false
+
+  private let model: TemplateImagenModel
+
+  private var generateImagesTask: Task<Void, Never>?
+
+  init(firebaseService: FirebaseAI) {
+    model = firebaseService.templateImagenModel()
+  }
+
+  func generateImageFromTemplate(prompt: String) async {
+    stop()
+
+    generateImagesTask = Task {
+      inProgress = true
+      defer {
+        inProgress = false
+      }
+
+      do {
+        let response = try await model.generateImages(
+          templateID: "image-generation-basic",
+          inputs: [
+            "prompt": prompt,
+          ]
+        )
+
+        if !Task.isCancelled {
+          images = response.images.compactMap { UIImage(data: $0.data) }
+        }
+      } catch {
+        if !Task.isCancelled {
+          logger.error("Error generating images from template: \(error)")
+        }
+      }
+    }
+  }
+
+  func stop() {
+    generateImagesTask?.cancel()
+    generateImagesTask = nil
+  }
+}