|
1 | 1 | """Entrypoint for SM metrics""" |
| 2 | + |
2 | 3 | import argparse |
| 4 | +import sys |
| 5 | + |
| 6 | +try: |
| 7 | + import asr_metrics.wer.__main__ as wer_metrics |
| 8 | + |
| 9 | + WER_AVAILABLE = True |
| 10 | +except ImportError: |
| 11 | + WER_AVAILABLE = False |
| 12 | + |
| 13 | +try: |
| 14 | + import asr_metrics.diarization.sm_diarization_metrics.cookbook as diarization_metrics |
3 | 15 |
|
4 | | -import asr_metrics.diarization.sm_diarization_metrics.cookbook as diarization_metrics |
5 | | -import asr_metrics.wer.__main__ as wer_metrics |
| 16 | + DIARIZATION_AVAILABLE = True |
| 17 | +except ImportError: |
| 18 | + DIARIZATION_AVAILABLE = False |
6 | 19 |
|
7 | 20 |
|
8 | 21 | def main(): |
9 | | - parser = argparse.ArgumentParser(description="Your CLI description") |
| 22 | + parser = argparse.ArgumentParser( |
| 23 | + description="Speechmatics metrics tool for WER and diarization" |
| 24 | + ) |
10 | 25 |
|
11 | 26 | # Create subparsers |
12 | 27 | subparsers = parser.add_subparsers( |
13 | 28 | dest="mode", help="Metrics mode. Choose from 'wer' or 'diarization'" |
14 | 29 | ) |
15 | | - subparsers.required = True # Make sure a subparser id always provided |
| 30 | + subparsers.required = True # Make sure a subparser is always provided |
16 | 31 |
|
17 | | - wer_parser = subparsers.add_parser("wer", help="Entrypoint for WER metrics") |
18 | | - wer_metrics.get_wer_args(wer_parser) |
| 32 | + if WER_AVAILABLE: |
| 33 | + wer_parser = subparsers.add_parser("wer", help="Entrypoint for WER metrics") |
| 34 | + wer_metrics.get_wer_args(wer_parser) |
| 35 | + else: |
| 36 | + wer_parser = subparsers.add_parser( |
| 37 | + "wer", help="Entrypoint for WER metrics (requires additional dependencies)" |
| 38 | + ) |
19 | 39 |
|
20 | | - diarization_parser = subparsers.add_parser( |
21 | | - "diarization", help="Entrypoint for diarization metrics" |
22 | | - ) |
23 | | - diarization_metrics.get_diarization_args(diarization_parser) |
| 40 | + if DIARIZATION_AVAILABLE: |
| 41 | + diarization_parser = subparsers.add_parser( |
| 42 | + "diarization", help="Entrypoint for diarization metrics" |
| 43 | + ) |
| 44 | + diarization_metrics.get_diarization_args(diarization_parser) |
| 45 | + else: |
| 46 | + diarization_parser = subparsers.add_parser( |
| 47 | + "diarization", |
| 48 | + help="Entrypoint for diarization metrics (requires pyannote dependencies)", |
| 49 | + ) |
| 50 | + diarization_parser.add_argument( |
| 51 | + "--help-install", |
| 52 | + action="store_true", |
| 53 | + help="Show instructions for installing diarization dependencies", |
| 54 | + ) |
24 | 55 |
|
25 | 56 | args = parser.parse_args() |
26 | 57 |
|
27 | 58 | if args.mode == "wer": |
28 | | - wer_metrics.main(args) |
| 59 | + if WER_AVAILABLE: |
| 60 | + wer_metrics.main(args) |
| 61 | + else: |
| 62 | + print("Error: WER metrics require additional dependencies.") |
| 63 | + print("Please install them with: pip install speechmatics-python[metrics]") |
| 64 | + sys.exit(1) |
29 | 65 | elif args.mode == "diarization": |
30 | | - diarization_metrics.main(args) |
| 66 | + if DIARIZATION_AVAILABLE: |
| 67 | + diarization_metrics.main(args) |
| 68 | + else: |
| 69 | + print("Error: Diarization metrics require additional dependencies.") |
| 70 | + print("Please install them with: pip install speechmatics-python[metrics]") |
| 71 | + sys.exit(1) |
31 | 72 | else: |
32 | 73 | print("Unsupported mode. Please use 'wer' or 'diarization'") |
33 | 74 |
|
|
0 commit comments