From cb06fed9b6d55e8a66f322d229e5138768cba547 Mon Sep 17 00:00:00 2001 From: Aaron Chantrill Date: Wed, 26 Jun 2024 16:48:48 -0400 Subject: [PATCH] Phonetisaurus Fix Import sphinxvocab and PhonetisaurusG2P modules after installing phonetisaurus with pip. Add locale to default pocketsphinx hmm_dir. --- plugins/stt/pocketsphinx-stt/sphinxplugin.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/plugins/stt/pocketsphinx-stt/sphinxplugin.py b/plugins/stt/pocketsphinx-stt/sphinxplugin.py index b8a2f40..32c88d7 100644 --- a/plugins/stt/pocketsphinx-stt/sphinxplugin.py +++ b/plugins/stt/pocketsphinx-stt/sphinxplugin.py @@ -18,7 +18,13 @@ 'pip', 'install', 'phonetisaurus' ] completedprocess = run_command(cmd) - if completedprocess.returncode != 0: + if completedprocess.returncode == 0: + if importlib.util.find_spec("phonetisaurus"): + from . import sphinxvocab + from .g2p import PhonetisaurusG2P + else: + raise Exception("Phonetisaurus install failed") + else: # check what architecture we are on architecture = platform.machine() phonetisaurus_url = "" @@ -189,7 +195,7 @@ def settings(self): if (not hmm_dir): # Make a list of possible paths to check hmm_dir_paths = [ - paths.sub("pocketsphinx", "standard") + paths.sub("pocketsphinx", "standard", language) ] # see if any of these paths exist for path in hmm_dir_paths: