Build UI for template generate content

Author: paulb777

firebaseai/FirebaseAIExample.xcodeproj/project.pbxproj

@@ -60,6 +60,10 @@
 		AEE793DF2E256D3900708F02 /* GoogleSearchSuggestionView.swift in Sources */ = {isa = PBXBuildFile; fileRef = AEE793DC2E256D3900708F02 /* GoogleSearchSuggestionView.swift */; };
 		AEE793E02E256D3900708F02 /* GroundedResponseView.swift in Sources */ = {isa = PBXBuildFile; fileRef = AEE793DD2E256D3900708F02 /* GroundedResponseView.swift */; };
 		DE26D95F2DBB3E9F007E6668 /* FirebaseAI in Frameworks */ = {isa = PBXBuildFile; productRef = DE26D95E2DBB3E9F007E6668 /* FirebaseAI */; };
+		DE907A812EAAE53E00AE56CE /* GenerateContentFromTemplateScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = DE907A802EAAE53E00AE56CE /* GenerateContentFromTemplateScreen.swift */; };
+		DE907A822EAAE53E00AE56CE /* GenerateContentFromTemplateScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = DE907A802EAAE53E00AE56CE /* GenerateContentFromTemplateScreen.swift */; };
+		DE907A842EAAE55600AE56CE /* GenerateContentFromTemplateViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = DE907A832EAAE55600AE56CE /* GenerateContentFromTemplateViewModel.swift */; };
+		DE907A852EAAE55600AE56CE /* GenerateContentFromTemplateViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = DE907A832EAAE55600AE56CE /* GenerateContentFromTemplateViewModel.swift */; };
 		DEFECAA92D7B4CCD00EF9621 /* ImagenViewModel.swift in Sources */ = {isa = PBXBuildFile; fileRef = DEFECAA72D7B4CCD00EF9621 /* ImagenViewModel.swift */; };
 		DEFECAAA2D7B4CCD00EF9621 /* ImagenScreen.swift in Sources */ = {isa = PBXBuildFile; fileRef = DEFECAA62D7B4CCD00EF9621 /* ImagenScreen.swift */; };
 /* End PBXBuildFile section */
@@ -94,6 +98,8 @@
 		88E10F5C2B11135000C08E95 /* BouncingDots.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = BouncingDots.swift; sourceTree = "<group>"; };
 		AEE793DC2E256D3900708F02 /* GoogleSearchSuggestionView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = GoogleSearchSuggestionView.swift; sourceTree = "<group>"; };
 		AEE793DD2E256D3900708F02 /* GroundedResponseView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = GroundedResponseView.swift; sourceTree = "<group>"; };
+		DE907A802EAAE53E00AE56CE /* GenerateContentFromTemplateScreen.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = GenerateContentFromTemplateScreen.swift; sourceTree = "<group>"; };
+		DE907A832EAAE55600AE56CE /* GenerateContentFromTemplateViewModel.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = GenerateContentFromTemplateViewModel.swift; sourceTree = "<group>"; };
 		DEFECAA62D7B4CCD00EF9621 /* ImagenScreen.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ImagenScreen.swift; sourceTree = "<group>"; };
 		DEFECAA72D7B4CCD00EF9621 /* ImagenViewModel.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ImagenViewModel.swift; sourceTree = "<group>"; };
 /* End PBXFileReference section */
@@ -171,6 +177,7 @@
 		88209C1A2B0FBDC300F64795 /* Screens */ = {
 			isa = PBXGroup;
 			children = (
+				DE907A802EAAE53E00AE56CE /* GenerateContentFromTemplateScreen.swift */,
 				88209C1B2B0FBDC300F64795 /* GenerateContentScreen.swift */,
 			);
 			path = Screens;
@@ -179,6 +186,7 @@
 		88209C1C2B0FBDC300F64795 /* ViewModels */ = {
 			isa = PBXGroup;
 			children = (
+				DE907A832EAAE55600AE56CE /* GenerateContentFromTemplateViewModel.swift */,
 				88209C1D2B0FBDC300F64795 /* GenerateContentViewModel.swift */,
 			);
 			path = ViewModels;
@@ -468,6 +476,8 @@
 			isa = PBXSourcesBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
+				DE907A852EAAE55600AE56CE /* GenerateContentFromTemplateViewModel.swift in Sources */,
+				DE907A812EAAE53E00AE56CE /* GenerateContentFromTemplateScreen.swift in Sources */,
 				86BB55EA2E8B2D6D0054B8B5 /* FunctionCallingScreen.swift in Sources */,
 				86BB55EB2E8B2D6D0054B8B5 /* BouncingDots.swift in Sources */,
 				86BB55EC2E8B2D6D0054B8B5 /* FunctionCallingViewModel.swift in Sources */,
@@ -494,6 +504,8 @@
 			isa = PBXSourcesBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
+				DE907A842EAAE55600AE56CE /* GenerateContentFromTemplateViewModel.swift in Sources */,
+				DE907A822EAAE53E00AE56CE /* GenerateContentFromTemplateScreen.swift in Sources */,
 				86C1F4832BC726150026816F /* FunctionCallingScreen.swift in Sources */,
 				886F95DF2B17D5010036F07A /* BouncingDots.swift in Sources */,
 				86C1F4842BC726150026816F /* FunctionCallingViewModel.swift in Sources */,
@@ -827,8 +839,8 @@
 			isa = XCRemoteSwiftPackageReference;
 			repositoryURL = "https://github.com/firebase/firebase-ios-sdk.git";
 			requirement = {
-				kind = upToNextMajorVersion;
-				minimumVersion = 12.0.0;
+				branch = "pb-spt";
+				kind = branch;
 			};
 		};
 /* End XCRemoteSwiftPackageReference section */

firebaseai/FirebaseAIExample/ContentView.swift

@@ -55,6 +55,11 @@ struct ContentView: View {
           } label: {
             Label("Generate Content", systemImage: "doc.text")
           }
+          NavigationLink {
+            GenerateContentFromTemplateScreen(firebaseService: firebaseService)
+          } label: {
+            Label("Generate Content from Template", systemImage: "doc.text.fill")
+          }
           NavigationLink {
             PhotoReasoningScreen(firebaseService: firebaseService)
           } label: {

firebaseai/GenerativeAITextExample/Screens/GenerateContentFromTemplateScreen.swift

@@ -0,0 +1,89 @@
+// Copyright 2023 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 MarkdownUI
+import SwiftUI
+#if canImport(FirebaseAILogic)
+  import FirebaseAILogic
+#else
+  import FirebaseAI
+#endif
+import GenerativeAIUIComponents
+
+struct GenerateContentFromTemplateScreen: View {
+  let firebaseService: FirebaseAI
+  @StateObject var viewModel: GenerateContentFromTemplateViewModel
+  @State var userInput = ""
+
+  init(firebaseService: FirebaseAI) {
+    self.firebaseService = firebaseService
+    _viewModel =
+      StateObject(wrappedValue: GenerateContentFromTemplateViewModel(firebaseService: firebaseService))
+  }
+
+  enum FocusedField: Hashable {
+    case message
+  }
+
+  @FocusState
+  var focusedField: FocusedField?
+
+  var body: some View {
+    VStack {
+      VStack(alignment: .leading) {
+        Text("Enter your name, then tap on _Go_ to run generateContent from template on it.")
+          .padding(.horizontal, 6)
+        InputField("Enter your name", text: $userInput) {
+          Text("Go")
+        }
+        .focused($focusedField, equals: .message)
+        .onSubmit { onGenerateContentTapped() }
+      }
+      .padding(.horizontal, 16)
+
+      List {
+        HStack(alignment: .top) {
+          if viewModel.inProgress {
+            ProgressView()
+          } else {
+            Image(systemName: "cloud.circle.fill")
+              .font(.title2)
+          }
+
+          Markdown("\(viewModel.outputText)")
+        }
+        .listRowSeparator(.hidden)
+      }
+      .listStyle(.plain)
+    }
+    .onTapGesture {
+      focusedField = nil
+    }
+    .navigationTitle("Template Text example")
+  }
+
+  private func onGenerateContentTapped() {
+    focusedField = nil
+
+    Task {
+      await viewModel.generateContentFromTemplate(name: userInput)
+    }
+  }
+}
+
+#Preview {
+  NavigationStack {
+    GenerateContentFromTemplateScreen(firebaseService: FirebaseAI.firebaseAI())
+  }
+}

firebaseai/GenerativeAITextExample/ViewModels/GenerateContentFromTemplateViewModel.swift

@@ -0,0 +1,71 @@
+// Copyright 2023 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
+
+@MainActor
+class GenerateContentFromTemplateViewModel: ObservableObject {
+  private var logger = Logger(subsystem: Bundle.main.bundleIdentifier!, category: "generative-ai")
+
+  @Published
+  var outputText = ""
+
+  @Published
+  var errorMessage: String?
+
+  @Published
+  var inProgress = false
+
+  private var model: TemplateGenerativeModel?
+
+  init(firebaseService: FirebaseAI) {
+    model = firebaseService.templateGenerativeModel()
+   // model = firebaseService.generativeModel(modelName: "gemini-2.0-flash-001")
+  }
+
+  func generateContentFromTemplate(name: String) async {
+    defer {
+      inProgress = false
+    }
+    guard let model else {
+      return
+    }
+
+    do {
+      inProgress = true
+      errorMessage = nil
+      outputText = ""
+
+      let response = try await model.generateContent(
+        templateID: "new-greeting",
+        inputs: [
+          "name": name,
+          "language": "English",
+        ]
+      )
+      if let text = response.text {
+        outputText = text
+      }
+    } catch {
+      logger.error("\(error.localizedDescription)")
+      errorMessage = error.localizedDescription
+    }
+  }
+}