import os
import random
from PIL import Image

def randomly_insert_wally(directory_path, wally_path, output_directory):
    # Get all image files in the directory
    image_files = [f for f in os.listdir(directory_path) if f.endswith(('.jpg', '.jpeg', '.png'))]
    
    # If no images found, exit
    if not image_files:
        print("No images found in the specified directory.")
        return

    # Select a random image from the directory
    random_image_path = os.path.join(directory_path, random.choice(image_files))
    
    # Open the background (randomly selected) image and Wally image
    background = Image.open(random_image_path).convert("RGBA")
    wally = Image.open(wally_path).convert("RGBA")

    # Resize Wally if it's too large for the background
    if wally.size[0] > background.size[0] or wally.size[1] > background.size[1]:
        wally = wally.resize((background.size[0] // 4, background.size[1] // 4))  # Resize to 1/4th of background size

    # Get background and Wally dimensions
    bg_width, bg_height = background.size
    wally_width, wally_height = wally.size

    # Generate random coordinates within the background for Wally placement
    x = random.randint(0, bg_width - wally_width)
    y = random.randint(0, bg_height - wally_height)

    # Create a copy of the background to paste Wally onto it
    combined = background.copy()

    # Paste Wally onto the background at the random coordinates with transparency
    combined.paste(wally, (x, y), wally)

    # Convert back to RGB if saving as JPEG (JPEG does not support alpha channel)
    final_image = combined.convert("RGB")

    # Ensure output directory exists
    if not os.path.exists(output_directory):
        os.makedirs(output_directory)

    # Save the result
    output_path = os.path.join(output_directory, f"output_{os.path.basename(random_image_path)}")
    final_image.save(output_path, format="JPEG")
    print(f"Saved combined image with Wally at {output_path}")

# Usage
directory_path = "images"
# Make sure this is a transparent .png file of Wally

wally_paths = ['symposium/waldo_transparent.png', 'symposium/woof_transparent.png', 'symposium/waldina_transparent.png']

# Run the function for each character

for postava in wally_paths:
    randomly_insert_wally(directory_path, postava, directory_path)


def append_hex_to_random_images(input_folder, output_folder, wally):
    # Check if the input folder exists
    if not os.path.exists(input_folder):
        print(f"The folder '{input_folder}' does not exist.")
        return

    # Ensure output folder exists
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)

    # Get all images in the input folder with allowed extensions
    images = [f for f in os.listdir(input_folder) if f.lower().endswith(('.jpg', '.jpeg', '.png'))]

    # Check if there are at least two images to proceed
    if len(images) < 2:
        print("Not enough images in the folder to perform the operation. Need at least two images.")
        return

    # Select two random images
    first_image_name = random.choice(images)
    first_image_path = os.path.join(input_folder, first_image_name)
    second_image_path = wally
    
    # Ensure the second image is different from the first
    while first_image_path == second_image_path:
        second_image_path = os.path.join(input_folder, random.choice(images))

    # Read the first image in binary mode
    with open(first_image_path, 'rb') as first_image_file:
        first_image_data = first_image_file.read()

    # Read the second image in binary mode
    with open(second_image_path, 'rb') as second_image_file:
        second_image_data = second_image_file.read()

    # Find the JPEG end marker (FFD9 in hex) in the first image data
    jpeg_end_marker = b'\xFF\xD9'
    end_marker_index = first_image_data.rfind(jpeg_end_marker)
    
    if end_marker_index == -1:
        print("End of JPEG marker not found in the first image.")
        return

    # Split the first image data at the end marker
    first_image_main = first_image_data[:end_marker_index + 2]  # Include the marker
    first_image_remainder = first_image_data[end_marker_index + 2:]

    # Concatenate: first image main data + hex of the second image + remainder of first image (if any)
    combined_data = first_image_main + second_image_data + first_image_remainder

    # Construct the output file path, using the same name as the first image
    output_image_path = os.path.join(output_folder, first_image_name)

    # Write the combined data to the output file
    with open(output_image_path, 'wb') as output_file:
        output_file.write(combined_data)

    print(f"Hex data of '{second_image_path}' appended to '{first_image_path}', saved as '{output_image_path}'.")

odlaw_path = 'symposium/odlaw_transparent.png'

#append_hex_to_random_images('images', 'images', odlaw_path)