from PIL import Image, ImageDraw, ImageFont
import os


def merge_imgs(img1_path, img2_path, save_path, similar=None, label=None):
        position = (50, 50)  # 文字的左上角坐标
        color = (255, 0, 0)  # 红色文字，格式为 RGB
        if not os.path.exists(os.sep.join([save_path, str(label)])):
            os.makedirs(os.sep.join([save_path, str(label)]))
        save_path = os.sep.join([save_path, str(label)])
        img_name = os.path.basename(img1_path).split('.')[0]+'_'+os.path.basename(img2_path).split('.')[0]+'.png'
        img1 = Image.open(img1_path)
        img2 = Image.open(img2_path)
        img1 = img1.resize((224,224))
        img2 = img2.resize((224,224))
        print('img1_path', img1)
        print('img2_path', img2)
        assert img1.height == img2.height

        new_img = Image.new('RGB', (img1.width + img2.width + 10, img1.height))

        # print('new_img', new_img)
        new_img.paste(img1, (0, 0))
        new_img.paste(img2, (img1.width + 10, 0))

        if similar is not None:
            similar = str(similar)+'_'+str(label)
            draw = ImageDraw.Draw(new_img)
            draw.text(position, str(similar), color, font_size=36)
        os.makedirs(save_path, exist_ok=True)
        img_save = os.path.join(save_path, img_name)
        new_img.save(img_save)