174 lines
5.6 KiB
Swift
174 lines
5.6 KiB
Swift
import SwiftUI
|
|
import Combine
|
|
import AppKit
|
|
|
|
struct ParakeetModelCardRowView: View {
|
|
let model: ParakeetModel
|
|
@ObservedObject var whisperState: WhisperState
|
|
|
|
var isCurrent: Bool {
|
|
whisperState.currentTranscriptionModel?.name == model.name
|
|
}
|
|
|
|
var isDownloaded: Bool {
|
|
whisperState.isParakeetModelDownloaded
|
|
}
|
|
|
|
var isDownloading: Bool {
|
|
whisperState.isDownloadingParakeet
|
|
}
|
|
|
|
var body: some View {
|
|
HStack(alignment: .top, spacing: 16) {
|
|
VStack(alignment: .leading, spacing: 6) {
|
|
headerSection
|
|
metadataSection
|
|
descriptionSection
|
|
progressSection
|
|
}
|
|
.frame(maxWidth: .infinity, alignment: .leading)
|
|
|
|
actionSection
|
|
}
|
|
.padding(16)
|
|
.background(CardBackground(isSelected: isCurrent, useAccentGradientWhenSelected: isCurrent))
|
|
}
|
|
|
|
private var headerSection: some View {
|
|
HStack(alignment: .firstTextBaseline) {
|
|
Text(model.displayName)
|
|
.font(.system(size: 13, weight: .semibold))
|
|
.foregroundColor(Color(.labelColor))
|
|
|
|
Text("Experimental")
|
|
.font(.system(size: 11, weight: .medium))
|
|
.padding(.horizontal, 6)
|
|
.padding(.vertical, 2)
|
|
.background(Capsule().fill(Color.orange.opacity(0.8)))
|
|
.foregroundColor(.white)
|
|
|
|
statusBadge
|
|
Spacer()
|
|
}
|
|
}
|
|
|
|
private var statusBadge: some View {
|
|
Group {
|
|
if isCurrent {
|
|
Text("Default")
|
|
.font(.system(size: 11, weight: .medium))
|
|
.padding(.horizontal, 6)
|
|
.padding(.vertical, 2)
|
|
.background(Capsule().fill(Color.accentColor))
|
|
.foregroundColor(.white)
|
|
} else if isDownloaded {
|
|
Text("Downloaded")
|
|
.font(.system(size: 11, weight: .medium))
|
|
.padding(.horizontal, 6)
|
|
.padding(.vertical, 2)
|
|
.background(Capsule().fill(Color(.quaternaryLabelColor)))
|
|
.foregroundColor(Color(.labelColor))
|
|
}
|
|
}
|
|
}
|
|
|
|
private var metadataSection: some View {
|
|
HStack(spacing: 12) {
|
|
Label(model.language, systemImage: "globe")
|
|
Label(model.size, systemImage: "internaldrive")
|
|
HStack(spacing: 3) {
|
|
Text("Speed")
|
|
progressDotsWithNumber(value: model.speed * 10)
|
|
}
|
|
HStack(spacing: 3) {
|
|
Text("Accuracy")
|
|
progressDotsWithNumber(value: model.accuracy * 10)
|
|
}
|
|
}
|
|
.font(.system(size: 11))
|
|
.foregroundColor(Color(.secondaryLabelColor))
|
|
.lineLimit(1)
|
|
}
|
|
|
|
private var descriptionSection: some View {
|
|
Text(model.description)
|
|
.font(.system(size: 11))
|
|
.foregroundColor(Color(.secondaryLabelColor))
|
|
.lineLimit(2)
|
|
.fixedSize(horizontal: false, vertical: true)
|
|
.padding(.top, 4)
|
|
}
|
|
|
|
private var progressSection: some View {
|
|
Group {
|
|
if isDownloading {
|
|
ProgressView() // Indeterminate for now
|
|
.progressViewStyle(LinearProgressViewStyle())
|
|
.frame(maxWidth: 200)
|
|
.padding(.top, 8)
|
|
}
|
|
}
|
|
}
|
|
|
|
private var actionSection: some View {
|
|
HStack(spacing: 8) {
|
|
if isCurrent {
|
|
Text("Default Model")
|
|
.font(.system(size: 12))
|
|
.foregroundColor(Color(.secondaryLabelColor))
|
|
} else if isDownloaded {
|
|
Button(action: {
|
|
Task {
|
|
await whisperState.setDefaultTranscriptionModel(model)
|
|
}
|
|
}) {
|
|
Text("Set as Default")
|
|
.font(.system(size: 12))
|
|
}
|
|
.buttonStyle(.bordered)
|
|
.controlSize(.small)
|
|
} else {
|
|
Button(action: {
|
|
Task {
|
|
await whisperState.downloadParakeetModel()
|
|
}
|
|
}) {
|
|
HStack(spacing: 4) {
|
|
Text(isDownloading ? "Downloading..." : "Download")
|
|
Image(systemName: "arrow.down.circle")
|
|
}
|
|
.font(.system(size: 12, weight: .medium))
|
|
.foregroundColor(.white)
|
|
.padding(.horizontal, 12)
|
|
.padding(.vertical, 6)
|
|
.background(Capsule().fill(Color.accentColor))
|
|
}
|
|
.buttonStyle(.plain)
|
|
.disabled(isDownloading)
|
|
}
|
|
|
|
if isDownloaded {
|
|
Menu {
|
|
Button(action: {
|
|
whisperState.deleteParakeetModel()
|
|
}) {
|
|
Label("Delete Model", systemImage: "trash")
|
|
}
|
|
|
|
Button {
|
|
whisperState.showParakeetModelInFinder()
|
|
} label: {
|
|
Label("Show in Finder", systemImage: "folder")
|
|
}
|
|
} label: {
|
|
Image(systemName: "ellipsis.circle")
|
|
.font(.system(size: 14))
|
|
}
|
|
.menuStyle(.borderlessButton)
|
|
.menuIndicator(.hidden)
|
|
.frame(width: 20, height: 20)
|
|
}
|
|
}
|
|
}
|
|
}
|