Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions riva/client/realtime.py
Original file line number Diff line number Diff line change
Expand Up @@ -701,6 +701,9 @@ async def _update_session(self, timeout=1):
requested_voice = self.args.voice
input_text_synthesis["voice_name"] = requested_voice
overrides.append("voice_name")
else:
# Explicitly clear voice so the server selects based on language_code
input_text_synthesis["voice_name"] = ""
if input_text_synthesis:
session_payload["input_text_synthesis"] = input_text_synthesis

Expand Down
1 change: 1 addition & 0 deletions scripts/asr/realtime_asr_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@
add_asr_config_argparse_parameters,
add_realtime_config_argparse_parameters,
add_connection_argparse_parameters,
cli_main,
)
try:
from riva.client.argparse_utils import cli_main
Expand Down
6 changes: 1 addition & 5 deletions scripts/asr/riva_streaming_asr_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,12 +15,8 @@
from riva.client.argparse_utils import (
add_asr_config_argparse_parameters,
add_connection_argparse_parameters,
cli_main,
)
try:
from riva.client.argparse_utils import cli_main
except ImportError:
def cli_main(func):
return func


def parse_args() -> argparse.Namespace:
Expand Down
8 changes: 2 additions & 6 deletions scripts/asr/transcribe_file.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,13 +9,9 @@
from riva.client.argparse_utils import (
add_asr_config_argparse_parameters,
add_connection_argparse_parameters,
cli_main,
EXIT_BAD_INPUT,
)
try:
from riva.client.argparse_utils import cli_main, EXIT_BAD_INPUT
except ImportError:
EXIT_BAD_INPUT = 2
def cli_main(func):
return func


def parse_args() -> argparse.Namespace:
Expand Down
8 changes: 2 additions & 6 deletions scripts/asr/transcribe_file_offline.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,13 +10,9 @@
from riva.client.argparse_utils import (
add_asr_config_argparse_parameters,
add_connection_argparse_parameters,
cli_main,
EXIT_BAD_INPUT,
)
try:
from riva.client.argparse_utils import cli_main, EXIT_BAD_INPUT
except ImportError:
EXIT_BAD_INPUT = 2
def cli_main(func):
return func


def parse_args() -> argparse.Namespace:
Expand Down
8 changes: 2 additions & 6 deletions scripts/asr/transcribe_mic.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,13 +8,9 @@
from riva.client.argparse_utils import (
add_asr_config_argparse_parameters,
add_connection_argparse_parameters,
cli_main,
EXIT_BAD_INPUT,
)
try:
from riva.client.argparse_utils import cli_main, EXIT_BAD_INPUT
except ImportError:
EXIT_BAD_INPUT = 2
def cli_main(func):
return func

try:
import riva.client.audio_io
Expand Down
14 changes: 1 addition & 13 deletions scripts/nmt/nmt.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,13 +34,7 @@
import riva.client.proto.riva_nmt_pb2_grpc as riva_nmt_srv

import riva.client
from riva.client.argparse_utils import add_connection_argparse_parameters
try:
from riva.client.argparse_utils import cli_main, EXIT_BAD_INPUT
except ImportError:
EXIT_BAD_INPUT = 2
def cli_main(func):
return func
from riva.client.argparse_utils import add_connection_argparse_parameters, cli_main, EXIT_BAD_INPUT


def read_dnt_phrases_file(file_path):
Expand Down Expand Up @@ -146,7 +140,6 @@ def request(inputs):
if not os.path.exists(args.text_file):
print(f"Invalid input file path: {args.text_file}", file=sys.stderr)
return EXIT_BAD_INPUT
translated_any = False
with open(args.text_file, "r") as f:
batch = []
for line in f:
Expand All @@ -155,14 +148,9 @@ def request(inputs):
batch.append(line)
if len(batch) == args.batch_size:
request(batch)
translated_any = True
batch = []
if len(batch) > 0:
request(batch)
translated_any = True
if not translated_any:
print(f"{args.text_file} contained no non-empty lines", file=sys.stderr)
return EXIT_BAD_INPUT
return

if not args.text or not args.text.strip():
Expand Down
7 changes: 1 addition & 6 deletions scripts/nmt/nmt_speech_to_speech.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,12 +7,7 @@
import riva.client
import riva.client.proto.riva_asr_pb2 as riva_asr_pb2
import riva.client.proto.riva_nmt_pb2 as riva_nmt_pb2
from riva.client.argparse_utils import add_connection_argparse_parameters
try:
from riva.client.argparse_utils import cli_main
except ImportError:
def cli_main(func):
return func
from riva.client.argparse_utils import add_connection_argparse_parameters, cli_main


def parse_arguments():
Expand Down
7 changes: 1 addition & 6 deletions scripts/nmt/nmt_speech_to_text.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,12 +5,7 @@
import riva.client
import riva.client.proto.riva_asr_pb2 as riva_asr_pb2
import riva.client.proto.riva_nmt_pb2 as riva_nmt_pb2
from riva.client.argparse_utils import add_connection_argparse_parameters
try:
from riva.client.argparse_utils import cli_main
except ImportError:
def cli_main(func):
return func
from riva.client.argparse_utils import add_connection_argparse_parameters, cli_main

def parse_arguments():
parser = argparse.ArgumentParser(description='Riva Speech-to-Text Translation Client')
Expand Down
7 changes: 1 addition & 6 deletions scripts/tts/realtime_tts_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,12 +20,7 @@
import websockets
from websockets.exceptions import WebSocketException

from riva.client.argparse_utils import add_connection_argparse_parameters
try:
from riva.client.argparse_utils import cli_main
except ImportError:
def cli_main(func):
return func
from riva.client.argparse_utils import add_connection_argparse_parameters, cli_main

from riva.client.realtime import RealtimeClientTTS

Expand Down
12 changes: 5 additions & 7 deletions scripts/tts/talk.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,13 +9,11 @@
from pathlib import Path

import riva.client
from riva.client.argparse_utils import add_connection_argparse_parameters
try:
from riva.client.argparse_utils import cli_main, EXIT_BAD_INPUT
except ImportError:
EXIT_BAD_INPUT = 2
def cli_main(func):
return func
from riva.client.argparse_utils import (
add_connection_argparse_parameters,
cli_main,
EXIT_BAD_INPUT,
)
from riva.client.proto.riva_audio_pb2 import AudioEncoding

def read_file_to_dict(file_path):
Expand Down