feat: show outputs location in verbose mode

This commit is contained in:
Ayo Ayco 2025-09-04 21:11:40 +02:00
parent 8cf2827b24
commit 06c1217e22

8
tts.py
View file

@ -7,13 +7,10 @@ import torch
import argparse import argparse
from kokoro import KPipeline from kokoro import KPipeline
import soundfile as sf import soundfile as sf
# import vlc
from tqdm import tqdm from tqdm import tqdm
import pyperclip import pyperclip
from yaspin import yaspin from yaspin import yaspin
# See voices: https://huggingface.co/hexgrad/Kokoro-82M/blob/main/VOICES.md # See voices: https://huggingface.co/hexgrad/Kokoro-82M/blob/main/VOICES.md
voices = { voices = {
'pro': 'af_heart', 'pro': 'af_heart',
@ -168,12 +165,11 @@ def main():
start_time = time() start_time = time()
output_files = generate_audio(generator, name, voice) output_files = generate_audio(generator, name, voice)
generation_time = time() - start_time generation_time = time() - start_time
directory,f = os.path.split(output_files[0])
if args.verbose: if args.verbose:
print(f"[TTS] {len(output_files)} chunks generated in {generation_time:.2f} seconds") print(f"[TTS] {len(output_files)} chunks generated in {generation_time:.2f} seconds")
print("[TTS] Now playing generated audio...") print(f"[TTS] Output files are in: {directory}/*")
directory,f = os.path.split(output_files[0])
if args.skip_play: if args.skip_play:
print(f"Audio player disabled: {directory}/*") print(f"Audio player disabled: {directory}/*")