File size: 2,007 Bytes
0711651 |
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 42 43 44 45 46 47 48 49 50 51 52 53 |
import os
from TTS.api import TTS
from datetime import datetime
class TextToSpeech:
def __init__(self, model_name="tts_models/en/ljspeech/vits", device="cpu"):
"""
Initialize the TTS model.
:param model_name: The name of the TTS model to use.
:param device: The device to run the model on ("cuda" for GPU, "cpu" for CPU).
"""
self.model_name = model_name
self.device = device
self.tts = TTS(model_name).to(device)
def synthesize(self, text, output_dir="output_audio"):
"""
Convert text to speech and save as a .wav file.
:param text: The text to convert to speech.
:param output_dir: Directory to save the audio file.
:return: The path of the saved audio file.
"""
# Create output directory if it doesn't exist
os.makedirs(output_dir, exist_ok=True)
# Generate output filename with timestamp
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
if not output_dir.endswith(".wav"): # Ensure correct file extension
output_file = os.path.join(output_dir, f"tts_output_{timestamp}.wav")
else:
output_file = output_dir
# Generate speech with custom parameters
self.tts.tts_to_file(
text=text,
file_path=output_file,
speed=1.4, # Adjust speed (1.0 = normal)
noise_scale=0.8, # Control expressiveness
noise_scale_w=0.5, # Control speech rhythm variation
length_scale=1.1 # Control speech speed and pauses
)
print(f"Speech synthesis complete. Saved as {output_file}.")
return output_file
# Example usage
""" if __name__ == "__main__":
tts = TextToSpeech()
text_input = "Hello! This is a test for text-to-speech conversion."
output_path = tts.synthesize(text_input)
print("\nGenerated Audio File:", output_path) """
|