【初心者歓迎】画像の背景切り抜きツールを作ろう

import os
import customtkinter as ctk
from customtkintercontrols import App
from customtkintercontrols import FolderSelectionControl
from appconfig import AppConfig 
import manager 
# -----------------------------------------------
# コールバック関数の定義 
# -----------------------------------------------

# ウィンドウを閉じる時に呼ばれる関数
def close_window():
    # 画面入力値を保存
    conf.set("input_folder",input_folder.get())
    conf.set("output_folder",output_folder.get())
    conf.write()
    # プログラムの終了
    app.quit()

# 変換ボタンが押された時に呼ばれる関数
def remove():
    manager.bg_remove_from_folder(input_folder.get(),output_folder.get())
           
# -----------------------------------------------
# メインウインドウの定義
# -----------------------------------------------

# 前回入力値の保存・復元用インスタンスの生成
conf = AppConfig("./config.json")

# メインウィンドウの作成
app = App()
app.minsize(width=400, height=200)
app.maxsize(width=1200, height=200)
app.geometry(f"600x200")
app.title("画像の背景切り抜きツール")

# customtkinterをダークモードに設定
ctk.set_appearance_mode("light")

# ウィンドウを閉じる時に呼びたい関数を登録
app.protocol("WM_DELETE_WINDOW", close_window)

# -----------------------------------------------
# 画面レイアウト（ウィジェット）の定義 
# -----------------------------------------------

# フォルダ選択ダイアログを配置
input_folder = FolderSelectionControl(app)
input_folder.label.configure(text="入力フォルダ名")
input_folder.pack(padx=10,pady=10,fill="x")

# フォルダ選択ダイアログを配置
output_folder = FolderSelectionControl(app)
output_folder.label.configure(text="出力フォルダ名")
output_folder.pack(padx=10,pady=10,fill="x")

# 実行ボタンの配置
split_button = ctk.CTkButton(app,text="背景切り抜き",height=40,width=450,command=remove)
split_button.pack(padx=20,pady=10)

# -----------------------------------------------
# 前回実行時の画面入力値を復元
# -----------------------------------------------

input_folder.set(conf.get("input_folder"))
if input_folder.get() == "":
    input_folder.set(os.getcwd())

output_folder.set(conf.get("output_folder"))
if output_folder.get() == "":
    output_folder.set(os.getcwd())


# -----------------------------------------------
# CustomTkinterの起動 
# -----------------------------------------------

# メインループを開始
app.mainloop()

import os
import glob
from bg_remove import bg_remove


def get_files(folder,recursive=False,include_hidden=True):
    """
    指定されたフォルダ内のファイルを再帰的に取得する関数
    :param folder: フォルダのパス
    :param recursive: 再帰的に取得するかどうかのフラグ
    :param include_hidden: 隠しファイルを含めるかどうかのフラグ
    :return: ファイルのリスト
    """
    if os.path.isdir(folder):
        folder = os.path.join(folder,"*")
    if recursive:
        dir,name = os.path.split(folder)
        folder = os.path.join(dir,"**",name)
    return glob.glob(folder,recursive=recursive,include_hidden=include_hidden)


def bg_remove_from_folder(input_folder,output_folder):
    """
    指定されたフォルダ内のファイルの背景を切り抜きます
    :param input_folder: 入力フォルダのパス
    :param output_folder: 出力フォルダのパス
    """
    files = get_files(input_folder)
    for file in files:
        _,name = os.path.split(file)
        path = os.path.join(output_folder,name)
        bg_remove(file,path)
        print(file)

from rembg import remove
from PIL import Image
import numpy as np

def bg_remove(input_path, output_path):
    """
    指定されたファイルの背景を切り抜きます
    :param input_path: 入力ファイルのパス
    :param output_path: 出力ファイルのパス
    """
    # 入力画像を開く
    input_image = Image.open(input_path)
    
    # 画像をRGBAモードに変換（背景を切り抜くためにはアルファチャンネルが必要）
    if input_image.mode != 'RGBA':
        input_image = input_image.convert('RGBA')
    
    # 画像データをnumpy配列に変換
    input_array = np.array(input_image)
    
    # 背景を除去
    output_array = remove(input_array)
    
    # numpy配列をPIL画像に変換
    output_image = Image.fromarray(output_array)
    
    # 画像をPNG形式で保存（透明度を保持するため）
    output_image.save(output_path, 'PNG')