Clean up

VoiceInk/Services/ParakeetTranscriptionService.swift

@@ -1,35 +1,28 @@
-import Foundation  
-import CoreML  
-import AVFoundation  
-import FluidAudio  
-import os.log  
-  
+import Foundation
+import CoreML
+import AVFoundation
+import FluidAudio
+import os.log
+
 class ParakeetTranscriptionService: TranscriptionService {
     private var asrManager: AsrManager?
     private var vadManager: VadManager?
     private var activeVersion: AsrModelVersion?
-    private let logger = Logger(
-        subsystem: "com.prakashjoshipax.voiceink.parakeet",
-        category: "ParakeetTranscriptionService"
-    )
-    
+    private let logger = Logger(subsystem: "com.prakashjoshipax.voiceink.parakeet", category: "ParakeetTranscriptionService")
+
     private func version(for model: any TranscriptionModel) -> AsrModelVersion {
         model.name.lowercased().contains("v2") ? .v2 : .v3
     }
 
     private func ensureModelsLoaded(for version: AsrModelVersion) async throws {
         if let manager = asrManager, activeVersion == version {
-            logger.notice("🦜 Parakeet \(version == .v2 ? "V2" : "V3") models already loaded, skipping")
             return
         }
 
         cleanup()
 
         let manager = AsrManager(config: .default)
-        let cacheDirectory = AsrModels.defaultCacheDirectory(for: version)
-        logger.notice("🦜 Loading Parakeet models from \(cacheDirectory.path)")
-        let models = try await AsrModels.load(
-            from: cacheDirectory,
+        let models = try await AsrModels.loadFromCache(
             configuration: nil,
             version: version
         )
@@ -41,78 +34,72 @@ class ParakeetTranscriptionService: TranscriptionService {
     func loadModel(for model: ParakeetModel) async throws {
         try await ensureModelsLoaded(for: version(for: model))
     }
-  
+
     func transcribe(audioURL: URL, model: any TranscriptionModel) async throws -> String {
         let targetVersion = version(for: model)
         try await ensureModelsLoaded(for: targetVersion)
-        
+
         guard let asrManager = asrManager else {
-            logger.notice("🦜 ASR manager not initialized, cannot transcribe")
             throw ASRError.notInitialized
         }
-          
+
         let audioSamples = try readAudioSamples(from: audioURL)
-  
+
         let durationSeconds = Double(audioSamples.count) / 16000.0
         let isVADEnabled = UserDefaults.standard.object(forKey: "IsVADEnabled") as? Bool ?? true
-  
-        let speechAudio: [Float]
-        if durationSeconds < 20.0 || !isVADEnabled {
-            speechAudio = audioSamples
-        } else {
+
+        var speechAudio = audioSamples
+        if durationSeconds >= 20.0, isVADEnabled {
             let vadConfig = VadConfig(threshold: 0.7)
             if vadManager == nil {
                 do {
                     vadManager = try await VadManager(config: vadConfig)
                 } catch {
-                    logger.notice("🦜 VAD initialization failed, using full audio: \(error.localizedDescription)")
+                    logger.notice("VAD init failed; falling back to full audio: \(error.localizedDescription)")
+                    vadManager = nil
                 }
             }
 
-            do {
-                if let vadManager {
+            if let vadManager {
+                do {
                     let segments = try await vadManager.segmentSpeechAudio(audioSamples)
                     speechAudio = segments.isEmpty ? audioSamples : segments.flatMap { $0 }
-                } else {
+                } catch {
+                    logger.notice("VAD segmentation failed; using full audio: \(error.localizedDescription)")
                     speechAudio = audioSamples
                 }
-            } catch {
-                logger.notice("🦜 VAD segmentation failed, using full audio: \(error.localizedDescription)")
-                speechAudio = audioSamples
-            }  
-        }  
-  
+            }
+        }
+
         let result = try await asrManager.transcribe(speechAudio)
-        
-        logger.notice("🦜 Parakeet transcription result: \(result.text)")
-          
-        return result.text  
-    }  
-  
-    private func readAudioSamples(from url: URL) throws -> [Float] {  
-        do {  
-            let data = try Data(contentsOf: url)  
-            guard data.count > 44 else {  
-                throw ASRError.invalidAudioData  
-            }  
-  
-            let floats = stride(from: 44, to: data.count, by: 2).map {  
-                return data[$0..<$0 + 2].withUnsafeBytes {  
-                    let short = Int16(littleEndian: $0.load(as: Int16.self))  
-                    return max(-1.0, min(Float(short) / 32767.0, 1.0))  
-                }  
-            }  
-              
-            return floats  
-        } catch {  
-            throw ASRError.invalidAudioData  
-        }  
-    }  
-      
+
+        return result.text
+    }
+
+    private func readAudioSamples(from url: URL) throws -> [Float] {
+        do {
+            let data = try Data(contentsOf: url)
+            guard data.count > 44 else {
+                throw ASRError.invalidAudioData
+            }
+
+            let floats = stride(from: 44, to: data.count, by: 2).map {
+                return data[$0..<$0 + 2].withUnsafeBytes {
+                    let short = Int16(littleEndian: $0.load(as: Int16.self))
+                    return max(-1.0, min(Float(short) / 32767.0, 1.0))
+                }
+            }
+
+            return floats
+        } catch {
+            throw ASRError.invalidAudioData
+        }
+    }
+
     func cleanup() {
         asrManager?.cleanup()
         asrManager = nil
         vadManager = nil
         activeVersion = nil
-    }  
+    }
 }