import os
from PIL import Image
import glob


def crop_images_in_folder(input_folder, output_folder, first_image_top=0.50, first_image_bottom=0.92,
                          other_images_top=0.08, other_images_bottom=0.92):
    """
    裁剪输入文件夹中的图片：
    - 第一张图片裁剪掉上35%和下8%
    - 其他图片裁剪掉上8%和下8%
    并将裁剪后的图片保存到输出文件夹中。

    :param input_folder: 输入文件夹路径，包含要裁剪的图片
    :param output_folder: 输出文件夹路径，用于保存裁剪后的图片
    :param first_image_top: 第一张图片上边裁剪百分比（0到1之间的值）
    :param first_image_bottom: 第一张图片下边裁剪百分比（0到1之间的值）
    :param other_images_top: 其他图片上边裁剪百分比（0到1之间的值）
    :param other_images_bottom: 其他图片下边裁剪百分比（0到1之间的值）
    """
    # 确保输出文件夹存在
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)

    # 获取输入文件夹中的所有图片文件
    image_files = glob.glob(os.path.join(input_folder, '*'))
    image_files.sort()  # 按文件名排序，确保顺序一致

    for i, image_file in enumerate(image_files):
        try:
            # 打开图片
            image = Image.open(image_file)

            # 获取图片的宽度和高度
            width, height = image.size

            # 定义裁剪区域
            if i == 0:
                # 第一张图片的裁剪比例
                top_crop = int(height * first_image_top)
                bottom_crop = int(height * first_image_bottom)
            else:
                # 其他图片的裁剪比例
                top_crop = int(height * other_images_top)
                bottom_crop = int(height * other_images_bottom)

            # 裁剪图片
            cropped_image = image.crop((0, top_crop, width, bottom_crop))

            # 构建保存路径
            file_name = os.path.basename(image_file)
            output_file = os.path.join(output_folder, file_name)

            # 保存裁剪后的图片
            cropped_image.save(output_file)

            # 如果你想查看裁剪后的图片，可以取消下面的注释
            # cropped_image.show()

            print(f'Processed and saved: {output_file}')

        except Exception as e:
            print(f'Error processing file {image_file}: {e}')


def process_all_folders(base_input_folder, base_output_folder):
    """
    对基础输入文件夹下的每个子文件夹中的图片进行裁剪处理，并将结果保存到对应的输出文件夹中。

    :param base_input_folder: 基础输入文件夹路径，包含子文件夹
    :param base_output_folder: 基础输出文件夹路径，用于保存裁剪后的图片
    """
    # 获取基础输入文件夹中的所有子文件夹
    subfolders = [f.path for f in os.scandir(base_input_folder) if f.is_dir()]

    for subfolder in subfolders:
        # 构建对应的输出子文件夹路径
        folder_name = os.path.basename(subfolder)
        output_folder = os.path.join(base_output_folder, folder_name)

        # 执行裁剪操作
        crop_images_in_folder(subfolder, output_folder)


# 使用示例
base_input_folder = r'E:\Project\PaddleOCR\bilu_txt_dispose\bilu_img_2'
base_output_folder = r'E:\Project\PaddleOCR\bilu_txt_dispose\Crop_img'
process_all_folders(base_input_folder, base_output_folder)