70 lines
3.1 KiB
Python
70 lines
3.1 KiB
Python
#!/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])
|