import time
from eden.client import Client
from eden.datatypes import Image

import subprocess
import socket

# Get IP address of eden-server service
hostname = 'eden-server'
port = 5656
network_name = 'eden-network'
import docker
client = docker.from_env()
project_name = 'not_so_minimal'
container_name = f'{project_name}_{hostname}_1'
container = client.containers.get(container_name)
ip_address = container.attrs['NetworkSettings']['Networks'][network_name]['IPAddress']
print(ip_address)
url = f"http://{ip_address}:{port}"

## set up a client
c = Client(url=url, username="abraham")

# get server's identity
generator_id = c.get_generator_identity()
print(generator_id)

## define input args to be sent
config = {
    "width": 2000,  ## width
    "height": 1000,  ## height
    "input_image": Image(
        "/home/mm/Downloads/FF06F0EC-1B54-458A-BF12-FF7FC2A43C10.jpeg"
    ),  ## images require eden.datatypes.Image()
}

# start the task
run_response = c.run(config)

print("Intitial response")
# check status of the task, returns the output too if the task is complete
results = c.fetch(token=run_response["token"])
print(results)

# one eternity later
# time.sleep(5)

print("Trying")
while results["status"].get("status") != "complete":
    results = c.fetch(token=run_response["token"])
    print(results)
    time.sleep(0.1)

## check status again, hopefully the task is complete by now
# results = c.fetch(token=run_response["token"])
# print(results)
# results['output']['image'].show()