57 lines
1.4 KiB
Python
57 lines
1.4 KiB
Python
from ..converter import TranscriptConverter
|
|
|
|
|
|
|
|
|
|
class GentleConverter(TranscriptConverter):
|
|
|
|
name = 'gentle'
|
|
transcript_type = dict
|
|
|
|
def __init__(self, json_data, language_code='en-US'):
|
|
super().__init__(json_data)
|
|
|
|
def get_word_objects(self, json_data):
|
|
return json_data['words']
|
|
|
|
@staticmethod
|
|
def get_word_start(word_object):
|
|
return word_object['start']
|
|
|
|
@staticmethod
|
|
def get_word_end(word_object):
|
|
return word_object['end']
|
|
|
|
@staticmethod
|
|
def get_word_confidence(word_object):
|
|
return 1
|
|
|
|
@staticmethod
|
|
def get_word_word(word_object):
|
|
return word_object['alignedWord']
|
|
|
|
def convert_words(self, word_objects, words, tagged_words=None):
|
|
converted_words = []
|
|
num_words = len(words)
|
|
|
|
for i, w in enumerate(word_objects):
|
|
word_obj = self.get_word_object(w, i, tagged_words, word_objects)
|
|
|
|
converted_words.append({
|
|
'start': word_obj.start,
|
|
'end': word_obj.end,
|
|
'confidence': word_obj.confidence,
|
|
'word': word_obj.word,
|
|
'always_capitalized': self.check_if_always_capitalized(
|
|
word_obj.word,
|
|
i,
|
|
tagged_words),
|
|
'punc_after': False,
|
|
'punc_before': False,
|
|
})
|
|
|
|
punc_after = False
|
|
|
|
return converted_words
|
|
|