diff --git a/main.py b/main.py index 5e98963..15ef141 100644 --- a/main.py +++ b/main.py @@ -3,61 +3,63 @@ 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 = [ - ['получить кредит обучение'], - ['получить кредит образование'], - ['купить товар рассрочка'], - ['кредит ставка ипотека'], - ['самая низкий ставка кредиту'], - ['оформить кредит'], - ['ставка кредит'], - ['ипотека кредит ставка', 'какой ставка'], - ['получить деньга карта', 'заблокировать'], - ['снять деньга карта'], - ['какой баланс карта'], - ['лимит карта'], - ['проходить оплата', 'быть деньга'], - ['деньга карта'], - ['деньга счёт'], - ['разблокировать карта'], - ['заблокировать карта'], - ['какой условие ипотека'], - ['какой условие кредит жильё'], - ['какой срок кредит жильё'], - ['какой срок ипотека'], - ['потерять карта'], - ['застрять банкомат'], - ['украсть карта'], - ['какой остаток карта'], - ['купить дом'], - ['купить дом'], - ['купить товар кредит'], - ['срок ипотека'], - ['купить дом кредит'], - ['снизить ставка кредиту'], - ['рефинансировать кредит'], - ['снизить ставка кредиту'], - ['ли друг банка банк, перевести банк'], - ['провести рефинансирование кредит'], - ['рефинансировать ипотека'] +test_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) + 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')