UDP server (pixels lost)

# UDP Server with pixels lost
import socket
import pickle
from fl_networking_tools import ImageViewer


viewer = ImageViewer()

udp_server = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
#udp_server.bind(("0.0.0.0", 20001))
udp_server.bind(("127.0.0.1", 20001))

def get_pixel_data():
    received_pixels = 0

    lost_pixels = 0
    last_pixel_updated = (-1, -1)

    data, client_address = udp_server.recvfrom(1024)


    while True:
        data, client_address = udp_server.recvfrom(1024)
        message = pickle.loads(data)
        pos = message[0]
        rgba = message[1]
        received_pixels += 1
        viewer.put_pixel(pos, rgba)
        if (pos[0] - last_pixel_updated[0] > 1) or (pos[1] - last_pixel_updated[1] > 1):
            lost_pixels += 1
        viewer.text = " Pixels lost: " + str(lost_pixels)

        last_pixel_updated = pos
        if message == ((9,9), (0,0,0,0)):
            received_pixels -= 1
            break

    print("Pixels lost: " + str(lost_pixels))


viewer.start(get_pixel_data)