Update app.py

"회색 추가를 어두운 노이즈로 변경

app.py

@@ -26,17 +26,14 @@ def change_hair_to_blonde(image):
     image_blonde = cv2.cvtColor(hsv, cv2.COLOR_HSV2RGB)
     return image_blonde
 
-def add_gray_noise(image):
+def add_dark_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
-    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
+    # Generate random dark noise
+    noise = np.random.randint(0, 50, image_np.shape, dtype=np.uint8)
+    # Subtract noise from the image
+    noisy_image = cv2.subtract(image_np, noise)
+    return noisy_image
 
 def apply_deepfake(image):
     # Convert PIL image to bytes
@@ -70,9 +67,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 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)
+        # Add dark noise to the original image
+        dark_noisy_image = add_dark_noise(image)
+        st.image(dark_noisy_image, caption='Image with Dark Noise.', use_column_width=True)
     elif action == 'Deepfake':
         # Apply deepfake transformation
         deepfake_image = apply_deepfake(image)