helper/main.py

#!/usr/bin/env python3

import sys

from src.myontology import MyOntology

data = [
    ['получить кредит обучение'],
    ['получить кредит образование'],
    ['купить товар рассрочка'],
    ['кредит ставка ипотека'],
    ['самая низкий ставка кредиту'],
    ['оформить кредит'],
    ['ставка кредит'],
    ['ипотека кредит ставка', 'какой ставка'],
    ['получить деньга карта', 'заблокировать'],
    ['снять деньга карта'],
    ['какой баланс карта'],
    ['лимит карта'],
    ['проходить оплата', 'быть деньга'],
    ['деньга карта'],
    ['деньга счёт'],
    ['разблокировать карта'],
    ['заблокировать карта'],
    ['какой условие ипотека'],
    ['какой условие кредит жильё'],
    ['какой срок кредит жильё'],
    ['какой срок ипотека'],
    ['потерять карта'],
    ['застрять банкомат'],
    ['украсть карта'],
    ['какой остаток карта'],
    ['купить дом'],
    ['купить дом'],
    ['купить товар кредит'],
    ['срок ипотека'],
    ['купить дом кредит'],
    ['снизить ставка кредиту'],
    ['рефинансировать кредит'],
    ['снизить ставка кредиту'],
    ['ли друг банка банк, перевести банк'],
    ['провести рефинансирование кредит'],
    ['рефинансировать ипотека']
]


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}')
    # terms = NLP().get_terms(parse_tree)
    for terms in data:
        print(f'Extracted terms:\n{", ".join(terms)}')
        result: str = MyOntology().get_event_description(terms)
        print(f'Result:\n{result}')
        print()


if __name__ == '__main__':
    if len(sys.argv) < 2:
        print(f'Usage: {sys.argv[0]} FILE')
        exit(1)
    _main(sys.argv[1])