Update app.py

app.py

@@ -23,22 +23,26 @@ def image_to_sketch(image, kernel_size=1):
 
 # Function to blur the background of the image
 def blur_background(image, blur_strength=31):
-    # Convert the image to grayscale
-    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+    # # Convert the image to grayscale
+    # gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
     
-    # Use thresholding to create a binary mask
-    _, mask = cv2.threshold(gray, 120, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)
+    # # Use thresholding to create a binary mask
+    # _, mask = cv2.threshold(gray, 120, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)
     
-    # Create the inverse mask
-    mask_inv = cv2.bitwise_not(mask)
+    # # Create the inverse mask
+    # mask_inv = cv2.bitwise_not(mask)
     
-    # Blur the entire image
-    blurred = cv2.GaussianBlur(image, (blur_strength, blur_strength), 0)
+    # # Blur the entire image
+    # blurred = cv2.GaussianBlur(image, (blur_strength, blur_strength), 0)
     
-    # Use the mask to combine the original image with the blurred image
-    background = cv2.bitwise_and(blurred, blurred, mask=mask)
-    foreground = cv2.bitwise_and(image, image, mask=mask_inv)
-    combined = cv2.add(background, foreground)
+    # # Use the mask to combine the original image with the blurred image
+    # background = cv2.bitwise_and(blurred, blurred, mask=mask)
+    # foreground = cv2.bitwise_and(image, image, mask=mask_inv)
+    # combined = cv2.add(background, foreground)
+    ksize = (10, 10) 
+  
+# Using cv2.blur() method  
+    combined = cv2.blur(image, ksize)  
     
     return combined
 
@@ -97,7 +101,7 @@ example_image = cv2.imread(example_image_path)
 if example_image is not None:
     # Convert BGR to RGB for correct color display
     example_image_rgb = cv2.cvtColor(example_image, cv2.COLOR_BGR2RGB)
-    example_image_blurred = blur_background(example_image_rgb)
+    example_image_blurred = blur_background(example_image)
     example_sketch = image_to_sketch(example_image_blurred)
 
     col1, col2 = st.columns(2)