import cv2
import os
y_folder = "./Input_Images"

def crop_and_save_images(folder_path):
    # Get a list of all files in the folder
    files = os.listdir(folder_path)
    
    for file in files:
        # Construct the full file path
        file_path = os.path.join(folder_path, file)
        
        # Load the image
        img = cv2.imread(file_path)
        
        # Get the dimensions of the image
        h, w = img.shape[:2]
        
        # Determine the size of the crop
        crop_size = min(h, w)
        
        # Calculate the start coordinates of the crop
        start_y = (h - crop_size) // 2
        start_x = (w - crop_size) // 2
        
        # Perform the crop
        img_cropped = img[start_y : start_y + crop_size, start_x : start_x + crop_size]
        
        # Save the cropped image, overwriting the original image
        cv2.imwrite(file_path, img_cropped)

# Example usage:
crop_and_save_images(y_folder)