Update app.py

app.py

@@ -24,11 +24,15 @@ def add_noise(image, password):
 
 def remove_noise(noisy_image, password):
     """Attempt to remove noise from the image using the same password."""
+    # Check if the noisy_image is a numpy array and convert it to PIL Image if so
+    if isinstance(noisy_image, np.ndarray):
+        noisy_image = Image.fromarray(noisy_image.astype('uint8'))
+
     seed = hash_password(password) % (2**32)
     np.random.seed(seed)  # Reset the seed to generate the same noise parameters
     mean = np.random.uniform(-0.05, 0.05)
     var = np.random.uniform(0.001, 0.01)
-    image = noisy_image.convert('L')
+    image = noisy_image.convert('L')  # Ensure image is in grayscale
     image_np = np.array(image)
     sigma_est = np.mean(estimate_sigma(image_np, multichannel=False))
     denoised_image = denoise_nl_means(image_np, h=1.15 * sigma_est, fast_mode=True,
@@ -36,6 +40,7 @@ def remove_noise(noisy_image, password):
     denoised_image = (255 * denoised_image).astype(np.uint8)  # Scale back to 0-255
     return Image.fromarray(denoised_image)
 
+
 # Define Gradio interface with tabs for adding and removing noise
 with gr.Blocks() as interface:
     gr.Markdown("### Image Noise Encryption and Decryption App")