29 lines
902 B
Python
29 lines
902 B
Python
import requests
|
|
|
|
from convlab.nlg.template.multiwoz import TemplateNLG
|
|
|
|
|
|
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
|
|
|
|
class NaturalLanguageGeneration:
|
|
info_dict = None
|
|
def generate(self, system_act):
|
|
if len(system_act) == 0:
|
|
return "Nie rozumiem."
|
|
tnlg = TemplateNLG(is_user=False)
|
|
response_en = tnlg.generate(system_act)
|
|
return translate_text(response_en)
|
|
|
|
|
|
def init_session(self):
|
|
pass # Dodanie pustej metody init_session
|