app.py

import gradio as gr
from PIL import Image
import numpy as np

matrices = {
    'true': [ [ 0.299, 0.587, 0.114, 0, 0, 0, 0, 0, 0 ], [ 0, 0, 0, 0, 0, 0, 0.299, 0.587, 0.114 ] ],
    'mono': [ [ 0.299, 0.587, 0.114, 0, 0, 0, 0, 0, 0 ], [ 0, 0, 0, 0.299, 0.587, 0.114, 0.299, 0.587, 0.114 ] ],
    'color': [ [ 1, 0, 0, 0, 0, 0, 0, 0, 0 ], [ 0, 0, 0, 0, 1, 0, 0, 0, 1 ] ],
    'halfcolor': [ [ 0.299, 0.587, 0.114, 0, 0, 0, 0, 0, 0 ], [ 0, 0, 0, 0, 1, 0, 0, 0, 1 ] ],
    'optimized': [ [ 0, 0.7, 0.3, 0, 0, 0, 0, 0, 0 ], [ 0, 0, 0, 0, 1, 0, 0, 0, 1 ] ],
}

def make_anaglyph(left_img, right_img):
    """Generate an optimized anaglyph from left and right images"""
    if left_img is None or right_img is None:
        return None
    
    # Convert from numpy array (from Gradio) to PIL Image
    left = Image.fromarray(left_img)
    right = Image.fromarray(right_img)
    
    # Check if both images have the same dimensions
    if left.size != right.size:
        # Resize right image to match left image dimensions
        right = right.resize(left.size, Image.LANCZOS)
    
    # Create a copy of the left image to modify
    result = left.copy()
    
    # Get the pixel maps
    width, height = left.size
    leftMap = left.load()
    rightMap = right.load()
    resultMap = result.load()
    
    # Use the optimized color matrix
    m = matrices['optimized']
    
    # Apply the anaglyph transformation
    for y in range(0, height):
        for x in range(0, width):
            r1, g1, b1 = leftMap[x, y]
            r2, g2, b2 = rightMap[x, y]
            resultMap[x, y] = (
                int(r1*m[0][0] + g1*m[0][1] + b1*m[0][2] + r2*m[1][0] + g2*m[1][1] + b2*m[1][2]),
                int(r1*m[0][3] + g1*m[0][4] + b1*m[0][5] + r2*m[1][3] + g2*m[1][4] + b2*m[1][5]),
                int(r1*m[0][6] + g1*m[0][7] + b1*m[0][8] + r2*m[1][6] + g2*m[1][7] + b2*m[1][8])
            )
    
    # Convert back to numpy array for Gradio
    return np.array(result)

# Create the Gradio interface
with gr.Blocks(title="3D Anaglyph Generator") as app:
    gr.Markdown("# 3D Anaglyph Generator")
    gr.Markdown("Upload left and right images to create a 3D anaglyph with optimized color settings.")
    
    with gr.Row():
        with gr.Column():
            left_input = gr.Image(label="Left Image")
        with gr.Column():
            right_input = gr.Image(label="Right Image")
    
    generate_btn = gr.Button("Generate Anaglyph", variant="primary")
    
    output = gr.Image(label="Generated Anaglyph (Red-Cyan)")
    
    gr.Markdown("""
    ### How to use:
    1. Upload a left-eye image
    2. Upload a right-eye image
    3. Click "Generate Anaglyph"
    4. View or download the resulting anaglyph
    
    For best results, use red-cyan 3D glasses to view the generated anaglyph.
    """)
    
    generate_btn.click(
        fn=make_anaglyph,
        inputs=[left_input, right_input],
        outputs=output
    )

# Launch the app
if __name__ == "__main__":
    app.launch()