Update app.py

app.py

@@ -10,8 +10,8 @@ matrices = {
     '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"""
+def make_anaglyph(left_img, right_img, color_method):
+    """Generate an anaglyph from left and right images using the specified color method"""
     if left_img is None or right_img is None:
         return None
     
@@ -33,8 +33,8 @@ def make_anaglyph(left_img, right_img):
     rightMap = right.load()
     resultMap = result.load()
     
-    # Use the optimized color matrix
-    m = matrices['optimized']
+    # Use the selected color matrix
+    m = matrices[color_method]
     
     # Apply the anaglyph transformation
     for y in range(0, height):
@@ -50,10 +50,55 @@ def make_anaglyph(left_img, right_img):
     # Convert back to numpy array for Gradio
     return np.array(result)
 
+def make_stereopair(left_img, right_img, color_method):
+    """Generate a stereo pair 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)
+    
+    width, height = left.size
+    leftMap = left.load()
+    rightMap = right.load()
+    
+    # Create a new image twice as wide
+    pair = Image.new('RGB', (width * 2, height))
+    pairMap = pair.load()
+    
+    # Copy the left and right images side by side
+    for y in range(0, height):
+        for x in range(0, width):
+            pairMap[x, y] = leftMap[x, y]
+            pairMap[x + width, y] = rightMap[x, y]
+    
+    # Convert to monochrome if required
+    if color_method == 'mono':
+        pair = pair.convert('L')
+    
+    # Convert back to numpy array for Gradio
+    return np.array(pair)
+
+def process_images(left_img, right_img, method, color_method):
+    """Process images based on the selected method"""
+    if method == "anaglyph":
+        return make_anaglyph(left_img, right_img, color_method)
+    elif method == "parallel":
+        return make_stereopair(left_img, right_img, color_method)
+    elif method == "crossed":
+        return make_stereopair(right_img, left_img, color_method)
+    return None
+
 # 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.Blocks(title="3D Image Generator") as app:
+    gr.Markdown("# 3D Image Generator")
+    gr.Markdown("Upload left and right images to create 3D images using different methods.")
     
     with gr.Row():
         with gr.Column():
@@ -61,23 +106,43 @@ with gr.Blocks(title="3D Anaglyph Generator") as app:
         with gr.Column():
             right_input = gr.Image(label="Right Image")
     
-    generate_btn = gr.Button("Generate Anaglyph", variant="primary")
+    with gr.Row():
+        with gr.Column():
+            method = gr.Radio(
+                ["anaglyph", "parallel", "crossed"], 
+                label="Method", 
+                value="anaglyph",
+                info="Select the 3D image creation method"
+            )
+        with gr.Column():
+            color_method = gr.Radio(
+                ["optimized", "true", "mono", "color", "halfcolor"], 
+                label="Color Method", 
+                value="optimized",
+                info="Select the color processing method"
+            )
+    
+    generate_btn = gr.Button("Generate 3D Image", variant="primary")
     
-    output = gr.Image(label="Generated Anaglyph (Red-Cyan)")
+    output = gr.Image(label="Generated 3D Image")
     
     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
+    ### Methods:
+    - **anaglyph**: Creates a red-cyan 3D image (requires 3D glasses)
+    - **parallel**: Creates side-by-side images for parallel viewing
+    - **crossed**: Creates side-by-side images for cross-eyed viewing
     
-    For best results, use red-cyan 3D glasses to view the generated anaglyph.
+    ### Color Methods:
+    - **optimized**: Best for most images (default)
+    - **true**: True color anaglyph
+    - **mono**: Monochrome output
+    - **color**: Full color (may cause ghosting)
+    - **halfcolor**: Balance between color and depth
     """)
     
     generate_btn.click(
-        fn=make_anaglyph,
-        inputs=[left_input, right_input],
+        fn=process_images,
+        inputs=[left_input, right_input, method, color_method],
         outputs=output
     )