Kiran Lonikar
58 lines
2.0 KiB
Python
58 lines
2.0 KiB
Python
from pathlib import Path
|
|
import sys
|
|
from whisper_live.client import TranscriptionClient
|
|
import argparse
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--port', '-p',
|
|
type=int,
|
|
default=9090,
|
|
help="Websocket port to run the server on.")
|
|
parser.add_argument('--server', '-s',
|
|
type=str,
|
|
default='localhost',
|
|
help='hostname or ip address of server')
|
|
parser.add_argument('--files', '-f',
|
|
type=str,
|
|
nargs='+',
|
|
help='hostname or ip address of server')
|
|
parser.add_argument('--output_file', '-o',
|
|
type=str,
|
|
default='./output_recording.wav',
|
|
help='hostname or ip address of server')
|
|
args = parser.parse_args()
|
|
|
|
# Validate audio files
|
|
valid_files = []
|
|
for file_path in args.files:
|
|
path = Path(file_path)
|
|
if path.exists() and path.is_file():
|
|
valid_files.append(str(path))
|
|
else:
|
|
print(f"Warning: File not found: {file_path}")
|
|
|
|
if not valid_files:
|
|
print("Error: No valid audio files found!")
|
|
sys.exit(1)
|
|
|
|
print(f"Found {len(valid_files)} audio file(s) to stream:")
|
|
for file_path in valid_files:
|
|
print(f" - {file_path}")
|
|
|
|
for f in valid_files:
|
|
client = TranscriptionClient(
|
|
args.server,
|
|
args.port,
|
|
lang="en",
|
|
translate=False,
|
|
model="large-v3", # also support hf_model => `Systran/faster-whisper-small`
|
|
use_vad=False,
|
|
save_output_recording=False, # Only used for microphone input, False by Default
|
|
output_recording_filename=args.output_file, # Only used for microphone input
|
|
max_clients=4,
|
|
max_connection_time=600,
|
|
mute_audio_playback=True, # Only used for file input, False by Default
|
|
)
|
|
client(f)
|