Update app.py

"노이즈 회색으로"

app.py

@@ -26,14 +26,17 @@ def change_hair_to_blonde(image):
     image_blonde = cv2.cvtColor(hsv, cv2.COLOR_HSV2RGB)
     return image_blonde
 
-def add_noise(image):
+def add_gray_noise(image):
     # Convert to OpenCV format
     image_np = np.array(image)
+    # Change the whole image to gray
+    gray_image = np.full(image_np.shape, 128, dtype=np.uint8)
     # Generate random noise
-    noise = np.random.normal(0, 25, image_np.shape).astype(np.uint8)
-    # Add noise to the image
-    noisy_image = cv2.add(image_np, noise)
-    return noisy_image
+    num_spots = 50
+    for _ in range(num_spots):
+        x, y = np.random.randint(0, image_np.shape[1]), np.random.randint(0, image_np.shape[0])
+        cv2.circle(gray_image, (x, y), 10, (50, 50, 50), -1)  # Add dark gray noise
+    return gray_image
 
 def apply_deepfake(image):
     # Convert PIL image to bytes
@@ -67,9 +70,9 @@ if uploaded_file is not None:
         # Just display the original image
         st.image(image, caption='Original Image.', use_column_width=True)
     elif action == 'B':
-        # Add noise to the original image
-        noisy_image = add_noise(image)
-        st.image(noisy_image, caption='Image with Noise.', use_column_width=True)
+        # Add gray noise to the original image
+        gray_noisy_image = add_gray_noise(image)
+        st.image(gray_noisy_image, caption='Image with Gray Noise.', use_column_width=True)
     elif action == 'Deepfake':
         # Apply deepfake transformation
         deepfake_image = apply_deepfake(image)