From 73907026a72d37508624ca458fdaa687ea9abf84 Mon Sep 17 00:00:00 2001 From: PatatjeMC Date: Wed, 8 Jun 2022 20:31:25 +0200 Subject: [PATCH] Revert "Use sox for combining audio in TTS chunk combiner" This reverts commit 9d2c2cd88f58c348464187f3fe12799c3956cb1d. --- video_creation/TTSwrapper.py | 19 ++++++------------- 1 file changed, 6 insertions(+), 13 deletions(-) diff --git a/video_creation/TTSwrapper.py b/video_creation/TTSwrapper.py index 76a5a03..f5ae5e5 100644 --- a/video_creation/TTSwrapper.py +++ b/video_creation/TTSwrapper.py @@ -1,11 +1,10 @@ import re import base64 -import shutil import os import random import requests -import sox +from moviepy.editor import AudioFileClip, concatenate_audioclips, CompositeAudioClip #from profanity_filter import ProfanityFilter #pf = ProfanityFilter() # Code by @JasonLovesDoggo @@ -76,7 +75,6 @@ class TTTTSWrapper: # TikTok Text-to-Speech Wrapper chunks = [m.group().strip() for m in re.finditer(r' *((.{0,200})(\.|.$))', req_text)] audio_clips = [] - cbn = sox.Combiner() chunkId = 0 for chunk in chunks: @@ -84,21 +82,16 @@ class TTTTSWrapper: # TikTok Text-to-Speech Wrapper vstr = [r.json()["data"]["v_str"]][0] b64d = base64.b64decode(vstr) - with open(filename.replace(".mp3", f"-{chunkId}.mp3"), "wb") as out: + with open(f"{filename}-{chunkId}", "wb") as out: out.write(b64d) - audio_clips.append(filename.replace(".mp3", f"-{chunkId}.mp3")) + audio_clips.append(AudioFileClip(f"{filename}-{chunkId}")) chunkId = chunkId + 1 - if(len(audio_clips) > 1): - cbn.convert(samplerate=44100, n_channels=2) - cbn.build( - audio_clips, filename, 'concatenate' - ) - else: - os.rename(audio_clips[0], filename) - + audio_concat = concatenate_audioclips(audio_clips) + audio_composite = CompositeAudioClip([audio_concat]) + audio_composite.write_audiofile(filename, 44100, 2, 2000, None) @staticmethod def randomvoice():