Update app.py

app.py

@@ -4,21 +4,21 @@ import numpy as np
 from PIL import Image
 
 # Function to convert image to sketch
-def convert_to_sketch(image):
+def convert_to_sketch(image, blur_intensity, sketch_intensity):
     # Convert image to grayscale
     gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
     
     # Invert the grayscale image
     inverted_gray_image = 255 - gray_image
     
-    # Apply Gaussian Blur
-    blurred_image = cv2.GaussianBlur(inverted_gray_image, (21, 21), 0)
+    # Apply Gaussian Blur with adjustable intensity
+    blurred_image = cv2.GaussianBlur(inverted_gray_image, (blur_intensity, blur_intensity), 0)
     
     # Invert the blurred image
     inverted_blurred_image = 255 - blurred_image
     
-    # Create the pencil sketch image
-    sketch_image = cv2.divide(gray_image, inverted_blurred_image, scale=256.0)
+    # Create the pencil sketch image with adjustable intensity
+    sketch_image = cv2.divide(gray_image, inverted_blurred_image, scale=sketch_intensity)
     
     return sketch_image
 
@@ -33,13 +33,18 @@ st.image(logo_url, width=200)  # Adjust the width as needed
 # Upload image - allow all image formats
 uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png", "bmp", "gif", "tiff", "webp"])
 
+# Add settings in the sidebar
+st.sidebar.title("Settings")
+blur_intensity = st.sidebar.slider("Blur Intensity", min_value=1, max_value=99, value=21, step=2)
+sketch_intensity = st.sidebar.slider("Sketch Intensity", min_value=100, max_value=300, value=256, step=10)
+
 if uploaded_file is not None:
     # Read the image
     image = Image.open(uploaded_file)
     image = np.array(image)
     
-    # Convert the image to sketch
-    sketch_image = convert_to_sketch(image)
+    # Convert the image to sketch using the selected settings
+    sketch_image = convert_to_sketch(image, blur_intensity, sketch_intensity)
     
     # Display the original image
     st.image(image, caption='Original Image', use_column_width=True)