iOS 18.4の新機能「ImageCreator」で始める動的な画像生成

import ImagePlayground
import SwiftUI

struct ContentView: View {
    @State private var images: [Image] = []
    @State private var prompt = "A cat wearing mittens."
    @State private var errorText: Text? = nil
    @State private var isProgress: Bool = false

    @Environment(\.supportsImagePlayground) private var supportsImagePlayground

    let columns = [GridItem(.adaptive(minimum: 100, maximum: 200))]

    var body: some View {
        if supportsImagePlayground {
            contentView
        } else {
            Text("Apple Intelligence未対応のデバイスです。画像生成はできません")
                .padding()
        }
    }

    var contentView: some View {
        VStack {
            // 生成した画像のリスト
            ScrollView {
                LazyVGrid(columns: columns, alignment: .leading) {
                    ForEach(images.indices, id: \.self) { index in
                        images[index]
                            .resizable()
                            .scaledToFit()
                    }
                }
            }

            Group {
                if isProgress {
                    // 画像生成中はプログレスビューを表示して、ボタンを連打できないようにする
                    ProgressView()
                        .padding()
                } else {
                    if let errorText {
                        errorText
                            .padding()
                    }

                    HStack(alignment: .center) {
                        TextField(text: $prompt) {
                            Text("Enter text")
                        }

                        Button(action: {
                            Task {
                                await createImage()
                            }
                        }) {
                            Text("Generate Images")
                        }
                    }
                }
            }
        }
        .padding()
    }

    @available(iOS 18.4, *)
    private func createImage() async {
        errorText = nil
        isProgress = true
        defer {
            isProgress = false
        }

        do {
            // 1. ImageCreatorインスタンスの生成
            let imageCreator = try await ImageCreator()

            // 2. スタイルの選択
            guard let style = imageCreator.availableStyles.first else {
                errorText = Text("利用可能なスタイルが見つかりません")
                return
            }

            // 3. プロンプトやスタイルを指定して画像生成する
            let generatedImages = imageCreator.images(
                for: [.text(prompt)],
                style: style,
                limit: 1
            )

            // 4. 生成された画像を受け取る
            for try await image in generatedImages {
                let cgImage = image.cgImage
                let image = Image(cgImage, scale: 1, label: Text("生成された画像です"))

                await MainActor.run {
                    images.append(image)
                }
            }
        } catch ImageCreator.Error.notSupported {
            errorText = Text("このデバイスでは画像生成はできません")
        } catch ImageCreator.Error.creationFailed {
            errorText = Text("画像生成に失敗しました")
        } catch {
            errorText = Text("エラー: \(error.localizedDescription)")
        }
    }
}