Spaces:
Running
on
Zero
Running
on
Zero
File size: 1,145 Bytes
fba9477 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
import re
from textstat import textstat
def contains_chinese(text):
# 正则表达式，用于匹配中文字符 + 数字 -> 都认为是 zh
if re.search(r'[\u4e00-\u9fff0-9]', text):
return True
return False
def get_text_syllable_num(text):
chinese_char_pattern = re.compile(r'[\u4e00-\u9fff]')
number_char_pattern = re.compile(r'[0-9]')
syllable_num = 0
tokens = re.findall(r'[\u4e00-\u9fff]+|[a-zA-Z]+|[0-9]+', text)
# print(tokens)
if contains_chinese(text):
for token in tokens:
if chinese_char_pattern.search(token) or number_char_pattern.search(token):
syllable_num += len(token)
else:
syllable_num += textstat.syllable_count(token)
else:
syllable_num = textstat.syllable_count(text)
return syllable_num
def get_text_tts_dur(text):
min_speed = 3 # 2.18 #
max_speed = 5.50
ratio = 0.8517 if contains_chinese(text) else 1.0
syllable_num = get_text_syllable_num(text)
max_dur = syllable_num * ratio / max_speed
min_dur = syllable_num * ratio / min_speed
return max_dur, min_dur |