123 lines
4.2 KiB
Swift
123 lines
4.2 KiB
Swift
import SwiftUI
|
|
|
|
struct DownloadButton: View {
|
|
@ObservedObject private var llamaState: LlamaState
|
|
private var modelName: String
|
|
private var modelUrl: String
|
|
private var filename: String
|
|
|
|
@State private var status: String
|
|
|
|
@State private var downloadTask: URLSessionDownloadTask?
|
|
@State private var progress = 0.0
|
|
@State private var observation: NSKeyValueObservation?
|
|
|
|
private static func getFileURL(filename: String) -> URL {
|
|
FileManager.default.urls(for: .documentDirectory, in: .userDomainMask)[0].appendingPathComponent(filename)
|
|
}
|
|
|
|
private func checkFileExistenceAndUpdateStatus() {
|
|
}
|
|
|
|
init(llamaState: LlamaState, modelName: String, modelUrl: String, filename: String) {
|
|
self.llamaState = llamaState
|
|
self.modelName = modelName
|
|
self.modelUrl = modelUrl
|
|
self.filename = filename
|
|
|
|
let fileURL = DownloadButton.getFileURL(filename: filename)
|
|
status = FileManager.default.fileExists(atPath: fileURL.path) ? "downloaded" : "download"
|
|
}
|
|
|
|
private func download() {
|
|
status = "downloading"
|
|
print("Downloading model \(modelName) from \(modelUrl)")
|
|
guard let url = URL(string: modelUrl) else { return }
|
|
let fileURL = DownloadButton.getFileURL(filename: filename)
|
|
|
|
downloadTask = URLSession.shared.downloadTask(with: url) { temporaryURL, response, error in
|
|
if let error = error {
|
|
print("Error: \(error.localizedDescription)")
|
|
return
|
|
}
|
|
|
|
guard let response = response as? HTTPURLResponse, (200...299).contains(response.statusCode) else {
|
|
print("Server error!")
|
|
return
|
|
}
|
|
|
|
do {
|
|
if let temporaryURL = temporaryURL {
|
|
try FileManager.default.copyItem(at: temporaryURL, to: fileURL)
|
|
print("Writing to \(filename) completed")
|
|
|
|
llamaState.cacheCleared = false
|
|
|
|
status = "downloaded"
|
|
}
|
|
} catch let err {
|
|
print("Error: \(err.localizedDescription)")
|
|
}
|
|
}
|
|
|
|
observation = downloadTask?.progress.observe(\.fractionCompleted) { progress, _ in
|
|
self.progress = progress.fractionCompleted
|
|
}
|
|
|
|
downloadTask?.resume()
|
|
}
|
|
|
|
var body: some View {
|
|
VStack {
|
|
if status == "download" {
|
|
Button(action: download) {
|
|
Text("Download " + modelName)
|
|
}
|
|
} else if status == "downloading" {
|
|
Button(action: {
|
|
downloadTask?.cancel()
|
|
status = "download"
|
|
}) {
|
|
Text("\(modelName) (Downloading \(Int(progress * 100))%)")
|
|
}
|
|
} else if status == "downloaded" {
|
|
Button(action: {
|
|
let fileURL = DownloadButton.getFileURL(filename: filename)
|
|
if !FileManager.default.fileExists(atPath: fileURL.path) {
|
|
download()
|
|
return
|
|
}
|
|
do {
|
|
try llamaState.loadModel(modelUrl: fileURL)
|
|
} catch let err {
|
|
print("Error: \(err.localizedDescription)")
|
|
}
|
|
}) {
|
|
Text("\(modelName) (Downloaded)")
|
|
}
|
|
} else {
|
|
Text("Unknown status")
|
|
}
|
|
}
|
|
.onDisappear() {
|
|
downloadTask?.cancel()
|
|
}
|
|
.onChange(of: llamaState.cacheCleared) { newValue in
|
|
if newValue {
|
|
downloadTask?.cancel()
|
|
let fileURL = DownloadButton.getFileURL(filename: filename)
|
|
status = FileManager.default.fileExists(atPath: fileURL.path) ? "downloaded" : "download"
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// #Preview {
|
|
// DownloadButton(
|
|
// llamaState: LlamaState(),
|
|
// modelName: "TheBloke / TinyLlama-1.1B-1T-OpenOrca-GGUF (Q4_0)",
|
|
// modelUrl: "https://huggingface.co/TheBloke/TinyLlama-1.1B-1T-OpenOrca-GGUF/resolve/main/tinyllama-1.1b-1t-openorca.Q4_0.gguf?download=true",
|
|
// filename: "tinyllama-1.1b-1t-openorca.Q4_0.gguf"
|
|
// )
|
|
// }
|