File size: 4,162 Bytes
c920f65 |
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 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 |
gradio_url="http://127.0.0.1:7860/"
from gradio_client import Client
import json
import os
import shutil
from rich.console import Console
from rich.text import Text
import simpleaudio as sa
# Initialize the Gradio client
client = Client(gradio_url)
console = Console()
voice_dict = None
voices = {}
output_dir = "api_output"
def get_voice_names():
global voice_dict
# Get the result from the client
result = client.predict(api_name="/get_voice_names")
# Convert the result string into a Python dictionary
voice_dict = json.loads(result)
id = 1
for key in voice_dict:
if len(voice_dict[key]) > 0:
for i in voice_dict[key]:
voices[id] = i
id += 1
return voice_dict
def display_voice_names():
global voice_dict
id_num = 1
for key in voice_dict:
if len(voice_dict[key]) > 0:
gender = key.replace("_", " ").capitalize()
console.print(f"[bold cyan]{gender}:[/bold cyan]")
for voice in voice_dict[key]:
console.print(f"[green]{id_num}. {voice}[/green]")
id_num += 1
# console.print()
def text_to_speech(
text="Hello!!",
model_name="kokoro-v0_19.pth",
voice_name="af_bella",
speed=1,
pad_between_segments=0,
remove_silence=False,
minimum_silence=0.05,
custom_voicepack=None,
):
# Call the API with provided parameters
result = client.predict(
text=text,
model_name=model_name,
voice_name=voice_name,
speed=speed,
pad_between_segments=pad_between_segments,
remove_silence=remove_silence,
minimum_silence=minimum_silence,
custom_voicepack=custom_voicepack,
api_name="/text_to_speech"
)
# Save the audio file in the specified directory
save_at = f"{output_dir}/{os.path.basename(result)}"
shutil.move(result, save_at)
return save_at
def choose_voice():
while True:
display_voice_names()
user_input = console.input("[bold yellow]Choose a voice ID number or type 'exit' to quit: [/bold yellow]")
try:
voice_number = int(user_input)
if voice_number in voices:
selected_voice = voices[voice_number]
console.print(f"[bold green]You selected: {selected_voice}[/bold green]\n")
return selected_voice
else:
console.print("[bold red]Invalid number. Please choose a valid voice number.[/bold red]")
except ValueError:
if user_input.lower() == "exit":
console.print("[bold red]Exiting voice selection.[/bold red]")
return None
else:
console.print("[bold red]Invalid input. Please enter a valid number or 'exit'.[/bold red]")
def play_audio(filename):
wave_obj = sa.WaveObject.from_wave_file(filename)
play_obj = wave_obj.play()
play_obj.wait_done()
# Ensure the output directory exists
os.makedirs(output_dir, exist_ok=True)
voice_dict = get_voice_names()
def text_to_speech_cli():
while True:
voice_name = choose_voice()
if not voice_name:
break
# print(
# "Type 'C' to change voice or 'q' to quit"
# )
while True:
text = console.input(
"[bold green]Enter Text ('C' to change voice, 'q' for quit): [/bold green]"
)
if text.lower() == "q":
console.print("[bold red]Exiting text-to-speech CLI.[/bold red]")
return
elif text.lower() == "c":
console.print("[bold yellow]Changing voice...[/bold yellow]")
break # Break the inner loop to select a new voice
else:
audio_path = text_to_speech(text=text, voice_name=voice_name)
play_audio(audio_path)
# console.print("[bold green]Audio played successfully![/bold green]\n")
if __name__ == "__main__":
console.print("[bold blue]Welcome to the Text-to-Speech CLI![/bold blue]\n")
text_to_speech_cli() |