helper/src/main.py

#!/usr/bin/env python3

import sys
import warnings

from src.nlp import NLP
from src.ontology import Ontology
from syntax import Syntax

if not sys.warnoptions:
    warnings.simplefilter("ignore")


def _main():
    if len(sys.argv) < 2:
        print(f'Usage: {sys.argv[0]} FILE')
        exit(1)
    # wav_file = sys.argv[1]
    # speech_server = 'http://vosk.athene.tech'
    # text = Speech().run_recognition(wav_file, speech_server)
    text = 'не могу оплатить из-за ограничений карты'
    print(f'Text: {text}')
    syntax_server = 'http://syntaxnet.athene.tech'
    parse_tree = Syntax().get_parse_tree(text, syntax_server)
    print(f'Parse tree:\n{parse_tree}')
    nouns = NLP().get_nouns(parse_tree)
    print(f'Nouns:\n{nouns}')
    onto = Ontology()
    print(f'Test: {onto.get_event()}')


if __name__ == '__main__':
    _main()