echodict/llama.cpp

1import SwiftUI2import UniformTypeIdentifiers3 4struct LoadCustomButton: View {5    @ObservedObject private var llamaState: LlamaState6    @State private var showFileImporter = false7 8    init(llamaState: LlamaState) {9        self.llamaState = llamaState10    }11 12    var body: some View {13        VStack {14            Button(action: {15                showFileImporter = true16            }) {17                Text("Load Custom Model")18            }19        }20        .fileImporter(21            isPresented: $showFileImporter,22            allowedContentTypes: [UTType(filenameExtension: "gguf", conformingTo: .data)!],23            allowsMultipleSelection: false24        ) { result in25            switch result {26            case .success(let files):27                files.forEach { file in28                    let gotAccess = file.startAccessingSecurityScopedResource()29                    if !gotAccess { return }30 31                    do {32                        try llamaState.loadModel(modelUrl: file.absoluteURL)33                    } catch let err {34                        print("Error: \(err.localizedDescription)")35                    }36 37                    file.stopAccessingSecurityScopedResource()38                }39            case .failure(let error):40                print(error)41            }42        }43    }44}45