import os
import subprocess
import shlex

import shlex
import cv2

import cv2
import gradio as gr
import numpy as np 
# from src.demo.demo import *
# pretrained_model_path = "SG161222/RealVisXL_V5.0_Lightning"#




# model.run_moving





examples_moving = [
    [
    "examples/layer/01_horse/00.jpg", #bg
    "examples/layer/01_horse/mask0.jpg", #bg_mask
    0, 0, 1.2, "None", "left/right",  #l1_dx, l1_dy, l1_resize
    ],
    [
    "examples/moving/01_ball/0.jpg", #bg
    "examples/moving/01_ball/mask0.jpg", #bg_mask
    -0.2, -0.1, 0.8, "None", "None",  #l1_dx, l1_dy, l1_resize
    ],

    [
    "examples/moving/01_ball/0.jpg", #bg
    "examples/moving/01_ball/mask0.jpg", #bg_mask
    -1.0, -0.1, 0.8, "None", "None",  #l1_dx, l1_dy, l1_resize
    ],
    [
    "examples/moving/02_bell/0.jpg", #bg
    "examples/moving/02_bell/mask0.jpg", #bg_mask
    0, 0, 0.75, "None", "None",  #l1_dx, l1_dy, l1_resize
    ],
]


import uuid 
from PIL import Image

products_base_folder = "/home/rnd/Documents/Ameer/Dream/Lora_Replace_inpaint/products"

background_base_folder = "/home/rnd/Documents/Ameer/Dream/Lora_Replace_inpaint/results_lora_base_model"


print("os.listdir(products_base_folder) len ",os.listdir(products_base_folder))
for product_path  in ['15.png']:

    mask_image  =  Image.open(f"{products_base_folder}/{product_path}")
    folder_id  =  product_path.split(".")[0]
    for idx in range(2):
        
        from src.demo.model import DesignEdit
        
        pretrained_model_path = "stabilityai/stable-diffusion-xl-base-1.0"
        model =  DesignEdit(pretrained_model_path=pretrained_model_path)

        unique_id = str(uuid.uuid4())
        background_path_image =  f"{background_base_folder}/{folder_id}_{idx}/refiner_image_{idx}.png"
        background_image  =  Image.open(background_path_image)

        index  =   2

        bg_img, bg_mask   =  np.array(background_image), np.array(mask_image)

        bg_img = cv2.cvtColor(bg_img, cv2.COLOR_BGR2RGB)
        bg_mask = cv2.cvtColor(bg_mask, cv2.COLOR_BGR2RGB)


        bg_ori = bg_img
        l1_dx, l1_dy, l1_resize, l1_w_flip, l1_h_flip   =  -1.0, -0.1, 1.0, "None", "None"


        resized_img = model.run_moving(bg_img, bg_ori, bg_mask, l1_dx, l1_dy, l1_resize)


        cv2.imwrite(f"final_inpaint_image/{unique_id}_bg.png", bg_img)
        cv2.imwrite(f"final_inpaint_image/{unique_id}_mask.png", bg_mask)

        cv2.imwrite(f"final_inpaint_image/{unique_id}.png", resized_img[0])
    
        print(f"final_inpaint_image/{unique_id}.png")

        exit()