import riva.client
from riva.client.argparse_utils import add_asr_config_argparse_parameters, add_connection_argparse_parameters
import os
from dotenv import load_dotenv


# Load environment variables from .env file
load_dotenv()

uri = "grpc.nvcf.nvidia.com:443"

auth = riva.client.Auth(
    uri=uri,
    use_ssl=True,
    metadata_args=[
        ["function-id", "b702f636-f60c-4a3d-a6f4-f3568c13bd7d"],
        ["authorization", f"Bearer {os.environ['NVIDIA_API']}"],
    ]
)

# assuming you already created `auth`
asr = riva.client.ASRService(auth)

# list all available ASR models
models = asr.list_models()

for m in models:
    print("Model name:", m.name)
    print("  Description:", m.description)
    print("  Type:", m.type)  # 'online' or 'offline'
    print("  Sample rates:", m.supported_sample_rates)
    print("  Languages:", m.languages)
    print()