[FirebaseAI ] Integrate conversationkit

firebaseai/ChatExample/Models/ChatMessage.swift

@@ -15,30 +15,30 @@
 import FirebaseAI
 import Foundation
 
-enum Participant {
+public enum Participant {
   case system
   case user
 }
 
-struct ChatMessage: Identifiable, Equatable {
-  let id = UUID().uuidString
-  var message: String
-  var groundingMetadata: GroundingMetadata?
-  let participant: Participant
-  var pending = false
+public struct ChatMessage: Identifiable, Equatable {
+  public let id = UUID().uuidString
+  public var message: String
+  public let participant: Participant
+  public var groundingMetadata: GroundingMetadata?
+  public var pending = false
 
-  static func pending(participant: Participant) -> ChatMessage {
+  public static func pending(participant: Participant) -> ChatMessage {
     Self(message: "", participant: participant, pending: true)
   }
 
   // TODO(andrewheard): Add Equatable conformance to GroundingMetadata and remove this
-  static func == (lhs: ChatMessage, rhs: ChatMessage) -> Bool {
+  public static func == (lhs: ChatMessage, rhs: ChatMessage) -> Bool {
     lhs.id == rhs.id && lhs.message == rhs.message && lhs.participant == rhs.participant && lhs
       .pending == rhs.pending
   }
 }
 
-extension ChatMessage {
+public extension ChatMessage {
   static var samples: [ChatMessage] = [
     .init(message: "Hello. What can I do for you today?", participant: .system),
     .init(message: "Show me a simple loop in Swift.", participant: .user),
@@ -71,7 +71,7 @@ extension ChatMessage {
   static var sample = samples[0]
 }
 
-extension ChatMessage {
+public extension ChatMessage {
   static func from(_ modelContent: ModelContent) -> ChatMessage? {
     // TODO: add non-text parts to message when multi-model support is added
     let text = modelContent.parts.compactMap { ($0 as? TextPart)?.text }.joined()

firebaseai/ChatExample/Screens/ChatScreen.swift

@@ -0,0 +1,77 @@
+// 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 FirebaseAI
+import SwiftUI
+
+struct ChatScreen: View {
+  let firebaseService: FirebaseAI
+  @StateObject var viewModel: ChatViewModel
+
+  init(firebaseService: FirebaseAI, sample: Sample? = nil) {
+    self.firebaseService = firebaseService
+    _viewModel =
+      StateObject(wrappedValue: ChatViewModel(firebaseService: firebaseService,
+                                              sample: sample))
+  }
+
+  var body: some View {
+    NavigationStack {
+      ConversationView(messages: $viewModel.messages,
+                       userPrompt: viewModel.initialPrompt) { message in
+        MessageView(message: message)
+      }
+      .disableAttachments()
+      .errorState(viewModel.error)
+      .onSendMessage { prompt in
+        Task {
+          await viewModel.sendMessage(prompt, streaming: true)
+        }
+      }
+      .toolbar {
+        ToolbarItem(placement: .primaryAction) {
+          Button(action: newChat) {
+            Image(systemName: "square.and.pencil")
+          }
+        }
+      }
+      .navigationTitle(viewModel.title)
+      .navigationBarTitleDisplayMode(.inline)
+    }
+  }
+
+  private func newChat() {
+    viewModel.startNewChat()
+  }
+}
+
+struct ChatScreen_Previews: PreviewProvider {
+  struct ContainerView: View {
+    @StateObject var viewModel = ChatViewModel(firebaseService: FirebaseAI
+      .firebaseAI(), sample: nil) // Example service init
+
+    var body: some View {
+      ChatScreen(firebaseService: FirebaseAI.firebaseAI())
+        .onAppear {
+          viewModel.messages = ChatMessage.samples
+        }
+    }
+  }
+
+  static var previews: some View {
+    NavigationStack {
+      ChatScreen(firebaseService: FirebaseAI.firebaseAI())
+    }
+  }
+}

firebaseai/ChatExample/ViewModels/ConversationViewModel.swift → firebaseai/ChatExample/ViewModels/ChatViewModel.swift

@@ -15,10 +15,9 @@
 import FirebaseAI
 import Foundation
 import UIKit
-import GenerativeAIUIComponents
 
 @MainActor
-class ConversationViewModel: ObservableObject {
+class ChatViewModel: ObservableObject {
   /// This array holds both the user's and the system's chat messages
   @Published var messages = [ChatMessage]()
 

firebaseai/ChatExample/Views/MessageView.swift

@@ -83,24 +83,41 @@ struct ResponseTextView: View {
 struct MessageView: View {
   var message: ChatMessage
 
+  private var participantLabel: String {
+    message.participant == .user ? "User" : "Model"
+  }
+
   var body: some View {
-    HStack {
-      if message.participant == .user {
-        Spacer()
-      }
-      MessageContentView(message: message)
-        .padding(10)
-        .background(message.participant == .system
-          ? Color(UIColor.systemFill)
-          : Color(UIColor.systemBlue))
-        .roundedCorner(10,
-                       corners: [
-                         .topLeft,
-                         .topRight,
-                         message.participant == .system ? .bottomRight : .bottomLeft,
-                       ])
-      if message.participant == .system {
-        Spacer()
+    VStack(alignment: message.participant == .user ? .trailing : .leading, spacing: 4) {
+      // Sender label
+      Text(participantLabel)
+        .font(.caption2)
+        .fontWeight(.medium)
+        .foregroundColor(.secondary)
+        .textCase(.uppercase)
+        .padding(.horizontal, 8)
+        .padding(.vertical, 2)
+        .frame(maxWidth: .infinity, alignment: message.participant == .user ? .trailing : .leading)
+
+      // Message content
+      HStack {
+        if message.participant == .user {
+          Spacer()
+        }
+        MessageContentView(message: message)
+          .padding(10)
+          .background(message.participant == .system
+            ? Color(UIColor.systemFill)
+            : Color(UIColor.systemBlue))
+          .roundedCorner(10,
+                         corners: [
+                           .topLeft,
+                           .topRight,
+                           message.participant == .system ? .bottomRight : .bottomLeft,
+                         ])
+        if message.participant == .system {
+          Spacer()
+        }
       }
     }
     .listRowSeparator(.hidden)