@ -8,7 +8,23 @@ from moviepy.audio.AudioClip import concatenate_audioclips, CompositeAudioClip
from moviepy . audio . io . AudioFileClip import AudioFileClip
from requests . exceptions import JSONDecodeError
voices = [ ' Brian ' , ' Emma ' , ' Russell ' , ' Joey ' , ' Matthew ' , ' Joanna ' , ' Kimberly ' , ' Amy ' , ' Geraint ' , ' Nicole ' , ' Justin ' , ' Ivy ' , ' Kendra ' , ' Salli ' , ' Raveena ' ]
voices = [
" Brian " ,
" Emma " ,
" Russell " ,
" Joey " ,
" Matthew " ,
" Joanna " ,
" Kimberly " ,
" Amy " ,
" Geraint " ,
" Nicole " ,
" Justin " ,
" Ivy " ,
" Kendra " ,
" Salli " ,
" Raveena " ,
]
# valid voices https://lazypy.ro/tts/
@ -16,29 +32,33 @@ voices = ['Brian', 'Emma', 'Russell', 'Joey', 'Matthew', 'Joanna', 'Kimberly', '
class POLLY :
def __init__ ( self ) :
self . url = ' https://streamlabs.com/polly/speak '
self . url = " https://streamlabs.com/polly/speak "
def tts (
self ,
req_text : str = " Amazon Text To Speech " ,
filename : str = " title.mp3 " ,
random_speaker = False ,
cense r= False ,
censo r= False ,
) :
if random_speaker :
voice = self . randomvoice ( )
else :
if not os . getenv ( ' VOICE ' ) :
return ValueError ( ' Please set the environment variable VOICE to a valid voice. options are: {} ' . format ( voices ) )
if not os . getenv ( " VOICE " ) :
return ValueError (
" Please set the environment variable VOICE to a valid voice. options are: {} " . format (
voices
)
)
voice = str ( os . getenv ( " VOICE " ) ) . capitalize ( )
body = { ' voice ' : voice , ' text ' : req_text , ' service ' : ' polly ' }
body = { " voice " : voice , " text " : req_text , " service " : " polly " }
response = requests . post ( self . url , data = body )
try :
voice_data = requests . get ( response . json ( ) [ ' speak_url ' ] )
with open ( filename , ' wb ' ) as f :
voice_data = requests . get ( response . json ( ) [ " speak_url " ] )
with open ( filename , " wb " ) as f :
f . write ( voice_data . content )
except ( KeyError , JSONDecodeError ) :
if response . json ( ) [ ' error ' ] == ' Text length is too long! ' :
if response . json ( ) [ " error " ] == " Text length is too long! " :
chunks = [
m . group ( ) . strip ( ) for m in re . finditer ( r " *((. { 0,499})( \ .|.$)) " , req_text )
]
@ -48,9 +68,9 @@ class POLLY:
chunkId = 0
for chunk in chunks :
body = { ' voice ' : voice , ' text ' : chunk , ' service ' : ' polly ' }
body = { " voice " : voice , " text " : chunk , " service " : " polly " }
resp = requests . post ( self . url , data = body )
voice_data = requests . get ( resp . json ( ) [ ' speak_url ' ] )
voice_data = requests . get ( resp . json ( ) [ " speak_url " ] )
with open ( filename . replace ( " .mp3 " , f " - { chunkId } .mp3 " ) , " wb " ) as out :
out . write ( voice_data . content )
@ -63,12 +83,14 @@ class POLLY:
cbn . build ( audio_clips , filename , " concatenate " )
else :
os . rename ( audio_clips [ 0 ] , filename )
except ( sox . core . SoxError ,
FileNotFoundError ) : # https://github.com/JasonLovesDoggo/RedditVideoMakerBot/issues/67#issuecomment-1150466339
except (
sox . core . SoxError ,
FileNotFoundError ,
) : # https://github.com/JasonLovesDoggo/RedditVideoMakerBot/issues/67#issuecomment-1150466339
for clip in audio_clips :
i = audio_clips . index ( clip ) # get the index of the clip
audio_clips = (
audio_clips [ : i ] + [ AudioFileClip ( clip ) ] + audio_clips [ i + 1 : ]
audio_clips [ : i ] + [ AudioFileClip ( clip ) ] + audio_clips [ i + 1 : ]
) # replace the clip with an AudioFileClip
audio_concat = concatenate_audioclips ( audio_clips )
audio_composite = CompositeAudioClip ( [ audio_concat ] )
@ -79,7 +101,7 @@ class POLLY:
Amazon Polly fails to read some symbols properly such as ' & (and) ' .
So we normalize input text before passing it to the service
"""
text = text . replace ( ' & ' , ' and ' )
text = text . replace ( " & " , " and " )
return text
def randomvoice ( self ) :
Jason
3 years ago
committed by
GitHub