jake/vOOice
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Improve model loading during recording

Browse Source
This commit is contained in:
Beingpax 2025-03-22 14:42:27 +05:45
parent 6a376dd5ad
commit ca264b3300
1 changed files with 30 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

60
VoiceInk/Whisper/WhisperState.swift
View File

@ -132,30 +132,50 @@ class WhisperState: NSObject, ObservableObject, AVAudioRecorderDelegate {
if granted {
Task {
do {
// Start window configuration in parallel with recording setup
async let windowConfigTask = ActiveWindowService.shared.applyConfigurationForCurrentApp()
let file = try FileManager.default.url(for: .documentDirectory,
in: .userDomainMask,
appropriateFor: nil,
create: true)
.appending(path: "output.wav")
try await self.recorder.startRecording(toOutputFile: file, delegate: self)
// Start recording setup and window configuration in parallel
async let recordingTask = self.recorder.startRecording(toOutputFile: file, delegate: self)
async let windowConfigTask = ActiveWindowService.shared.applyConfigurationForCurrentApp()
// Start model loading in parallel if needed
async let modelLoadingTask: Void = {
if let currentModel = await self.currentModel, await self.whisperContext == nil {
logger.notice("🔄 Loading model in parallel with recording: \(currentModel.name)")
do {
try await self.loadModel(currentModel)
} catch {
logger.error("❌ Model preloading failed: \(error.localizedDescription)")
await MainActor.run {
self.messageLog += "Error preloading model: \(error.localizedDescription)\n"
}
}
}
}()
// Wait for recording and window configuration
try await recordingTask
await windowConfigTask
self.isRecording = true
self.isVisualizerActive = true
self.recordedFile = file
self.transcriptionStartTime = Date()
// Wait for window configuration to complete
await windowConfigTask
// Start background tasks for model loading and screen capture
Task.detached(priority: .background) {
await self.performBackgroundTasks()
// After recording and window config are done, handle enhancement service
if let enhancementService = self.enhancementService,
enhancementService.isEnhancementEnabled &&
enhancementService.useScreenCaptureContext {
await enhancementService.captureScreenContext()
}
// Wait for model loading to complete (this won't block the UI)
await modelLoadingTask
} catch {
self.messageLog += "\(error.localizedDescription)\n"
self.isRecording = false
@ -169,26 +189,6 @@ class WhisperState: NSObject, ObservableObject, AVAudioRecorderDelegate {
}
}
private func performBackgroundTasks() async {
if let currentModel = self.currentModel, self.whisperContext == nil {
logger.notice("🔄 Preloading model in background: \(currentModel.name)")
do {
try await self.loadModel(currentModel)
} catch {
logger.error("❌ Background model preloading failed: \(error.localizedDescription)")
await MainActor.run {
self.messageLog += "Error preloading model: \(error.localizedDescription)\n"
}
}
}
if let enhancementService = self.enhancementService,
enhancementService.isEnhancementEnabled &&
enhancementService.useScreenCaptureContext {
await enhancementService.captureScreenContext()
}
}
private func requestRecordPermission(response: @escaping (Bool) -> Void) {
#if os(macOS)
response(true)