import socket
import sys


HOST = "127.0.0.1"
PORT = 10001


message = "hello server"
if len(sys.argv) > 1:
    message = " ".join(sys.argv[1:])

client = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
client.sendto(message.encode("utf-8"), (HOST, PORT))

data, _ = client.recvfrom(1024)
print(f"Response from server: {data.decode('utf-8')}")

client.close()