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.
107 lines
3.9 KiB
107 lines
3.9 KiB
//
|
|
// PromptView.swift
|
|
// Diffusion-macOS
|
|
//
|
|
// Created by Cyril Zakka on 1/12/23.
|
|
//
|
|
|
|
import SwiftUI
|
|
import CompactSlider
|
|
|
|
|
|
struct PromptView: View {
|
|
|
|
var models = ["hf-default-model"]
|
|
|
|
@State private var model = "hf-default-model"
|
|
@State private var positivePrompt = ""
|
|
@State private var negativePrompt = ""
|
|
@State private var steps = 50.0
|
|
@State private var numImages = 1.0
|
|
@State private var seed = 386.0
|
|
|
|
var body: some View {
|
|
VStack(alignment: .leading) {
|
|
|
|
Label("Adjustments", systemImage: "gearshape.2")
|
|
.font(.headline)
|
|
.fontWeight(.bold)
|
|
Divider()
|
|
|
|
ScrollView {
|
|
Group {
|
|
DisclosureGroup {
|
|
Picker("", selection: $model) {
|
|
ForEach(models, id: \.self) {
|
|
Text($0)
|
|
}
|
|
}
|
|
} label: {
|
|
Label("Model", systemImage: "cpu").foregroundColor(.secondary)
|
|
}
|
|
|
|
Divider()
|
|
|
|
DisclosureGroup {
|
|
Group {
|
|
TextField("Positive prompt", text: $positivePrompt,
|
|
axis: .vertical).lineLimit(5)
|
|
.textFieldStyle(.squareBorder)
|
|
.listRowInsets(EdgeInsets(top: 0, leading: -20, bottom: 0, trailing: 20))
|
|
TextField("Negative prompt", text: $negativePrompt,
|
|
axis: .vertical).lineLimit(5)
|
|
.textFieldStyle(.squareBorder)
|
|
}.padding(.leading, 10)
|
|
} label: {
|
|
Label("Prompts", systemImage: "text.quote").foregroundColor(.secondary)
|
|
}
|
|
|
|
Divider()
|
|
|
|
DisclosureGroup {
|
|
CompactSlider(value: $steps, in: 0...250, step: 5) {
|
|
Text("Steps")
|
|
Spacer()
|
|
Text("\(Int(steps))")
|
|
}.padding(.leading, 10)
|
|
} label: {
|
|
Label("Step count", systemImage: "square.3.layers.3d.down.left").foregroundColor(.secondary)
|
|
}
|
|
|
|
Divider()
|
|
DisclosureGroup() {
|
|
CompactSlider(value: $numImages, in: 0...10, step: 1) {
|
|
Text("Number of Images")
|
|
Spacer()
|
|
Text("\(Int(numImages))")
|
|
}.padding(.leading, 10)
|
|
} label: {
|
|
Label("Number of images", systemImage: "photo.stack").foregroundColor(.secondary)
|
|
}
|
|
Divider()
|
|
DisclosureGroup() {
|
|
CompactSlider(value: $seed, in: 0...1000, step: 1) {
|
|
Text("Random seed")
|
|
Spacer()
|
|
Text("\(Int(seed))")
|
|
}.padding(.leading, 10)
|
|
} label: {
|
|
Label("Random Seed", systemImage: "leaf").foregroundColor(.secondary)
|
|
}
|
|
}
|
|
|
|
}
|
|
Button {
|
|
// Generate image here
|
|
} label: {
|
|
Text("Generate")
|
|
.frame(maxWidth: .infinity)
|
|
.frame(height: 50)
|
|
}
|
|
.buttonStyle(.borderedProminent)
|
|
// StatusView()
|
|
}.padding()
|
|
}
|
|
}
|
|
|