Add swift online punctuation (#1661)

swift-api-examples/SherpaOnnx.swift

@@ -1095,6 +1095,52 @@ class SherpaOnnxOfflinePunctuationWrapper {
   }
 }
 
+func sherpaOnnxOnlinePunctuationModelConfig(
+  cnnBiLstm: String,
+  bpeVocab: String,
+  numThreads: Int = 1,
+  debug: Int = 0,
+  provider: String = "cpu"
+) -> SherpaOnnxOnlinePunctuationModelConfig {
+  return SherpaOnnxOnlinePunctuationModelConfig(
+    cnn_bilstm: toCPointer(cnnBiLstm),
+    bpe_vocab: toCPointer(bpeVocab),
+    num_threads: Int32(numThreads),
+    debug: Int32(debug),
+    provider: toCPointer(provider))
+}
+
+func sherpaOnnxOnlinePunctuationConfig(
+  model: SherpaOnnxOnlinePunctuationModelConfig
+) -> SherpaOnnxOnlinePunctuationConfig {
+  return SherpaOnnxOnlinePunctuationConfig(model: model)
+}
+
+class SherpaOnnxOnlinePunctuationWrapper {
+  /// A pointer to the underlying counterpart in C
+  let ptr: OpaquePointer!
+
+  /// Constructor taking a model config
+  init(
+    config: UnsafePointer<SherpaOnnxOnlinePunctuationConfig>!
+  ) {
+    ptr = SherpaOnnxCreateOnlinePunctuation(config)
+  }
+
+  deinit {
+    if let ptr {
+      SherpaOnnxDestroyOnlinePunctuation(ptr)
+    }
+  }
+
+  func addPunct(text: String) -> String {
+    let cText = SherpaOnnxOnlinePunctuationAddPunct(ptr, toCPointer(text))
+    let ans = String(cString: cText!)
+    SherpaOnnxOnlinePunctuationFreeText(cText)
+    return ans
+  }
+}
+
 func sherpaOnnxOfflineSpeakerSegmentationPyannoteModelConfig(model: String)
   -> SherpaOnnxOfflineSpeakerSegmentationPyannoteModelConfig
 {

swift-api-examples/add-punctuation-online.swift

@@ -0,0 +1,35 @@
+func run() {
+    let model = "./sherpa-onnx-online-punct-en-2024-08-06/model.onnx"
+    let bpe = "./sherpa-onnx-online-punct-en-2024-08-06/bpe.vocab"
+    
+    // Create model config
+    let modelConfig = sherpaOnnxOnlinePunctuationModelConfig(
+        cnnBiLstm: model,
+        bpeVocab: bpe
+    )
+    
+    // Create punctuation config
+    var config = sherpaOnnxOnlinePunctuationConfig(model: modelConfig)
+    
+    // Create punctuation instance
+    let punct = SherpaOnnxOnlinePunctuationWrapper(config: &config)
+    
+    // Test texts
+    let textList = [
+        "how are you i am fine thank you",
+        "The African blogosphere is rapidly expanding bringing more voices online in the form of commentaries opinions analyses rants and poetry"
+    ]
+    
+    // Process each text
+  for i in 0..<textList.count {
+    let t = punct.addPunct(text: textList[i])
+    print("\nresult is:\n\(t)")
+  }
+}
+
+@main
+struct App {
+  static func main() {
+    run()
+  }
+}

swift-api-examples/run-add-punctuations-online.sh

@@ -0,0 +1,36 @@
+#!/usr/bin/env bash
+
+set -ex
+
+if [ ! -d ../build-swift-macos ]; then
+  echo "Please run ../build-swift-macos.sh first!"
+  exit 1
+fi
+
+# Download and extract the online punctuation model if not exists
+if [ ! -d ./sherpa-onnx-online-punct-en-2024-08-06 ]; then
+  curl -SL -O https://github.com/k2-fsa/sherpa-onnx/releases/download/punctuation-models/sherpa-onnx-online-punct-en-2024-08-06.tar.bz2
+  tar xvf sherpa-onnx-online-punct-en-2024-08-06.tar.bz2
+  rm sherpa-onnx-online-punct-en-2024-08-06.tar.bz2
+fi
+
+if [ ! -e ./add-punctuation-online ]; then
+  # Note: We use -lc++ to link against libc++ instead of libstdc++
+  swiftc \
+    -lc++ \
+    -I ../build-swift-macos/install/include \
+    -import-objc-header ./SherpaOnnx-Bridging-Header.h \
+    ./add-punctuation-online.swift  ./SherpaOnnx.swift \
+    -L ../build-swift-macos/install/lib/ \
+    -l sherpa-onnx \
+    -l onnxruntime \
+    -o ./add-punctuation-online
+
+  strip ./add-punctuation-online
+else
+  echo "./add-punctuation-online exists - skip building"
+fi
+
+# Set library path and run the executable
+export DYLD_LIBRARY_PATH=$PWD/../build-swift-macos/install/lib:$DYLD_LIBRARY_PATH
+./add-punctuation-online