| import argparse |
| import json |
| from pathlib import Path |
| import soundfile as sf |
| from .pipeline import StreamPAPipeline |
|
|
| def main(): |
| p=argparse.ArgumentParser(description='Append raw audio blocks and emit each StreamPA prefix') |
| p.add_argument('--audio',required=True) |
| p.add_argument('--bundle-dir',required=True) |
| p.add_argument('--device',default='cpu') |
| p.add_argument('--threads',type=int,default=2) |
| p.add_argument('--block-samples',type=int,default=1600) |
| p.add_argument('--output',required=True) |
| a=p.parse_args() |
| session=StreamPAPipeline(a.bundle_dir,a.device,a.threads).create_session() |
| dest=Path(a.output); dest.parent.mkdir(parents=True,exist_ok=True) |
| with sf.SoundFile(a.audio) as audio, dest.open('x',encoding='utf-8') as f: |
| if audio.samplerate!=16000 or audio.channels!=1: raise ValueError('Input must be mono 16000-Hz WAV') |
| while True: |
| block=audio.read(a.block_samples,dtype='float32') |
| if not len(block): break |
| for row in session.append(block): |
| f.write(json.dumps(row,allow_nan=False)+'\n'); f.flush() |
| f.write(json.dumps(session.finish(),allow_nan=False)+'\n') |
| print(json.dumps(dict(output=str(dest),prefixes=session.prefix_index))) |
|
|
| if __name__=='__main__': main() |
|
|
