Update app.py

app.py

@@ -25,38 +25,25 @@ def calculate_storage(shape, r_value):
     }
 
 def process_image(image_data, r_value):
-    # Open image as RGB
+    # Open image and convert to grayscale
     img = Image.open(io.BytesIO(image_data))
-    img_array = np.array(img)
     
-    # Check if image is RGB (3 channels)
-    if len(img_array.shape) == 3 and img_array.shape[2] == 3:
-        # Process each channel separately
-        reconstructed = np.zeros_like(img_array)
-        
-        for channel in range(3):
-            # Extract the channel
-            channel_data = img_array[:, :, channel]
-            
-            # Perform SVD on this channel
-            U, s, Vt = svd(channel_data, full_matrices=False)
-            
-            # Reconstruct image with r singular values
-            r = min(r_value, len(s))
-            reconstructed[:, :, channel] = np.dot(U[:, :r] * s[:r], Vt[:r, :])
-        
-        # Clip values to valid range and convert to uint8
-        reconstructed = np.clip(reconstructed, 0, 255).astype(np.uint8)
-    else:
-        # Fallback to grayscale processing if not RGB
-        img_array = np.array(img.convert('L'))
-        U, s, Vt = svd(img_array, full_matrices=False)
-        r = min(r_value, len(s))
-        reconstructed = np.dot(U[:, :r] * s[:r], Vt[:r, :])
-        reconstructed = np.clip(reconstructed, 0, 255).astype(np.uint8)
+    # Always convert to grayscale
+    grayscale_img = img.convert('L')
+    img_array = np.array(grayscale_img)
+    
+    # Perform SVD on the grayscale data
+    U, s, Vt = svd(img_array, full_matrices=False)
+    
+    # Reconstruct image with r singular values
+    r = min(r_value, len(s))
+    reconstructed = np.dot(U[:, :r] * s[:r], Vt[:r, :])
+    
+    # Clip values to valid range and convert to uint8
+    reconstructed = np.clip(reconstructed, 0, 255).astype(np.uint8)
     
     # Convert back to image
-    reconstructed_img = Image.fromarray(reconstructed)
+    reconstructed_img = Image.fromarray(reconstructed, mode='L')
     
     # Save to base64 string
     buffered = io.BytesIO()