import socket
import json
import builtins

HOST = "host.docker.internal"
PORT = 6050

def connect(data):
    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
        s.connect((HOST, PORT))
        data = data + "\n"
        s.sendall(data.encode("utf-8"))
        response = s.recv(4096)
        return response.decode("utf-8")

def print_output(text):
    data = {"function": "print_output", "container_id": CONTAINER_ID, "text": text}
    connect(json.dumps(data))

def _neurodock_print(*args, **kwargs):
    text = " ".join(str(a) for a in args)
    print_output(text)

builtins.print = _neurodock_print

## Following is the generated