StabilityUltra.swift

StabilityUltra.swift

import Foundation
import SwiftUI
import AIProxy

final actor ImageLoader {
    static func create(fromPrompt prompt: String) async -> UIImage? {
        let service = AIProxy.stabilityAIService(
            partialKey: "your-partial-key",
            serviceURL: "your-service-url")
        do {
            let body = StabilityAIUltraRequestBody(prompt: prompt)
            let response = try await service.ultraRequest(body: body)
            return UIImage(data: response.imageData)
        }  catch AIProxyError.unsuccessfulRequest(let statusCode, let responseBody) {
            print("Received non-200 status code: \(statusCode) with response body: \(responseBody)")
        } catch {
            print(error.localizedDescription)
        }
        return nil
    }
}

@Observable
final class ViewModel {
    var myImage: UIImage?
    init() {
        Task {
            await self.perform()
        }
    }

    func perform() async {
        await self.myImage = ImageLoader.create(fromPrompt: "Lighthouse on a cliff overlooking the ocean")
    }
}

struct ContentView: View {
    @State var viewModel = ViewModel()
    var body: some View {
        VStack {
            if let image = viewModel.myImage {
                Image(uiImage: image).imageScale(.large)
            }
        }
        .padding()
    }
}