remove_background_image.py

import os
from datetime import datetime
from rembg import remove

class BackgroundRemover:
    def __init__(self, input_folder, output_folder):
        self.input_folder = input_folder
        self.output_folder = output_folder
    
    def process_images(self):
        today = datetime.now().strftime('%Y-%m-%d %H-%M-%S')
        processed_folder = os.path.join(self.output_folder, today)

        os.makedirs(processed_folder, exist_ok=True)
        for filename in os.listdir(self.input_folder):
            if filename.endswith(('.png', '.jpg', '.jpeg')):
                input_path = os.path.join(self.input_folder, filename)
                output_path = os.path.join(processed_folder, filename)
                self._remove_background(input_path, output_path)
                self._move_originals(input_path, processed_folder)
                print("proceso terminado")

    def _remove_background(self, input_p, output_p):
        with open(input_p, 'rb') as inp, open(output_p, 'wb') as outp:
            background_ouput = remove(inp.read())
            outp.write(background_ouput)

    def _move_originals(self, input_p, dest_p):
        original_folder = os.path.join(dest_p, 'originals')
        os.makedirs(original_folder, exist_ok=True)

        filename = os.path.basename(input_p)
        new_path = os.path.join(original_folder, filename)
        os.rename(input_p, new_path)

if __name__ == '__main__':
    input_folder = "input"
    output_folder = "output"
    remover = BackgroundRemover(input_folder, output_folder)
    remover.process_images()