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.
47 lines
1.3 KiB
47 lines
1.3 KiB
//
|
|
// PreviewView.swift
|
|
// Diffusion
|
|
//
|
|
// Created by Fahim Farook on 15/12/2022.
|
|
//
|
|
|
|
import SwiftUI
|
|
|
|
struct PreviewView: View {
|
|
var image: Binding<CGImage?>
|
|
var state: Binding<GenerationState>
|
|
|
|
var body: some View {
|
|
switch state.wrappedValue {
|
|
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 .idle(let lastPrompt):
|
|
guard let theImage = image.wrappedValue else {
|
|
return AnyView(Image(systemName: "exclamationmark.triangle").resizable())
|
|
}
|
|
|
|
let imageView = Image(theImage, scale: 1, label: Text("generated"))
|
|
return AnyView(
|
|
VStack {
|
|
imageView.resizable().clipShape(RoundedRectangle(cornerRadius: 20))
|
|
ShareLink(item: imageView, preview: SharePreview(lastPrompt, image: imageView))
|
|
})
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
struct PreviewView_Previews: PreviewProvider {
|
|
static var previews: some View {
|
|
PreviewView(image: .constant(nil), state: .constant(.startup))
|
|
}
|
|
}
|