Add Create Image example to Demo app

Author: SunburstEnzo

Demo/App/APIProvidedView.swift

@@ -12,6 +12,7 @@ import SwiftUI
 struct APIProvidedView: View {
     @Binding var apiKey: String
     @StateObject var chatStore: ChatStore
+    @StateObject var imageStore: ImageStore
     @StateObject var miscStore: MiscStore
     @State var isShowingAPIConfigModal: Bool = true
 
@@ -29,6 +30,11 @@ struct APIProvidedView: View {
                 idProvider: idProvider
             )
         )
+        self._imageStore = StateObject(
+            wrappedValue: ImageStore(
+                openAIClient: OpenAI(apiToken: apiKey.wrappedValue)
+            )
+        )
         self._miscStore = StateObject(
             wrappedValue: MiscStore(
                 openAIClient: OpenAI(apiToken: apiKey.wrappedValue)
@@ -39,6 +45,7 @@ struct APIProvidedView: View {
     var body: some View {
         ContentView(
             chatStore: chatStore,
+            imageStore: imageStore,
             miscStore: miscStore
         )
         .onChange(of: apiKey) { newApiKey in

Demo/App/ContentView.swift

@@ -11,6 +11,7 @@ import SwiftUI
 
 struct ContentView: View {
     @ObservedObject var chatStore: ChatStore
+    @ObservedObject var imageStore: ImageStore
     @ObservedObject var miscStore: MiscStore
     @State private var selectedTab = 0
     @Environment(\.idProviderValue) var idProvider
@@ -33,6 +34,7 @@ struct ContentView: View {
             .tag(1)
 
             ImageView(
+                store: imageStore
             )
             .tabItem {
                 Label("Image", systemImage: "photo")
@@ -63,10 +65,3 @@ struct TranscribeView: View {
             .font(.largeTitle)
     }
 }
-
-struct ImageView: View {
-    var body: some View {
-        Text("Image: TBD")
-            .font(.largeTitle)
-    }
-}

Demo/DemoChat/Sources/ImageStore.swift

@@ -0,0 +1,33 @@
+//
+//  ImageStore.swift
+//  DemoChat
+//
+//  Created by Aled Samuel on 24/04/2023.
+//
+
+import Foundation
+import OpenAI
+
+public final class ImageStore: ObservableObject {
+    public var openAIClient: OpenAIProtocol
+    
+    @Published var images: [ImagesResult.URLResult] = []
+    
+    public init(
+        openAIClient: OpenAIProtocol
+    ) {
+        self.openAIClient = openAIClient
+    }
+    
+    @MainActor
+    func images(query: ImagesQuery) async {
+        images.removeAll()
+        do {
+            let response = try await openAIClient.images(query: query)
+            images = response.data
+        } catch {
+            // TODO: Better error handling
+            print(error.localizedDescription)
+        }
+    }
+}

Demo/DemoChat/Sources/UI/Images/ImageCreationView.swift

@@ -0,0 +1,93 @@
+//
+//  ImageCreationView.swift
+//  DemoChat
+//
+//  Created by Aled Samuel on 24/04/2023.
+//
+
+import SwiftUI
+import OpenAI
+import SafariServices
+
+public struct ImageCreationView: View {
+    @ObservedObject var store: ImageStore
+    
+    @State private var prompt: String = ""
+    @State private var n: Int = 1
+    @State private var size: String
+    @State private var showSafari = false
+    
+    private var sizes = ["256x256", "512x512", "1024x1024"]
+    
+    public init(store: ImageStore) {
+        self.store = store
+        size = sizes[0]
+    }
+    
+    public var body: some View {
+        List {
+            Section {
+                HStack {
+                    Text("Prompt")
+                    Spacer()
+                    TextEditor(text: $prompt)
+                        .multilineTextAlignment(.trailing)
+                }
+                HStack {
+                    Stepper("Amount: \(n)", value: $n, in: 1...10)
+                }
+                HStack {
+                    Picker("Size", selection: $size) {
+                        ForEach(sizes, id: \.self) {
+                            Text($0)
+                        }
+                    }
+                }
+            }
+            Section {
+                HStack {
+                    Button("Create Image" + (n == 1 ? "" : "s")) {
+                        Task {
+                            let query = ImagesQuery(prompt: prompt, n: n, size: size)
+                            await store.images(query: query)
+                        }
+                    }
+                    .foregroundColor(.accentColor)
+                    Spacer()
+                }
+            }
+            if !$store.images.isEmpty {
+                Section("Images") {
+                    ForEach($store.images, id: \.self) { image in
+                        let urlString = image.wrappedValue.url
+                        if let imageURL = URL(string: urlString), UIApplication.shared.canOpenURL(imageURL) {
+                            Button {
+                                showSafari.toggle()
+                            } label: {
+                                Text(urlString)
+                                    .foregroundStyle(.foreground)
+                            }.fullScreenCover(isPresented: $showSafari, content: {
+                                SafariView(url: imageURL)
+                            })
+                        } else {
+                            Text(urlString)
+                                .foregroundStyle(.secondary)
+                        }
+                    }
+                }
+            }
+        }
+        .listStyle(.insetGrouped)
+        .navigationTitle("Create Image")
+    }
+}
+
+private struct SafariView: UIViewControllerRepresentable {
+    var url: URL
+    
+    func makeUIViewController(context: UIViewControllerRepresentableContext<SafariView>) -> SFSafariViewController {
+        SFSafariViewController(url: url)
+    }
+    
+    func updateUIViewController(_ safariViewController: SFSafariViewController, context: UIViewControllerRepresentableContext<SafariView>) { }
+}

Demo/DemoChat/Sources/UI/Images/ImageView.swift

@@ -0,0 +1,63 @@
+//
+//  ImageView.swift
+//  DemoChat
+//
+//  Created by Aled Samuel on 24/04/2023.
+//
+
+import SwiftUI
+
+public struct ImageView: View {
+    @ObservedObject var store: ImageStore
+    
+    public init(store: ImageStore) {
+        self.store = store
+    }
+    
+    public var body: some View {
+        NavigationStack {
+            List {
+                NavigationLink("Create Image", destination: ImageCreationView(store: store))
+                NavigationLink("Create Image Edit", destination: ImageEditView(store: store))
+                    .disabled(true)
+                NavigationLink("Create Image Variation", destination: ImageVariationView(store: store))
+                    .disabled(true)
+                
+            }
+            .listStyle(.insetGrouped)
+            .navigationTitle("Image")
+        }
+    }
+}
+
+public struct ImageEditView: View {
+    @ObservedObject var store: ImageStore
+    
+    public init(store: ImageStore) {
+        self.store = store
+    }
+    
+    public var body: some View {
+        List {
+            
+        }
+        .listStyle(.insetGrouped)
+        .navigationTitle("Create Image Edit")
+    }
+}
+
+public struct ImageVariationView: View {
+    @ObservedObject var store: ImageStore
+    
+    public init(store: ImageStore) {
+        self.store = store
+    }
+    
+    public var body: some View {
+        List {
+            
+        }
+        .listStyle(.insetGrouped)
+        .navigationTitle("Create Image Variation")
+    }
+}

Sources/OpenAI/Public/Models/ImagesResult.swift

@@ -15,3 +15,5 @@ public struct ImagesResult: Codable, Equatable {
     public let created: TimeInterval
     public let data: [URLResult]
 }
+
+extension ImagesResult.URLResult: Hashable { }