Update app.py

app.py

@@ -4,16 +4,17 @@ import numpy as np
 from PIL import Image
 import io
 
-# Function to convert image to sketch with darkening factor
-def image_to_sketch(image, darken_factor=1.0):
+# Function to convert image to sketch with adjustable outline thickness
+def image_to_sketch(image, kernel_size=3):
     gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
     inverted_image = 255 - gray_image
     blurred_image = cv2.GaussianBlur(inverted_image, (31, 31), 0)
     inverted_blurred = 255 - blurred_image
     sketch = cv2.divide(gray_image, inverted_blurred, scale=256.0)
     
-    # Apply darkening factor
-    sketch = cv2.multiply(sketch, np.array([darken_factor]))
+    # Apply morphological operation to thicken the outlines
+    kernel = np.ones((kernel_size, kernel_size), np.uint8)
+    sketch = cv2.dilate(sketch, kernel, iterations=1)
     
     return sketch
 
@@ -97,11 +98,11 @@ if uploaded_file is not None:
     
     st.write("Converting...")
 
-    # Darkening factor slider
-    darken_factor = st.slider('Adjust Darkening Factor', 0.1, 1.0, 1.0, 0.1)
+    # Kernel size slider for adjusting outline thickness
+    kernel_size = st.slider('Adjust Outline Thickness', 1, 10, 3)
     
-    # Convert the image to a sketch with the selected darkening factor
-    sketch = image_to_sketch(image_np, darken_factor)
+    # Convert the image to a sketch with the selected kernel size
+    sketch = image_to_sketch(image_np, kernel_size)
     
     col3, col4 = st.columns(2)