initial commit

ios/ExampleiOSApp/TTSService.swift

@@ -0,0 +1,114 @@
+import Foundation
+import OnnxRuntimeBindings
+
+final class TTSService {
+    enum Voice { case male, female }
+    enum Language: String, CaseIterable {
+        case en = "en"
+        case ko = "ko"
+        case es = "es"
+        case pt = "pt"
+        case fr = "fr"
+        
+        var displayName: String {
+            switch self {
+            case .en: return "English"
+            case .ko: return "한국어"
+            case .es: return "Español"
+            case .pt: return "Português"
+            case .fr: return "Français"
+            }
+        }
+    }
+
+    private let env: ORTEnv
+    private let textToSpeech: TextToSpeech
+    private let bundleOnnxDir: String
+    private let sampleRate: Int
+
+    init() throws {
+        bundleOnnxDir = try Self.locateOnnxDirInBundle()
+        env = try ORTEnv(loggingLevel: .warning)
+        textToSpeech = try loadTextToSpeech(bundleOnnxDir, false, env)
+        sampleRate = textToSpeech.sampleRate
+    }
+
+    func synthesize(text: String, nfe: Int, voice: Voice, language: Language) async throws -> URL {
+        // Load style for the selected voice
+        let styleURL = try Self.locateVoiceStyleURL(voice: voice)
+        let style = try loadVoiceStyle([styleURL.path], verbose: false)
+
+        // 2) Synthesize via packed TextToSpeech component
+        let (wav, duration) = try textToSpeech.call(text, language.rawValue, style, nfe)
+        let audioSeconds = Double(duration)
+        let wavLenSample = min(Int(Double(sampleRate) * audioSeconds), wav.count)
+        let wavOut = Array(wav[0..<wavLenSample])
+
+        let tmpURL = FileManager.default.temporaryDirectory.appendingPathComponent("supertonic_tts_\(UUID().uuidString).wav")
+        try writeWavFile(tmpURL.path, wavOut, sampleRate)
+
+        return tmpURL
+    }
+
+    // MARK: - Resource location helpers
+    private static func locateOnnxDirInBundle() throws -> String {
+        let bundle = Bundle.main
+        let fm = FileManager.default
+
+        func dirHasRequiredFiles(_ dir: URL) -> Bool {
+            let required = [
+                "tts.json",
+                "duration_predictor.onnx",
+                "text_encoder.onnx",
+                "vector_estimator.onnx",
+                "vocoder.onnx"
+            ]
+            return required.allSatisfy { fm.fileExists(atPath: dir.appendingPathComponent($0).path) }
+        }
+
+        var candidates: [URL] = []
+        if let dir = bundle.resourceURL?.appendingPathComponent("onnx", isDirectory: true) { candidates.append(dir) }
+        if let dir = bundle.resourceURL?.appendingPathComponent("assets/onnx", isDirectory: true) { candidates.append(dir) }
+        if let url = bundle.url(forResource: "tts", withExtension: "json", subdirectory: "onnx") { candidates.append(url.deletingLastPathComponent()) }
+        if let url = bundle.url(forResource: "tts", withExtension: "json", subdirectory: "assets/onnx") { candidates.append(url.deletingLastPathComponent()) }
+        if let url = bundle.url(forResource: "tts", withExtension: "json", subdirectory: nil) { candidates.append(url.deletingLastPathComponent()) }
+        if let root = bundle.resourceURL { candidates.append(root) }
+
+        for dir in candidates {
+            if dirHasRequiredFiles(dir) { return dir.path }
+        }
+        throw NSError(
+            domain: "TTS",
+            code: -100,
+            userInfo: [NSLocalizedDescriptionKey: "Could not find the onnx directory in the bundle. Please make sure the onnx folder (as a folder reference) is included in Copy Bundle Resources in Xcode."]
+        )
+    }
+
+    private static func locateVoiceStyleURL(voice: Voice) throws -> URL {
+        // Prefer M1/F1 defaults; search common subdirectories
+        let fileName = (voice == .male) ? "M1" : "F1"
+        let bundle = Bundle.main
+        let candidates: [URL?] = [
+            bundle.url(forResource: fileName, withExtension: "json", subdirectory: "voice_styles"),
+            bundle.url(forResource: fileName, withExtension: "json", subdirectory: "assets/voice_styles"),
+            bundle.url(forResource: fileName, withExtension: "json", subdirectory: nil)
+        ]
+        for url in candidates {
+            if let url = url { return url }
+        }
+        // Fallback: scan folders if needed
+        if let folder1 = bundle.resourceURL?.appendingPathComponent("voice_styles", isDirectory: true) {
+            let file = folder1.appendingPathComponent("\(fileName).json")
+            if FileManager.default.fileExists(atPath: file.path) { return file }
+        }
+        if let folder2 = bundle.resourceURL?.appendingPathComponent("assets/voice_styles", isDirectory: true) {
+            let file = folder2.appendingPathComponent("\(fileName).json")
+            if FileManager.default.fileExists(atPath: file.path) { return file }
+        }
+        throw NSError(
+            domain: "TTS",
+            code: -102,
+            userInfo: [NSLocalizedDescriptionKey: "Could not find the voice style JSON (\(fileName).json) in the bundle. Ensure voice_styles folder is included in Copy Bundle Resources."]
+        )
+    }
+}