From 7bb36b5bac1a65c92d12de3623fac9e50e99d435 Mon Sep 17 00:00:00 2001 From: 464962 Date: Tue, 4 Jun 2024 21:53:50 +0200 Subject: [PATCH] add translation of system answer --- Main.py | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/Main.py b/Main.py index 2575c44..50bce14 100644 --- a/Main.py +++ b/Main.py @@ -1,9 +1,24 @@ +import requests + from NaturalLanguageAnalyzer import NaturalLanguageAnalyzer from DialoguePolicy import DialoguePolicy from DialogueStateTracker import DialogueStateTracker from convlab.dialog_agent import PipelineAgent from MachineLearningNLG import MachineLearningNLG # Importujemy nowy komponent NLG + +def translate_text(text, target_language='pl'): + url = 'https://translate.googleapis.com/translate_a/single?client=gtx&sl=auto&tl={}&dt=t&q={}'.format( + target_language, text) + response = requests.get(url) + if response.status_code == 200: + translated_text = response.json()[0] + translated_text_joined = ''.join([sentence[0] for sentence in translated_text]) + return translated_text_joined + else: + return None + + if __name__ == "__main__": text = "chciałbym zarezerwować drogi hotel z parkingiem 1 stycznia w Warszawie w centrum" nlu = NaturalLanguageAnalyzer() @@ -13,4 +28,4 @@ if __name__ == "__main__": agent = PipelineAgent(nlu=nlu, dst=dst, policy=policy, nlg=nlg, name='sys') response = agent.response(text) - print(response) + print(translate_text(response))