Update app.py

app.py

@@ -17,18 +17,27 @@ def cartoonize_image(img, median_blur_value, block_size, c_value, d_value, sigma
                                   blockSize=block_size,
                                   C=c_value)
 
-    # Bilateral filter for color smoothing
+    # Bilateral filter for color smoothing (reduce colors)
     color = cv2.bilateralFilter(img, d=d_value, sigmaColor=sigma_color, sigmaSpace=sigma_space)
 
+    # Convert color image to grayscale and then back to BGR
+    color_gray = cv2.cvtColor(color, cv2.COLOR_BGR2GRAY)
+    
+    # Reduce colors by applying a threshold on the grayscale image
+    _, color_thresh = cv2.threshold(color_gray, 200, 255, cv2.THRESH_BINARY_INV)
+
     # Create a sketch effect by blending edges with the smoothed color image
     sketch = cv2.bitwise_and(color, color, mask=edges)
 
     # Ensure both images are of the same size before blending
-    if color.shape[:2] != sketch.shape[:2]:
-        sketch = cv2.resize(sketch, (color.shape[1], color.shape[0]))
+    if sketch.shape[:2] != img.shape[:2]:
+        sketch = cv2.resize(sketch, (img.shape[1], img.shape[0]))
+
+    # Combine the sketch with a desaturated version of the original image
+    desaturated_color = cv2.addWeighted(color, 0.3, np.zeros_like(color), 0.7, 0)  # Reduce saturation
 
-    # Combine the sketch with the original color image to create a more artistic effect
-    cartoon = cv2.addWeighted(color, 0.5, sketch, 0.5, 0)
+    # Blend sketch and desaturated color image
+    cartoon = cv2.addWeighted(desaturated_color, 0.5, sketch, 0.5, 0)
 
     return cartoon
 
@@ -51,8 +60,8 @@ if uploaded_file is not None:
         block_size = st.slider("Block Size", min_value=3, max_value=21, value=9, step=2)
         c_value = st.slider("C Value", min_value=0, max_value=20, value=5)  # Increased default value for more effect
         d_value = st.slider("Diameter (d)", min_value=1, max_value=15, value=9)
-        sigma_color = st.slider("Sigma Color", min_value=0, max_value=400, value=150)  # Reduced for more sketch effect
-        sigma_space = st.slider("Sigma Space", min_value=0, max_value=400, value=150)  # Reduced for more sketch effect
+        sigma_color = st.slider("Sigma Color", min_value=0, max_value=400, value=100)  # Reduced for more sketch effect
+        sigma_space = st.slider("Sigma Space", min_value=0, max_value=400, value=100)  # Reduced for more sketch effect
 
     # Show a spinner while processing the cartoonization
     with st.spinner('Processing...'):