import socket
import cv2
import struct
import numpy as np


# Port to listen on
SERVER_PORT = 5555








# Create a TCP socket
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server_socket.bind(("", SERVER_PORT))
server_socket.listen(1)

# Accept client connection
client_socket, client_address = server_socket.accept()
print(f"Connected to client {client_address}")
count =0
while True:
    print('recievving')
    # Receive frame size (4 bytes integer)
    data = client_socket.recv(4)
    if not data:
            break
    frame_size = int.from_bytes(data, byteorder='big')

  # Receive the frame data (based on size)
    data = b''
    while len(data) < frame_size:
      packet = client_socket.recv(4096)
      if not packet:
        break
      data += packet

    # Decode the received frame data
    frame_encoded = np.frombuffer(data, dtype=np.uint8)
    frame = cv2.imdecode(frame_encoded, cv2.IMREAD_COLOR)

    # Process the frame here (optional)
    # You can display the frame, save it, or perform further analysis

    # Display the frame (optional)
    cv2.imshow('Received Frame', frame)
    cv2.imwrite(f'./imagesMQZ/frame{count}.jpeg',frame)
    if cv2.waitKey(1) & 0xFF == ord('q'):
            break
        

    # Check for client disconnect
    if not frame:
        break
    
server_socket.close()
cv2.destroyAllWindows()