first

converters.py

@@ -0,0 +1,33 @@
+from decimal import Decimal
+from typing import Dict, Union, List
+
+from helpers import tag_words, PROPER_NOUN_TAGS
+
+
+def speechmatics_converter(data: Dict[str, Union[Dict[str, str], List[Dict[str, str]]]]):
+    converted_words = []
+    words = data['words']
+    tagged_words = tag_words([w['name'] for w in words])
+
+    for index, w in enumerate(words):
+        word_start = Decimal(w['time'])
+        word_end = word_start + Decimal(w['duration'])
+        confidence = Decimal(w['confidence'])
+        word = w['name']
+        space = '' if word == '.' else ' '
+        is_proper_noun = tagged_words[index][1] in PROPER_NOUN_TAGS
+        converted_words.append({
+            'wordStart': word_start,
+            'wordEnd': word_end,
+            'confidence': confidence,
+            'word': word,
+            'space': space,
+            'alwaysCapitalized': is_proper_noun or word == 'I',
+            'index': index,
+        })
+    return converted_words
+
+
+converters = {
+    'speechmatics': speechmatics_converter,
+}