import time

import paho.mqtt.client as mqtt

import mqtt_client_image

BROKER = "127.0.0.1"
PORT = 1883
TOPIC = "worker/01"

client = mqtt.Client(client_id="python-publisher")
client.connect(BROKER, PORT, 60)

while True:
    choice = int(input("image > "))
    if choice == 1:
        message = bytes(mqtt_client_image.image1)
    elif choice == 2:
        message = bytes(mqtt_client_image.image2)
    client.publish(TOPIC, message)
    print(f"send")