35 lines
1.0 KiB
Python
35 lines
1.0 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import sys
|
|
from typing import List
|
|
|
|
from src.myontology import MyOntology
|
|
from src.nlp import NLP
|
|
from src.parse_tree.parse_tree import ParseTree
|
|
from src.speech import Speech
|
|
from src.syntax import Syntax
|
|
|
|
|
|
def _main(wav_file: str):
|
|
text: str = Speech().run_recognition(wav_file)
|
|
# text: str = 'как получить кредит на обучение'
|
|
print(f'Text: {text}')
|
|
parse_tree: ParseTree = Syntax().get_parse_tree(text)
|
|
print(f'Parse tree:\n{parse_tree}')
|
|
nlp: NLP = NLP()
|
|
terms: List[str] = []
|
|
terms.extend(nlp.get_adj_noun(parse_tree))
|
|
terms.extend(nlp.get_nouns(parse_tree))
|
|
if len(terms) == 0:
|
|
terms.extend(nlp.get_terms_by_template(parse_tree, ['NOUN', 'VERB', 'ADJ']))
|
|
print(f'Extracted terms:\n{", ".join(terms)}')
|
|
result: str = MyOntology().get_event_description(terms)
|
|
print(f'Test:\n{result}')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
if len(sys.argv) < 2:
|
|
print(f'Usage: {sys.argv[0]} FILE')
|
|
exit(1)
|
|
_main(sys.argv[1])
|