import os
import rembg

def process_images(input_dir, output_dir):
    if not os.path.exists(output_dir):
        os.makedirs(output_dir)

    for filename in os.listdir(input_dir):
        if filename.lower().endswith('.png'):
            input_path = os.path.join(input_dir, filename)
            output_path = os.path.join(output_dir, f"{os.path.splitext(filename)[0]}_output.png")

            with open(input_path, 'rb') as input_file, open(output_path, 'wb') as output_file:
                output_file.write(rembg.remove(input_file.read()))

if __name__ == "__main__":
    input_directory = os.getcwd()  # current directory
    output_directory = os.path.join(input_directory, 'alpha')

    process_images(input_directory, output_directory)