Update app.py

app.py

@@ -4,13 +4,17 @@ import numpy as np
 from PIL import Image
 import io
 
-# Function to convert image to sketch
-def image_to_sketch(image):
+# Function to convert image to sketch with darkening factor
+def image_to_sketch(image, darken_factor=1.0):
     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]))
+    
     return sketch
 
 # Streamlit app layout
@@ -93,8 +97,11 @@ if uploaded_file is not None:
     
     st.write("Converting...")
 
-    # Convert the image to a sketch
-    sketch = image_to_sketch(image_np)
+    # Darkening factor slider
+    darken_factor = st.slider('Adjust Darkening Factor', 0.1, 1.0, 1.0, 0.1)
+    
+    # Convert the image to a sketch with the selected darkening factor
+    sketch = image_to_sketch(image_np, darken_factor)
     
     col3, col4 = st.columns(2)