from PIL import Image  
import os 

def merge_imgs(path1, path2, save_path):
    for img_name in os.listdir(path1):
        # img_dir1 = os.path.join(path1, img_dir)
        # if os.path.isdir(img_dir1):
        # img_dir2 = os.path.join(path2, img_dir)
        # for img_name in os.listdir(img_dir1):
        try:
            img1_path = os.path.join(path1, img_name)
            img2_path = os.path.join(path2, img_name)

            img1 = Image.open(img1_path)
            img2 = Image.open(img2_path)
            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))
            # save_dir = os.path.join(save_path, img_name)
            os.makedirs(save_path, exist_ok=True)
            img_save = os.path.join(save_path, img_name)
            # print('img-save', img_save)
            new_img.save(img_save)
        except Exception as e:
            print(e)
            #print(img_name)
        
        
#path1 = '/home/yujia/yj/yolov5-6.1/0518_cls10_v5s_new_delTP0.5/labelFn_5/'
# path1 = 'predict5_ori_v10s/'
# path1 = 'predict_best0524/'
# path2 = 'predict_0613_epoch27/'

# path1 = 'predict_labels110_0524/'
# path2 = 'predict_labels110_0613/'

path1 = '/home/lc/ieemoo-ai-gift/confusion_gift_cls4_0.45/FN/FN_3'
path2 = '/home/lc/ieemoo-ai-gift/confusion_gift_cls4_0.45/allBox/allBox_1'

#save_path = '/home/yujia/yj/yolov5-6.1/0518_cls10_v5s_new_delTP0.5/labelFn_5_allBox_merge/'
save_path = '/home/lc/ieemoo-ai-gift/confusion_gift_cls4_0.45/FN/FN_3_joint/'
# os.makedirs(save_path, exist_ok=True)
merge_imgs(path1, path2, save_path)