diff --git a/main.py b/main.py index 67d7538..5e98963 100644 --- a/main.py +++ b/main.py @@ -3,21 +3,59 @@ import sys from src.myontology import MyOntology -from src.nlp import NLP -from src.parse_tree.parse_tree import ParseTree -from src.syntax import Syntax + +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) - print(f'Extracted terms:\n{", ".join(terms)}') - result: str = MyOntology().get_event_description(terms) - print(f'Result:\n{result}') + # 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__':