dependencies: [
    .package(url: "https://github.com/Mythie/WhisperSwift.git", from: "1.0.0")
]

import WhisperSwift

// Load the model
let modelURL = Bundle.main.url(forResource: "ggml-tiny.en", withExtension: "bin")!
let transcriber = try await Transcriber(modelPath: modelURL)

// Transcribe an audio file
let audioURL = Bundle.main.url(forResource: "recording", withExtension: "wav")!
let result = try await transcriber.transcribe(file: audioURL)

print(result.text)
// "And so my fellow Americans, ask not what your country can do for you..."

// Access segments with timing
for segment in result.segments {
    print("[\(segment.startTime)s]: \(segment.text)")
}

import WhisperSwift
import AVFoundation

// Create streaming transcriber
let transcriber = try await StreamingTranscriber(modelPath: modelURL)
try await transcriber.start()

// Set up AVAudioEngine
let audioEngine = AVAudioEngine()
let inputNode = audioEngine.inputNode
let format = inputNode.outputFormat(forBus: 0)

// Feed audio to transcriber
inputNode.installTap(onBus: 0, bufferSize: 4096, format: format) { buffer, _ in
    Task { try await transcriber.process(buffer: buffer) }
}

try audioEngine.start()

// Receive transcriptions as they arrive
for try await segment in transcriber.segments {
    print(segment.text)
}

// Clean up
audioEngine.stop()
let finalResult = try await transcriber.stop()

import WhisperSwift
import AVFoundation

// Create audio engine - you manage its lifecycle
let engine = AVAudioEngine()

// Create the beta streaming transcriber
let transcriber = try await BetaStreamingTranscriber(
    modelPath: modelURL,
    engine: engine,
    configuration: .lowLatency  // or .default for balanced latency/accuracy
)

// Start audio engine first
try engine.start()

// Start transcription
try await transcriber.start()

// Receive segments as they arrive
for try await segment in transcriber.segments {
    print(segment.text)        // Transcribed text
    print(segment.isPartial)   // true = may be updated, false = final
}

// Stop and get final text
let finalText = try await transcriber.stop()
engine.stop()

let options = TranscriptionOptions(
    language: .english,           // Specify language (faster than auto-detect)
    translate: false,             // Translate to English
    tokenTimestamps: true,        // Word-level timing
    initialPrompt: "Meeting notes:",  // Condition the model
    samplingStrategy: .greedy     // or .beamSearch(beamSize: 5)
)

let result = try await transcriber.transcribe(file: audioURL, options: options)

// Default: GPU + Flash Attention
let transcriber = try await Transcriber(
    modelPath: modelURL,
    configuration: .default
)

// CPU only
let cpuTranscriber = try await Transcriber(
    modelPath: modelURL,
    configuration: .cpuOnly
)

// Custom
let config = WhisperConfiguration(
    useGPU: true,
    useFlashAttention: true,
    threadCount: 4
)

// Configure silence detection (default chunking strategy)
let silenceOptions = SilenceDetectorOptions(
    threshold: 0.01,           // RMS threshold
    minSilenceDuration: 0.3,   // Minimum silence gap
    searchDuration: 5.0        // Search window
)

let transcriber = try await StreamingTranscriber(
    modelPath: modelURL,
    silenceDetectorOptions: silenceOptions,
    minAudioDuration: 1.0,     // Minimum before transcription
    maxAudioDuration: 30.0     // Maximum before forced transcription
)

// Specify a language
let options = TranscriptionOptions(language: .spanish)

// Auto-detect
let options = TranscriptionOptions(language: nil)  // or .auto

// Check detected language
if let detected = result.detectedLanguage {
    print("Detected: \(detected.displayName)")  // "Spanish"
}

do {
    let result = try await transcriber.transcribe(file: audioURL)
} catch let error as WhisperError {
    switch error {
    case .modelNotFound(let url):
        print("Model not found: \(url.path)")
    case .audioLoadFailed(let url, let underlying):
        print("Audio error: \(underlying)")
    case .transcriptionFailed(let reason):
        print("Transcription failed: \(reason)")
    default:
        print(error.localizedDescription)
    }
}