aws-transcribe-labeling.py

#!/usr/bin/env python

import json
from dataclasses import dataclass
from pathlib import Path
from typing import List, Dict, Literal, Union, Optional

import typer


# #   "start_time": "1749.91",
# #   "speaker_label": "spk_0",
# #   "end_time": "1760.26",
# #   "items": [
# #     {
# #       "start_time": "1749.91",
# #       "speaker_label": "spk_0",
# #       "end_time": "1750.55"
# #     },
# @dataclass
# class Segments:
#     start_time: str
#     speaker_label: str
#     end_time: str
#     items: List[Dict[str, str]]

#   "start_time": "1759.77",
#   "speaker_label": "spk_0",
#   "end_time": "1760.25",
#   "alternatives": [
#     {
#       "confidence": "0.9997",
#       "content": "cinco"
#     }
#   ],
#   "type": "pronunciation"
@dataclass
class Item:
    alternatives: List[Dict[str, str]]
    type: Literal["pronunciation", "punctuation"]
    start_time: Optional[str] = None
    speaker_label: Optional[str] = None
    end_time: Optional[str] = None


def main(path: Path):
    j = json.loads(path.read_text())
    items = j['results']['items']

    current_speaker = None
    for i in items:
        item = Item(**i)
        if item.speaker_label and item.speaker_label != current_speaker:
            current_speaker = item.speaker_label
            print(f"\n\n{current_speaker}")

        if item.type == 'pronunciation':
            print(' ', end='')

        print(item.alternatives[0]['content'], end='')


if __name__ == "__main__":
    typer.run(main)