blackteay/ffmpeg-python
1
0
Fork 0
mirror of https://github.com/kkroening/ffmpeg-python.git synced 2025-04-06 04:15:44 +08:00
Code Issues Projects Releases Wiki Activity

Add --out-file in transcribe.py

Browse Source
This commit is contained in:
Karl Kroening 2018-01-17 00:21:47 +00:00
parent 4bb2fc4999
commit dc531d181e
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
examples/transcribe.py
View File

@ -10,7 +10,6 @@ import ffmpeg
import logging import logging
import subprocess import subprocess
import sys import sys
import IPython
logging.basicConfig(level=logging.INFO, format='%(message)s') logging.basicConfig(level=logging.INFO, format='%(message)s')
@ -20,6 +19,8 @@ logger.setLevel(logging.INFO)
parser = argparse.ArgumentParser(description='Convert speech audio to text using Google Speech API') parser = argparse.ArgumentParser(description='Convert speech audio to text using Google Speech API')
parser.add_argument('in_filename', help='Input filename (`-` for stdin)') parser.add_argument('in_filename', help='Input filename (`-` for stdin)')
parser.add_argument('--out-file', type=argparse.FileType('w'), default='-',
help='Output filename (defaults to stdout)')
parser.add_argument('--timing', action='store_true', help='Include timing info') parser.add_argument('--timing', action='store_true', help='Include timing info')
@ -53,12 +54,12 @@ def get_transcripts(audio_data, include_timing_info=False):
return client.recognize(config, audio) return client.recognize(config, audio)
def transcribe(in_filename, include_timing_info=False): def transcribe(in_filename, out_file=sys.stdout, include_timing_info=False):
audio_data = decode_audio(in_filename) audio_data = decode_audio(in_filename)
response = get_transcripts(audio_data, include_timing_info) response = get_transcripts(audio_data, include_timing_info)
print(MessageToJson(response, sort_keys=True)) out_file.write(MessageToJson(response, sort_keys=True).encode('utf-8'))
if __name__ == '__main__': if __name__ == '__main__':
args = parser.parse_args() args = parser.parse_args()
transcribe(args.in_filename, args.timing) transcribe(args.in_filename, args.out_file, args.timing)