from PIL import Image

def create_hybrid_image():
    # Define the hex color codes to be used in the logic
    green_hex_codes = ['#06653d', '#11A75F', '#81bf6c', '#bae383', '#e6f3a4']
    red_hex_codes = ['#fbc07e', '#f7885a', '#ea4f3b', '#fff0b5']
    white_hex_code = '#ab0535'

    # Open the two images with alpha channels
    rvi = Image.open('Downloads/rvi.png').convert('RGBA')
    rsm = Image.open('Downloads/rsm.png').convert('RGBA')
    
    # Ensure the images are of the same size
    if rvi.size != rsm.size:
        raise ValueError("The images rvi.png and rsm.png must be of the same size.")
    
    # Create a new image with the same size and mode 'RGBA'
    hybrid = Image.new('RGBA', rvi.size)
    
    # Get the pixel data
    rvi_pixels = rvi.load()
    rsm_pixels = rsm.load()
    hybrid_pixels = hybrid.load()
    
    width, height = rvi.size
    
    for x in range(width):
        for y in range(height):
            rvi_pixel = rvi_pixels[x, y]
            rsm_pixel = rsm_pixels[x, y]
            
            rvi_pixel_hex = '#%02x%02x%02x' % rvi_pixel[:3]
            rsm_pixel_hex = '#%02x%02x%02x' % rsm_pixel[:3]
            rvi_alpha = rvi_pixel[3]
            
            if rvi_alpha == 0:
                # If the pixel in rvi.png has zero alpha, set it in hybrid.png too
                hybrid_pixels[x, y] = (0, 0, 0, 0)
            elif rvi_pixel_hex in green_hex_codes and rsm_pixel_hex in green_hex_codes:
                hybrid_pixels[x, y] = (17, 167, 95, rvi_alpha)  # #11A75F
            elif rvi_pixel_hex in red_hex_codes and rsm_pixel_hex in red_hex_codes:
                hybrid_pixels[x, y] = (145, 16, 44, rvi_alpha)  # #91102C
            elif rvi_pixel_hex == white_hex_code and rsm_pixel_hex == white_hex_code:
                hybrid_pixels[x, y] = (255, 255, 255, rvi_alpha)  # #FFFFFF
            elif rvi_pixel_hex in red_hex_codes and rsm_pixel_hex not in red_hex_codes:
                hybrid_pixels[x, y] = (247, 136, 90, rvi_alpha)  # #F7885A
            elif rsm_pixel_hex in red_hex_codes and rvi_pixel_hex not in red_hex_codes:
                hybrid_pixels[x, y] = (60, 19, 97, rvi_alpha)  # #3C1361
            else:
                # If none of the conditions match, use a default color (optional)
                hybrid_pixels[x, y] = (0, 0, 0, rvi_alpha)  # Black with rvi alpha value
    
    # Save the new hybrid image
    hybrid.save('Downloads/hybrid.png')

# Call the function to create the hybrid image
#create_hybrid_image()