# waifu2xをインストールする
!pip install waifu2x

# 変換する関数
import os

def convert_images(input_dir, output_dir):
    for root, dirs, files in os.walk(input_dir):
        relative_path = os.path.relpath(root, input_dir)
        output_subdir = os.path.join(output_dir, relative_path)
        os.makedirs(output_subdir, exist_ok=True)

        for filename in files:
            if filename.lower().endswith(('.png', '.jpg', '.jpeg')):
                input_file = os.path.join(root, filename)
                output_file = os.path.join(output_subdir, filename)
                
                # ノイズレベルと拡大レートはお好みで 
                !waifu2x -g 0 -i {input_file} -o {output_file} --method noise_scale --noise_level 1 --scale_ratio 2


# Stable Diffusionのアウトプット先の指定
input_dir = "/content/stable-diffusion-webui/outputs"

# waifu2xで変換したファイルの出力先
output_dir = "/content/converted_images"

convert_images(input_dir, output_dir)