Update app.py

app.py

@@ -1,13 +1,39 @@
 import gradio as gr
-from PIL import Image
-import numpy as np
-
+# Modified matrices with improved red/cyan separation
 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 ] ],
+    '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]
+    ],
+    # New matrices with improved red/cyan separation
+    'enhanced': [
+        [0.9, 0, 0, 0, 0, 0, 0, 0, 0],     # Left image: strong red only
+        [0, 0, 0, 0, 0.95, 0, 0, 0, 0.95]  # Right image: strong green and blue
+    ],
+    'true_enhanced': [
+        [0.4, 0.5, 0.1, 0, 0, 0, 0, 0, 0],  # Left image: red channel from grayscale
+        [0, 0, 0, 0, 0, 0, 0, 0.6, 0.4]     # Right image: only blue and green
+    ],
+    'dark_red': [
+        [0.7, 0, 0, 0, 0, 0, 0, 0, 0],      # Left image: slightly reduced red
+        [0, 0, 0, 0, 1, 0, 0, 0, 1]         # Right image: full green and blue
+    ]
 }
 
 def make_anaglyph(left_img, right_img, color_method):
@@ -24,14 +50,14 @@ def make_anaglyph(left_img, right_img, color_method):
         # 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()
+    # Create a new image for the result (important change)
+    result = Image.new("RGB", left.size)
     
     # Get the pixel maps
     width, height = left.size
     leftMap = left.load()
     rightMap = right.load()
-    #resultMap = result.load()
+    resultMap = result.load()
     
     # Use the selected color matrix
     m = matrices[color_method]
@@ -41,14 +67,15 @@ def make_anaglyph(left_img, right_img, color_method):
         for x in range(0, width):
             r1, g1, b1 = leftMap[x, y]
             r2, g2, b2 = rightMap[x, y]
-            leftMap[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(left)
+    return np.array(result)
 
 def make_stereopair(left_img, right_img, color_method):
     """Generate a stereo pair from left and right images"""
@@ -124,7 +151,7 @@ with gr.Blocks(css=css) as app:
                 )
                     
                 color_method = gr.Radio(
-                    ["optimized", "true", "mono", "color", "halfcolor"], 
+                    ["optimized", "true", "mono", "color", "halfcolor", "enhanced", "true_enhanced", "dark_red"], 
                     label="Color Method", 
                     value="optimized",
                     info="Select the color processing method"