You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
38 lines
1.3 KiB
38 lines
1.3 KiB
//
|
|
// GeneratedImageView.swift
|
|
// Diffusion
|
|
//
|
|
// Created by Pedro Cuenca on 18/1/23.
|
|
// See LICENSE at https://github.com/huggingface/swift-coreml-diffusers/LICENSE
|
|
//
|
|
|
|
import SwiftUI
|
|
|
|
struct GeneratedImageView: View {
|
|
@EnvironmentObject var generation: GenerationContext
|
|
|
|
var body: some View {
|
|
switch generation.state {
|
|
case .startup: return AnyView(Image("placeholder").resizable())
|
|
case .running(let progress):
|
|
guard let progress = progress, progress.stepCount > 0 else {
|
|
// The first time it takes a little bit before generation starts
|
|
return AnyView(ProgressView())
|
|
}
|
|
let step = Int(progress.step) + 1
|
|
let fraction = Double(step) / Double(progress.stepCount)
|
|
let label = "Step \(step) of \(progress.stepCount)"
|
|
return AnyView(ProgressView(label, value: fraction, total: 1).padding())
|
|
case .complete(_, let image, _):
|
|
guard let theImage = image else {
|
|
return AnyView(Image(systemName: "exclamationmark.triangle").resizable())
|
|
}
|
|
|
|
return AnyView(Image(theImage, scale: 1, label: Text("generated"))
|
|
.resizable()
|
|
.clipShape(RoundedRectangle(cornerRadius: 20))
|
|
)
|
|
}
|
|
}
|
|
}
|