@@ -66,7 +66,7 @@ struct ImageWithPlaceholder: View {
66
66
let fraction = Double ( step) / Double( progress. stepCount)
67
67
let label = " Step \( step) of \( progress. stepCount) "
68
68
return AnyView ( ProgressView ( label, value: fraction, total: 1 ) . padding ( ) )
69
- case . complete( let lastPrompt, let image, let interval) :
69
+ case . complete( let lastPrompt, let image, _ , let interval) :
70
70
guard let theImage = image else {
71
71
return AnyView ( Image ( systemName: " exclamationmark.triangle " ) . resizable ( ) )
72
72
}
@@ -86,6 +86,8 @@ struct ImageWithPlaceholder: View {
86
86
)
87
87
} . frame ( maxHeight: 25 )
88
88
} )
89
+ case . failed( _) :
90
+ return AnyView ( Image ( systemName: " exclamationmark.triangle " ) . resizable ( ) )
89
91
}
90
92
}
91
93
}
@@ -97,10 +99,12 @@ struct TextToImage: View {
97
99
if case . running = generation. state { return }
98
100
Task {
99
101
generation. state = . running( nil )
100
- let interval : TimeInterval ?
101
- let image : CGImage ?
102
- let result = await generation. generate ( )
103
- generation. state = . complete( generation. positivePrompt, image, interval)
102
+ do {
103
+ let result = try await generation. generate ( )
104
+ generation. state = . complete( generation. positivePrompt, result. image, result. lastSeed, result. interval)
105
+ } catch {
106
+ generation. state = . failed( error)
107
+ }
104
108
}
105
109
}
106
110
0 commit comments