helper/main.py

#!/usr/bin/env python3

import sys

from src.speech import Speech
from src.myontology import MyOntology
from src.nlp import NLP
from src.parse_tree.parse_tree import ParseTree
from src.syntax import Syntax

test_data = [
    "как получить кредит на обучение",
    "Как получить кредит на образование?",
    "Можно ли купить товар в рассрочку?",
    "какая ставка кредита на ипотеку?",
    "Самая низкая ставка по кредиту?",
    "Как оформить кредит?",
    "Ставка кредита",
    "Какая ставка кредита на ипотеку?",
    "Как получить деньги с заблокированной карты?",
    "Как снять деньги с заблокированной карты",
    "Какой баланс карты",
    "Какой лимит карты",
    "Почему не проходит оплата, хотя деньги есть",
    "Сколько денег на карте",
    "Сколько денег на счете",
    "Как разблокировать карту?",
    "Как заблокировать карту?",
    "Какие условия по ипотеке?",
    "Какие условия кредита на жилье",
    "Какой срок кредита на жилье",
    "Какой срок ипотеки",
    "Я потеряла карту. Что делать?",
    "Карта застряла в банкомате. Что делать?",
    "У меня украли карту. Что делать?",
    "Какой остаток по карте?",
    "Хочу купить дом",
    "Хочу купить дом . Что делать?",
    "Можно ли купить товар в кредит?",
    "Какой срок по ипотеке",
    "Хочу купить дом в кредит Что делать?",
    "Можно ли снизить ставку по кредиту?",
    "Можно ли рефинансировать кредит?",
    "Можно ли снизить ставку по кредиту другого банка?",
    "У меня кредит в другом банке. Можно ли его перевести в ваш банк?",
    "Можно провести рефинансирование кредита?",
    "Можно ли рефинансировать ипотеку?",
]


def _main(wav_file: str):
    text: str = Speech().run_recognition(wav_file)
    # text: str = 'Как получить деньги с заблокированной карты?'
    # for text in test_data:
    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)
    print(f'Extracted terms:\n\t{", ".join(terms)}')
    result: str = MyOntology().get_events(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])