Python 调用 remove bg api 批量去除图片背景

import requests
import os
import logging
API_ENDPOINT = 'https://api.remove.bg/v1.0/removebg'
DEFAULT_OUTPUT_FOLDER = 'output'
class RemoveBg:
    def __init__(self, api_key):
        self.__api_key = api_key
    def remove_background_from_img_file(self, img_file_path, new_file_name=None):
        with open(img_file_path, 'rb') as file:
            response = requests.post(
                API_ENDPOINT,
                headers={'X-API-Key': self.__api_key},
                files={"image_file": file},
                data={'size': 'auto'},  # This line is added to request the image in its maximum size
                stream=True
            )
        if response.status_code == requests.codes.ok:
            with open(new_file_name if new_file_name else 'no-bg.png', 'wb') as out_file:
                for chunk in response.iter_content(chunk_size=8192):
                    out_file.write(chunk)
            print(f"Background removed for {img_file_path} and saved as {new_file_name}.")
        else:
            logging.error(f"Error processing {img_file_path}. Error: {response.text}")
    def process_images_in_folder(self, input_folder, output_folder):
        for file_name in os.listdir(input_folder):
            if file_name.endswith(('.png', '.jpg', '.jpeg')):
                img_path = os.path.join(input_folder, file_name)
                output_path = os.path.join(output_folder, file_name)
                self.remove_background_from_img_file(img_path, new_file_name=output_path)
def main():
    api_key = input("Please enter your remove.bg API key: ")
    remover = RemoveBg(api_key)
    input_folder = input("Enter the path to the folder containing images (default is current folder): ") or '.'
    output_folder = input(f"Enter the path for saving processed images (default is '{DEFAULT_OUTPUT_FOLDER}'): ") or DEFAULT_OUTPUT_FOLDER
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)
    remover.process_images_in_folder(input_folder, output_folder)
if __name__ == "__main__":
    main()