helper/main.py
2023-06-02 17:48:33 +04:00

66 lines
2.5 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

#!/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])