from PIL import Image
import os
import sys
import glob

def create_thumbnail(input_image_path, background_path, rainy_still_path):
    try:
        print(f"Processing input image: {input_image_path}")
        # Open the images and ensure they are in RGBA mode
        input_image = Image.open(input_image_path).convert("RGBA")
        background_image = Image.open(background_path).convert("RGBA")
        rainy_still_image = Image.open(rainy_still_path).convert("RGBA")

        # Scale the input image up by 5%
        scaled_width = int(input_image.width * 1.05)
        scaled_height = int(input_image.height * 1.05)
        input_image = input_image.resize((scaled_width, scaled_height), Image.LANCZOS)

        # Create a new image with the same size as the background
        combined_image = Image.new('RGBA', background_image.size)

        # Paste the background
        combined_image.paste(background_image, (0, 0))

        # Calculate position to center the input image
        input_x = (combined_image.width - input_image.width) // 2
        input_y = (combined_image.height - input_image.height) // 2

        # Paste the input image on top
        combined_image.paste(input_image, (input_x, input_y), input_image)

        # Move RainyStill down and to the right by 783 pixels
        rainy_still_position = (783, 783)  # Adjust as needed
        combined_image.paste(rainy_still_image, rainy_still_position, rainy_still_image)

        return combined_image  # Return the combined image

    except Exception as e:
        print(f"Error: {e}")  # Print the error message for debugging
        return None  # Return None if there was an error

if __name__ == "__main__":
    # Define the paths for the background and RainyStill images
    script_dir = os.path.dirname(os.path.abspath(__file__))
    background_path = os.path.join(script_dir, "background.png")
    rainy_still_path = os.path.join(script_dir, "RainyStill.png")

    # Check if the background and RainyStill images exist
    if not os.path.exists(background_path):
        print(f"Background image not found: {background_path}")
        sys.exit(1)

    if not os.path.exists(rainy_still_path):
        print(f"RainyStill image not found: {rainy_still_path}")
        sys.exit(1)

    # Find all output* files in the current directory
    output_files = glob.glob(os.path.join(script_dir, "output*"))

    # Process each output file
    for input_image_path in output_files:
        # Generate a unique output image path in the script directory
        output_image_path = os.path.join(script_dir, "output_thumb.jpg")  # Start with output_thumb.jpg

        # Ensure the output file does not overwrite existing files
        counter = 1
        while os.path.exists(output_image_path):
            output_image_path = os.path.join(script_dir, f"output_thumb_{counter}.jpg")
            counter += 1

        # Call the function to create the thumbnail
        combined_image = create_thumbnail(input_image_path, background_path, rainy_still_path)

        # Save the combined image if it was created successfully
        if combined_image is not None:
            # Convert to RGB mode for JPG (JPG doesn't support alpha)
            rgb_image = Image.new("RGB", combined_image.size, (255, 255, 255))
            rgb_image.paste(combined_image, mask=combined_image.split()[3])  # Use alpha as mask
            
            # Save as JPG with quality parameter to control file size
            rgb_image.save(output_image_path, format='JPEG', quality=100)  # Higher quality for better image fidelity
            print(f"JPG saved successfully at: {output_image_path}")  # Log success
        else:
            print(f"Failed to process {input_image_path}.")