77 lines
3.0 KiB
Swift
77 lines
3.0 KiB
Swift
import Foundation
|
|
import os
|
|
|
|
enum CloudTranscriptionError: Error, LocalizedError {
|
|
case unsupportedProvider
|
|
case missingAPIKey
|
|
case invalidAPIKey
|
|
case audioFileNotFound
|
|
case apiRequestFailed(statusCode: Int, message: String)
|
|
case networkError(Error)
|
|
case noTranscriptionReturned
|
|
case dataEncodingError
|
|
|
|
var errorDescription: String? {
|
|
switch self {
|
|
case .unsupportedProvider:
|
|
return "The model provider is not supported by this service."
|
|
case .missingAPIKey:
|
|
return "API key for this service is missing. Please configure it in the settings."
|
|
case .invalidAPIKey:
|
|
return "The provided API key is invalid."
|
|
case .audioFileNotFound:
|
|
return "The audio file to transcribe could not be found."
|
|
case .apiRequestFailed(let statusCode, let message):
|
|
return "The API request failed with status code \(statusCode): \(message)"
|
|
case .networkError(let error):
|
|
return "A network error occurred: \(error.localizedDescription)"
|
|
case .noTranscriptionReturned:
|
|
return "The API returned an empty or invalid response."
|
|
case .dataEncodingError:
|
|
return "Failed to encode the request body."
|
|
}
|
|
}
|
|
}
|
|
|
|
class CloudTranscriptionService: TranscriptionService {
|
|
|
|
private lazy var groqService = GroqTranscriptionService()
|
|
private lazy var elevenLabsService = ElevenLabsTranscriptionService()
|
|
private lazy var deepgramService = DeepgramTranscriptionService()
|
|
private lazy var mistralService = MistralTranscriptionService()
|
|
private lazy var geminiService = GeminiTranscriptionService()
|
|
private lazy var openAICompatibleService = OpenAICompatibleTranscriptionService()
|
|
|
|
func transcribe(audioURL: URL, model: any TranscriptionModel) async throws -> String {
|
|
var text: String
|
|
|
|
switch model.provider {
|
|
case .groq:
|
|
text = try await groqService.transcribe(audioURL: audioURL, model: model)
|
|
case .elevenLabs:
|
|
text = try await elevenLabsService.transcribe(audioURL: audioURL, model: model)
|
|
case .deepgram:
|
|
text = try await deepgramService.transcribe(audioURL: audioURL, model: model)
|
|
case .mistral:
|
|
text = try await mistralService.transcribe(audioURL: audioURL, model: model)
|
|
case .gemini:
|
|
text = try await geminiService.transcribe(audioURL: audioURL, model: model)
|
|
case .custom:
|
|
guard let customModel = model as? CustomCloudModel else {
|
|
throw CloudTranscriptionError.unsupportedProvider
|
|
}
|
|
text = try await openAICompatibleService.transcribe(audioURL: audioURL, model: customModel)
|
|
default:
|
|
throw CloudTranscriptionError.unsupportedProvider
|
|
}
|
|
|
|
if UserDefaults.standard.object(forKey: "IsTextFormattingEnabled") as? Bool ?? true {
|
|
text = WhisperTextFormatter.format(text)
|
|
}
|
|
|
|
return text
|
|
}
|
|
|
|
|
|
|
|
} |