Cleaned

tts.py

@@ -7,13 +7,22 @@ import shutil
 import argparse
 import torch
 
+'''
+    PLEASE NOTE:
+    tts_speed initially 1.0 and increases to 2.0 if necessary. This has been commented out for improved execution times due to CPU usage. 
+    Feel free to remove comments when used locally.
+'''
+
+# Accept TOS for tts
 os.environ["COQUI_TOS_AGREED"] = "1"
 
+# Adjust speed of audio segment
 def adjust_speed(input_file, speed_factor):
     output_file = input_file.replace(".wav", "_adjusted.wav")
     ffmpeg.input(input_file).filter('atempo', speed_factor).output(output_file, acodec='pcm_s16le').run()
     return output_file
 
+# Generate and process speech for each line
 def generate_speech(text, speaker_voice_map, output_file):
     combined_audio = AudioSegment.empty()
     temp_files = []
@@ -27,6 +36,7 @@ def generate_speech(text, speaker_voice_map, output_file):
         if not line.strip():
             continue
 
+        # Extract speaker ID, timestamps & text
         match = re.match(r"\[SPEAKER_(\d+)\] \[(\d+\.\d+)-(\d+\.\d+)\] (.+)", line)
         if not match:
             continue
@@ -39,21 +49,25 @@ def generate_speech(text, speaker_voice_map, output_file):
         if not speaker_wav:
             continue
 
+        # Create temp
         os.makedirs('./audio/temp', exist_ok=True)
         temp_file_path = f"./audio/temp/temp_output_part_{len(temp_files)}.wav"
         temp_files.append(temp_file_path)
 
+        # Initial TTS (original : 1.0 speed)
         tts_speed = 2.0 # original 1.0
         tts.tts_to_file(text=sentence, file_path=temp_file_path, speaker_wav=speaker_wav, language="es", speed=tts_speed)
 
         segment_audio = AudioSegment.from_wav(temp_file_path)
 
+        # Increase TTS speed if audio is longer than duration
         if segment_audio.duration_seconds * 1000 > segment_duration:
             #while tts_speed < 2.0 and segment_audio.duration_seconds * 1000 > segment_duration:
             #    tts_speed += 0.5
             #    tts.tts_to_file(text=sentence, file_path=temp_file_path, speaker_wav=speaker_wav, language="es", speed=tts_speed)
             #    segment_audio = AudioSegment.from_wav(temp_file_path)
 
+            # Speed up using FFmpeg if audio is longer than duration
             if segment_audio.duration_seconds * 1000 > segment_duration:
                 required_speed = segment_duration / (segment_audio.duration_seconds * 1000)
                 if required_speed < 1.0:
@@ -61,9 +75,11 @@ def generate_speech(text, speaker_voice_map, output_file):
                 temp_file_path = adjust_speed(temp_file_path, required_speed)
                 segment_audio = AudioSegment.from_wav(temp_file_path)
 
+        # Add silence at start of audio if needed
         if combined_audio.duration_seconds == 0 and start_time > 0:
             combined_audio = AudioSegment.silent(duration=start_time * 1000) + combined_audio
 
+        # Trim or pad audio to match segment duration (Should not trim since audio sped up)
         if segment_audio.duration_seconds * 1000 > segment_duration:
             segment_audio = segment_audio[:segment_duration]
         else:
@@ -71,11 +87,14 @@ def generate_speech(text, speaker_voice_map, output_file):
 
         combined_audio += segment_audio
 
+    # Export combined audio
     combined_audio.export(output_file, format="wav")
 
+    # Delete temp files
     for temp_file in temp_files:
         os.remove(temp_file)
 
+# Map speaker IDs to their voice files
 def map_speaker_ids(directory):
     speaker_voice_map = {}
     for file in os.listdir(directory):
@@ -85,10 +104,15 @@ def map_speaker_ids(directory):
     return speaker_voice_map
 
 def main(speaker_directory, aligned_text_file, output_audio_file):
+    # Generate speaker voice map and read translated text
     speaker_voice_map = map_speaker_ids(speaker_directory)
     with open(aligned_text_file, 'r') as file:
         translated_text = file.read()
+
+    # Generate speech
     generate_speech(translated_text, speaker_voice_map, output_audio_file)
+
+    # Remove temp folder
     if os.path.exists('./audio/temp'):
         shutil.rmtree('./audio/temp')