import requests
import json


class MTUOCAinaTranslator:
    def __init__(self, ip: str, port: str):
        self.ip = ip
        self.port = port

    def translate(self, text, source_lang=None, target_lang=None):
        myobj = {
            'id': '1',
            'src': text,
        }
        url = f'http://{self.ip}:{self.port}/translate'
        #url = 'http://' + self.ip + ':' + self.port + '/translate'
        x = requests.post(url, json=myobj)
        json_response = json.loads(x.text)
        return json_response['tgt']