import socket
import sys
import cv2
import pickle
import numpy as np
import struct ## new
import zlib
# pt imagini
from PIL import Image

# HOST=''
HOST='0.0.0.0'
# PORT=8485
PORT=8080


s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
print('Socket created')

s.bind((HOST,PORT))
print('Socket bind complete')
s.listen(10)
print('Socket now listening')

conn,addr=s.accept()

data = b""

# TODO: connection receive VIDEO SIZE
# TODO: connection receive {SIZE}
# TODO: connection receive START
# recevie image/video
# TODO: connection receive STOP

while True:
    while len(data) < payload_size:
        print("Recv: {}".format(len(data)))
        data += conn.recv(4096)

    print("Done Recv: {}".format(len(data)))
    packed_msg_size = data[:payload_size]
    data = data[payload_size:]
    msg_size = struct.unpack(">L", packed_msg_size)[0]
    print("msg_size: {}".format(msg_size))
    while len(data) < msg_size:
        data += conn.recv(4096)
    frame_data = data[:msg_size]
    data = data[msg_size:]

    frame=pickle.loads(frame_data, fix_imports=True, encoding="bytes")
    # merge doar pt imagini
    frame.show()