From 03f579dd241c78f698456cd2364dd9d06fca0899 Mon Sep 17 00:00:00 2001
From: anthony lloyd <contact@anthonylloyd.net>
Date: Fri, 15 Jul 2022 00:06:04 +1000
Subject: [PATCH] no conflicts version

---
 TTS/engine_wrapper.py | 105 ++++++++++++++++++++++++++++++------------
 1 file changed, 75 insertions(+), 30 deletions(-)

diff --git a/TTS/engine_wrapper.py b/TTS/engine_wrapper.py
index c5ed714..baeea7a 100644
--- a/TTS/engine_wrapper.py
+++ b/TTS/engine_wrapper.py
@@ -1,4 +1,5 @@
 #!/usr/bin/env python3
+import os
 from pathlib import Path
 from typing import Tuple
 import re
@@ -6,7 +7,10 @@ import re
 # import sox
 # from mutagen import MutagenError
 # from mutagen.mp3 import MP3, HeaderNotFoundError
+import numpy as np
 import translators as ts
+from moviepy.audio.AudioClip import AudioClip
+from moviepy.audio.fx.volumex import volumex
 from rich.progress import track
 from moviepy.editor import AudioFileClip, CompositeAudioClip, concatenate_audioclips
 from utils.console import print_step, print_substep
@@ -86,42 +90,83 @@ class TTSEngine:
             )
         ]
         offset = 0
+        try:
+            silence_duration = settings.config["settings"]["tts"]["silence_duration"]
+        except AttributeError:
+            silence_duration = 0.3
+        silence = AudioClip(make_frame=lambda t: np.sin(440 * 2 * np.pi * t), duration=silence_duration, fps=44100)
+        silence = volumex(silence, 0)
+        silence.write_audiofile(f"{self.path}/silence.mp3", fps=44100, verbose=False, logger=None)
+
+        idy = None
         for idy, text_cut in enumerate(split_text):
             # print(f"{idx}-{idy}: {text_cut}\n")
-            if not text_cut or text_cut.isspace():
-                offset += 1
+            newtext = process_text(text_cut)
+            if not newtext or newtext.isspace():
+                print("newtext was blank because sanitized split text resulted in none")
                 continue
-
-            self.call_tts(f"{idx}-{idy - offset}.part", text_cut)
-            split_files.append(AudioFileClip(f"{self.path}/{idx}-{idy - offset}.part.mp3"))
-
-        CompositeAudioClip([concatenate_audioclips(split_files)]).write_audiofile(
-            f"{self.path}/{idx}.mp3", fps=44100, verbose=False, logger=None
-        )
-
-        for i in split_files:
-            name = i.filename
-            i.close()
-            Path(name).unlink()
-
-        # for i in range(0, idy + 1):
-        # print(f"Cleaning up {self.path}/{idx}-{i}.part.mp3")
-
-        # Path(f"{self.path}/{idx}-{i}.part.mp3").unlink()
+            else:
+                self.call_tts(f"{idx}-{idy}.part", newtext)
+                with open(f"{self.path}/list.txt", 'w') as f:
+                    for idz in range(0, len(split_text)):
+                        f.write("file " + f"'{idx}-{idz}.part.mp3'" + "\n")
+                        split_files.append(str(f"{self.path}/{idx}-{idy}.part.mp3"))
+                    f.write("file " + f"'silence.mp3'" + "\n")
+                f.close()
+
+                os.system("ffmpeg -f concat -y -hide_banner -loglevel panic -safe 0 " +
+                          "-i " + f"{self.path}/list.txt " +
+                          "-c copy " + f"{self.path}/{idx}.mp3")
+                try:
+                    name = rf"{self.path}/{idx}-{idz}.part.mp3"
+                    os.unlink(name)
+                except FileNotFoundError:
+                    print("file not found error")
+                except OSError:
+                    print("OSError")
 
     def call_tts(self, filename: str, text: str):
-        self.tts_module.run(text=process_text(text), filepath=f"{self.path}/{filename}.mp3")
-        # try:
-        #     self.length += MP3(f"{self.path}/{filename}.mp3").info.length
-        # except (MutagenError, HeaderNotFoundError):
-        #     self.length += sox.file_info.duration(f"{self.path}/{filename}.mp3")
-        try:
-            clip = AudioFileClip(f"{self.path}/{filename}.mp3")
-            self.length += clip.duration
-            clip.close()
-        except:
-            self.length = 0
 
+        if filename == "title":
+            try:
+                self.tts_module.run(text, filepath=f"{self.path}/title_no_silence.mp3")
+                try:
+                    silence_duration = settings.config["settings"]["tts"]["silence_duration"]
+                except AttributeError:
+                    silence_duration = 0.3
+                silence = AudioClip(make_frame=lambda t: np.sin(440 * 2 * np.pi * t), duration=silence_duration,
+                                         fps=44100)
+                silence = volumex(silence, 0)
+                silence.write_audiofile(f"{self.path}/silence.mp3", fps=44100, verbose=False, logger=None)
+
+                with open(f"{self.path}/title.txt", 'w') as f:
+                    f.write("file " + f"'title_no_silence.mp3'" + "\n")
+                    f.write("file " + f"'silence.mp3'" + "\n")
+                f.close()
+                os.system("ffmpeg -f concat -y -hide_banner -loglevel panic -safe 0 " +
+                          "-i " + f"{self.path}/title.txt " +
+                          "-c copy " + f"{self.path}/title.mp3")
+                clip = AudioFileClip(f"{self.path}/title.mp3")
+                self.length += clip.duration
+                clip.close()
+                try:
+                    name = ["title_no_silence.mp3", "silence.mp3", "title.txt"]
+                    for i in range(0, len(name)):
+                        os.unlink(str(rf"{self.path}/"+name[i]))
+                except FileNotFoundError:
+                    print("file not found error")
+                except OSError:
+                    print("OSError")
+            except:
+                self.length = 0
+        else:
+            try:
+                self.tts_module.run(text=text, filepath=f"{self.path}/{filename}.mp3")
+                clip = AudioFileClip(f"{self.path}/{filename}.mp3")
+                self.length += clip.duration
+                clip.close()
+            except:
+                self.length = 0
 
 def process_text(text: str):
     lang = settings.config["reddit"]["thread"]["post_lang"]