34 lines
1018 B
Python
34 lines
1018 B
Python
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,
|
|
}
|