Rename context -> generation

Diffusion-macOS/ContentView.swift

@@ -8,7 +8,7 @@
 import SwiftUI
 
 struct ContentView: View {
-    @StateObject var context = GenerationContext()
+    @StateObject var generation = GenerationContext()
 
     var body: some View {
         NavigationSplitView {
@@ -26,7 +26,7 @@ struct ContentView: View {
                 }
 
         }
-        .environmentObject(context)
+        .environmentObject(generation)
     }
 }
 

Diffusion-macOS/PromptView.swift

@@ -18,7 +18,7 @@ enum PipelineState {
 }
 
 struct PromptView: View {
-    @EnvironmentObject var context: GenerationContext
+    @EnvironmentObject var generation: GenerationContext
 
     static let models = ModelInfo.MODELS
     static let modelNames = models.map { $0.modelVersion }
@@ -57,7 +57,7 @@ struct PromptView: View {
                 }
             }
             do {
-                context.pipeline = try await loader.prepare()
+                generation.pipeline = try await loader.prepare()
                 pipelineState = .ready
             } catch {
                 print("Could not load model, error: \(error)")

Diffusion/Views/Loading.swift

@@ -12,7 +12,7 @@ import Combine
 let model = ModelInfo.v2Base
 
 struct LoadingView: View {
-    @StateObject var context = GenerationContext()
+    @StateObject var generation = GenerationContext()
 
     @State private var preparationPhase = "Downloading…"
     @State private var downloadProgress: Double = 0
@@ -37,7 +37,7 @@ struct LoadingView: View {
             }
         }
         .animation(.easeIn, value: currentView)
-        .environmentObject(context)
+        .environmentObject(generation)
         .onAppear {
             Task.init {
                 let loader = PipelineLoader(model: model)
@@ -59,7 +59,7 @@ struct LoadingView: View {
                     }
                 }
                 do {
-                    context.pipeline = try await loader.prepare()
+                    generation.pipeline = try await loader.prepare()
                     self.currentView = .textToImage
                 } catch {
                     self.currentView = .error("Could not load model, error: \(error)")

Diffusion/Views/TextToImage.swift

@@ -94,18 +94,18 @@ struct ImageWithPlaceholder: View {
 }
 
 struct TextToImage: View {
-    @EnvironmentObject var context: GenerationContext
+    @EnvironmentObject var generation: GenerationContext
 
     @State private var prompt = "Labrador in the style of Vermeer"
 
     func submit() {
-        if case .running = context.state { return }
+        if case .running = generation.state { return }
         Task {
-            context.state = .running(nil)
+            generation.state = .running(nil)
             let interval: TimeInterval?
             let image: CGImage?
-            (image, interval) = await context.generate(prompt: prompt, steps: steps, seed: seed) ?? (nil, nil)
+            (image, interval) = await generation.generate(prompt: prompt, steps: steps, seed: seed) ?? (nil, nil)
-            context.state = .complete(prompt, image, interval)
+            generation.state = .complete(prompt, image, interval)
         }
     }
     
@@ -123,7 +123,7 @@ struct TextToImage: View {
                 .padding()
                 .buttonStyle(.borderedProminent)
             }
-            ImageWithPlaceholder(state: $context.state)
+            ImageWithPlaceholder(state: $generation.state)
                 .scaledToFit()
             Spacer()
         }