v6-place/main.py

import asyncio
import io
import os

from PIL import Image
from websockets import connect
from dotenv import load_dotenv

from differencing import get_pixel_differences
from network import upload

# Start a websocket
# In the initial image load, detect all changes between the target and

load_dotenv()

width, height = int(os.getenv("CANVAS_WIDTH")), int(os.getenv("CANVAS_HEIGHT"))


async def get_image(websocket):
    while True:
        data = await websocket.recv()
        if type(data) == bytes:
            return Image.open(io.BytesIO(data))


async def main():
    source_path = os.getenv("SOUCE_FILE")

    async with connect(os.getenv("WEBSOCKET_ADDRESS")) as websocket:
        while True:
            print(websocket.messages)

            original_image = Image.open(source_path)
            original_image = original_image.resize((width, height), Image.LANCZOS)

            source = await get_image(websocket)
            ips = get_pixel_differences(source, original_image)

            upload(ips)

            if len(ips) < 5:
                break

            original_image.close()

            break


if __name__ == "__main__":
    asyncio.run(main())