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