import os
import shutil

import gradio as gr
from PIL import Image
import uuid
import edimg


def replicate_edge(image, border_size):
    # Replicate the edges
    left = image.crop((0, 0, 1, image.height)).resize((border_size, image.height))
    right = image.crop((image.width - 1, 0, image.width, image.height)).resize((border_size, image.height))
    top = image.crop((0, 0, image.width, 1)).resize((image.width, border_size))
    bottom = image.crop((0, image.height - 1, image.width, image.height)).resize((image.width, border_size))

    # Create a new image with the replicated borders
    new_image = Image.new('RGB', (image.width + 2 * border_size, image.height + 2 * border_size))
    new_image.paste(image, (border_size, border_size))
    new_image.paste(left, (0, border_size))
    new_image.paste(right, (image.width + border_size, border_size))
    new_image.paste(top, (border_size, 0))
    new_image.paste(bottom, (border_size, image.height + border_size))

    # Replicate corners
    new_image.paste(image.getpixel((0, 0)), (0, 0, border_size, border_size))  # Top-left corner
    new_image.paste(image.getpixel((image.width - 1, 0)),
                    (image.width + border_size, 0, image.width + 2 * border_size, border_size))  # Top-right corner
    new_image.paste(image.getpixel((0, image.height - 1)),
                    (0, image.height + border_size, border_size, image.height + 2 * border_size))  # Bottom-left corner
    new_image.paste(image.getpixel((image.width - 1, image.height - 1)), (
    image.width + border_size, image.height + border_size, image.width + 2 * border_size,
    image.height + 2 * border_size))  # Bottom-right corner

    return new_image


def outpaint_image_with_pillow(image):

    # Define the amount of pixels to add on each side
    border_size = 42

    # Extend the image with replicated edges
    outpainted_image = replicate_edge(image, border_size)

    # Save the outpainted image

    return outpainted_image




def select(a, b, login):



    if a == None or b == None:
        gr.Warning('앞 뒷면 이미지를 먼저 만들어주세요')
        return None, None, None, None


    r, g, bs = b.getpixel((33, 33))
    b = edimg.rounded_color(b, r, g, bs)
    ap = outpaint_image_with_pillow(a)
    bp = outpaint_image_with_pillow(b)
    return ap, bp

def submit(ap, bp, login):

    if ap == None or bp == None:
        gr.Warning('앞 뒷면 이미지를 먼저 만들어주세요')
        return None
    # Generate a unique identifier
    unique_id = str(uuid.uuid4())[:18]

    # Define the target directory
    target_directory = f'/data/printing/{unique_id}'

    # Create the target directory if it doesn't exist
    if not os.path.exists(target_directory):
        os.makedirs(target_directory)

    ap.save(f"{target_directory}/front.png", dpi=(2400, 2400))
    bp.save(f"{target_directory}/back.png", dpi=(2400, 2400))
    #print(f'Files have been copied to {target_directory}')
    return unique_id

# Example usage
#outpaint_image('path_to_your_image.jpg', 'output_image.jpg')